mirror of
https://github.com/python-kasa/python-kasa.git
synced 2024-12-23 19:53:34 +00:00
657 lines
16 KiB
Python
657 lines
16 KiB
Python
|
import datetime
|
||
|
|
||
|
from unittest.mock import patch
|
||
|
|
||
|
import pytest
|
||
|
|
||
|
from pyHS100 import DeviceType, SmartStripException, SmartDeviceException
|
||
|
from .newfakes import (
|
||
|
BULB_SCHEMA,
|
||
|
PLUG_SCHEMA,
|
||
|
FakeTransportProtocol,
|
||
|
CURRENT_CONSUMPTION_SCHEMA,
|
||
|
TZ_SCHEMA,
|
||
|
)
|
||
|
from .conftest import (
|
||
|
turn_on,
|
||
|
handle_turn_on,
|
||
|
plug,
|
||
|
strip,
|
||
|
bulb,
|
||
|
color_bulb,
|
||
|
non_color_bulb,
|
||
|
has_emeter,
|
||
|
no_emeter,
|
||
|
dimmable,
|
||
|
non_dimmable,
|
||
|
variable_temp,
|
||
|
non_variable_temp,
|
||
|
)
|
||
|
|
||
|
|
||
|
@plug
|
||
|
def test_plug_sysinfo(dev):
|
||
|
assert dev.sys_info is not None
|
||
|
PLUG_SCHEMA(dev.sys_info)
|
||
|
|
||
|
assert dev.model is not None
|
||
|
|
||
|
assert dev.device_type == DeviceType.Plug or dev.device_type == DeviceType.Strip
|
||
|
assert dev.is_plug or dev.is_strip
|
||
|
|
||
|
|
||
|
@bulb
|
||
|
def test_bulb_sysinfo(dev):
|
||
|
assert dev.sys_info is not None
|
||
|
BULB_SCHEMA(dev.sys_info)
|
||
|
|
||
|
assert dev.model is not None
|
||
|
|
||
|
assert dev.device_type == DeviceType.Bulb
|
||
|
assert dev.is_bulb
|
||
|
|
||
|
|
||
|
def test_state_info(dev):
|
||
|
assert isinstance(dev.state_information, dict)
|
||
|
|
||
|
|
||
|
def test_invalid_connection(dev):
|
||
|
with patch.object(FakeTransportProtocol, "query", side_effect=SmartDeviceException):
|
||
|
with pytest.raises(SmartDeviceException):
|
||
|
dev.is_on
|
||
|
|
||
|
|
||
|
def test_query_helper(dev):
|
||
|
with pytest.raises(SmartDeviceException):
|
||
|
dev._query_helper("test", "testcmd", {})
|
||
|
# TODO check for unwrapping?
|
||
|
|
||
|
|
||
|
def test_deprecated_state(dev):
|
||
|
with pytest.deprecated_call():
|
||
|
dev.state = "OFF"
|
||
|
assert dev.state == "OFF"
|
||
|
assert not dev.is_on
|
||
|
|
||
|
with pytest.deprecated_call():
|
||
|
dev.state = "ON"
|
||
|
assert dev.state == "ON"
|
||
|
assert dev.is_on
|
||
|
|
||
|
with pytest.deprecated_call():
|
||
|
with pytest.raises(ValueError):
|
||
|
dev.state = "foo"
|
||
|
|
||
|
with pytest.deprecated_call():
|
||
|
with pytest.raises(ValueError):
|
||
|
dev.state = 1234
|
||
|
|
||
|
|
||
|
def test_deprecated_alias(dev):
|
||
|
with pytest.deprecated_call():
|
||
|
dev.alias = "foo"
|
||
|
|
||
|
|
||
|
def test_deprecated_mac(dev):
|
||
|
with pytest.deprecated_call():
|
||
|
dev.mac = 123123123123
|
||
|
|
||
|
|
||
|
@plug
|
||
|
def test_deprecated_led(dev):
|
||
|
with pytest.deprecated_call():
|
||
|
dev.led = True
|
||
|
|
||
|
|
||
|
@turn_on
|
||
|
def test_state(dev, turn_on):
|
||
|
handle_turn_on(dev, turn_on)
|
||
|
orig_state = dev.is_on
|
||
|
if orig_state:
|
||
|
dev.turn_off()
|
||
|
assert not dev.is_on
|
||
|
assert dev.is_off
|
||
|
dev.turn_on()
|
||
|
assert dev.is_on
|
||
|
assert not dev.is_off
|
||
|
else:
|
||
|
dev.turn_on()
|
||
|
assert dev.is_on
|
||
|
assert not dev.is_off
|
||
|
dev.turn_off()
|
||
|
assert not dev.is_on
|
||
|
assert dev.is_off
|
||
|
|
||
|
|
||
|
@no_emeter
|
||
|
def test_no_emeter(dev):
|
||
|
assert not dev.has_emeter
|
||
|
|
||
|
with pytest.raises(SmartDeviceException):
|
||
|
dev.get_emeter_realtime()
|
||
|
with pytest.raises(SmartDeviceException):
|
||
|
dev.get_emeter_daily()
|
||
|
with pytest.raises(SmartDeviceException):
|
||
|
dev.get_emeter_monthly()
|
||
|
with pytest.raises(SmartDeviceException):
|
||
|
dev.erase_emeter_stats()
|
||
|
|
||
|
|
||
|
@has_emeter
|
||
|
def test_get_emeter_realtime(dev):
|
||
|
if dev.is_strip:
|
||
|
pytest.skip("Disabled for HS300 temporarily")
|
||
|
|
||
|
assert dev.has_emeter
|
||
|
|
||
|
current_emeter = dev.get_emeter_realtime()
|
||
|
CURRENT_CONSUMPTION_SCHEMA(current_emeter)
|
||
|
|
||
|
|
||
|
@has_emeter
|
||
|
def test_get_emeter_daily(dev):
|
||
|
if dev.is_strip:
|
||
|
pytest.skip("Disabled for HS300 temporarily")
|
||
|
|
||
|
assert dev.has_emeter
|
||
|
|
||
|
assert dev.get_emeter_daily(year=1900, month=1) == {}
|
||
|
|
||
|
d = dev.get_emeter_daily()
|
||
|
assert len(d) > 0
|
||
|
|
||
|
k, v = d.popitem()
|
||
|
assert isinstance(k, int)
|
||
|
assert isinstance(v, float)
|
||
|
|
||
|
# Test kwh (energy, energy_wh)
|
||
|
d = dev.get_emeter_daily(kwh=False)
|
||
|
k2, v2 = d.popitem()
|
||
|
assert v * 1000 == v2
|
||
|
|
||
|
|
||
|
@has_emeter
|
||
|
def test_get_emeter_monthly(dev):
|
||
|
if dev.is_strip:
|
||
|
pytest.skip("Disabled for HS300 temporarily")
|
||
|
|
||
|
assert dev.has_emeter
|
||
|
|
||
|
assert dev.get_emeter_monthly(year=1900) == {}
|
||
|
|
||
|
d = dev.get_emeter_monthly()
|
||
|
assert len(d) > 0
|
||
|
|
||
|
k, v = d.popitem()
|
||
|
assert isinstance(k, int)
|
||
|
assert isinstance(v, float)
|
||
|
|
||
|
# Test kwh (energy, energy_wh)
|
||
|
d = dev.get_emeter_monthly(kwh=False)
|
||
|
k2, v2 = d.popitem()
|
||
|
assert v * 1000 == v2
|
||
|
|
||
|
|
||
|
@has_emeter
|
||
|
def test_emeter_status(dev):
|
||
|
if dev.is_strip:
|
||
|
pytest.skip("Disabled for HS300 temporarily")
|
||
|
|
||
|
assert dev.has_emeter
|
||
|
|
||
|
d = dev.get_emeter_realtime()
|
||
|
|
||
|
with pytest.raises(KeyError):
|
||
|
assert d["foo"]
|
||
|
|
||
|
assert d["power_mw"] == d["power"] * 1000
|
||
|
# bulbs have only power according to tplink simulator.
|
||
|
if not dev.is_bulb:
|
||
|
assert d["voltage_mv"] == d["voltage"] * 1000
|
||
|
|
||
|
assert d["current_ma"] == d["current"] * 1000
|
||
|
assert d["total_wh"] == d["total"] * 1000
|
||
|
|
||
|
|
||
|
@pytest.mark.skip("not clearing your stats..")
|
||
|
@has_emeter
|
||
|
def test_erase_emeter_stats(dev):
|
||
|
assert dev.has_emeter
|
||
|
|
||
|
dev.erase_emeter()
|
||
|
|
||
|
|
||
|
@has_emeter
|
||
|
def test_current_consumption(dev):
|
||
|
if dev.is_strip:
|
||
|
pytest.skip("Disabled for HS300 temporarily")
|
||
|
|
||
|
if dev.has_emeter:
|
||
|
x = dev.current_consumption()
|
||
|
assert isinstance(x, float)
|
||
|
assert x >= 0.0
|
||
|
else:
|
||
|
assert dev.current_consumption() is None
|
||
|
|
||
|
|
||
|
def test_alias(dev):
|
||
|
test_alias = "TEST1234"
|
||
|
original = dev.alias
|
||
|
assert isinstance(original, str)
|
||
|
|
||
|
dev.set_alias(test_alias)
|
||
|
assert dev.alias == test_alias
|
||
|
|
||
|
dev.set_alias(original)
|
||
|
assert dev.alias == original
|
||
|
|
||
|
|
||
|
@plug
|
||
|
def test_led(dev):
|
||
|
original = dev.led
|
||
|
|
||
|
dev.set_led(False)
|
||
|
assert not dev.led
|
||
|
dev.set_led(True)
|
||
|
|
||
|
assert dev.led
|
||
|
|
||
|
dev.set_led(original)
|
||
|
|
||
|
|
||
|
@plug
|
||
|
def test_on_since(dev):
|
||
|
assert isinstance(dev.on_since, datetime.datetime)
|
||
|
|
||
|
|
||
|
def test_icon(dev):
|
||
|
assert set(dev.icon.keys()), {"icon", "hash"}
|
||
|
|
||
|
|
||
|
def test_time(dev):
|
||
|
assert isinstance(dev.time, datetime.datetime)
|
||
|
# TODO check setting?
|
||
|
|
||
|
|
||
|
def test_timezone(dev):
|
||
|
TZ_SCHEMA(dev.timezone)
|
||
|
|
||
|
|
||
|
def test_hw_info(dev):
|
||
|
PLUG_SCHEMA(dev.hw_info)
|
||
|
|
||
|
|
||
|
def test_location(dev):
|
||
|
PLUG_SCHEMA(dev.location)
|
||
|
|
||
|
|
||
|
def test_rssi(dev):
|
||
|
PLUG_SCHEMA({"rssi": dev.rssi}) # wrapping for vol
|
||
|
|
||
|
|
||
|
def test_mac(dev):
|
||
|
PLUG_SCHEMA({"mac": dev.mac}) # wrapping for val
|
||
|
# TODO check setting?
|
||
|
|
||
|
|
||
|
@non_variable_temp
|
||
|
def test_temperature_on_nonsupporting(dev):
|
||
|
assert dev.valid_temperature_range == (0, 0)
|
||
|
|
||
|
# TODO test when device does not support temperature range
|
||
|
with pytest.raises(SmartDeviceException):
|
||
|
dev.set_color_temp(2700)
|
||
|
with pytest.raises(SmartDeviceException):
|
||
|
print(dev.color_temp)
|
||
|
|
||
|
|
||
|
@variable_temp
|
||
|
def test_out_of_range_temperature(dev):
|
||
|
with pytest.raises(ValueError):
|
||
|
dev.set_color_temp(1000)
|
||
|
with pytest.raises(ValueError):
|
||
|
dev.set_color_temp(10000)
|
||
|
|
||
|
|
||
|
@non_dimmable
|
||
|
def test_non_dimmable(dev):
|
||
|
assert not dev.is_dimmable
|
||
|
|
||
|
with pytest.raises(SmartDeviceException):
|
||
|
assert dev.brightness == 0
|
||
|
with pytest.raises(SmartDeviceException):
|
||
|
dev.set_brightness(100)
|
||
|
|
||
|
|
||
|
@dimmable
|
||
|
@turn_on
|
||
|
def test_dimmable_brightness(dev, turn_on):
|
||
|
handle_turn_on(dev, turn_on)
|
||
|
assert dev.is_dimmable
|
||
|
|
||
|
dev.set_brightness(50)
|
||
|
assert dev.brightness == 50
|
||
|
|
||
|
dev.set_brightness(10)
|
||
|
assert dev.brightness == 10
|
||
|
|
||
|
with pytest.raises(ValueError):
|
||
|
dev.set_brightness("foo")
|
||
|
|
||
|
|
||
|
@dimmable
|
||
|
def test_invalid_brightness(dev):
|
||
|
assert dev.is_dimmable
|
||
|
|
||
|
with pytest.raises(ValueError):
|
||
|
dev.set_brightness(110)
|
||
|
|
||
|
with pytest.raises(ValueError):
|
||
|
dev.set_brightness(-100)
|
||
|
|
||
|
|
||
|
@color_bulb
|
||
|
@turn_on
|
||
|
def test_hsv(dev, turn_on):
|
||
|
handle_turn_on(dev, turn_on)
|
||
|
assert dev.is_color
|
||
|
|
||
|
hue, saturation, brightness = dev.hsv
|
||
|
assert 0 <= hue <= 255
|
||
|
assert 0 <= saturation <= 100
|
||
|
assert 0 <= brightness <= 100
|
||
|
|
||
|
dev.set_hsv(hue=1, saturation=1, value=1)
|
||
|
|
||
|
hue, saturation, brightness = dev.hsv
|
||
|
assert hue == 1
|
||
|
assert saturation == 1
|
||
|
assert brightness == 1
|
||
|
|
||
|
|
||
|
@color_bulb
|
||
|
@turn_on
|
||
|
def test_invalid_hsv(dev, turn_on):
|
||
|
handle_turn_on(dev, turn_on)
|
||
|
|
||
|
assert dev.is_color
|
||
|
|
||
|
for invalid_hue in [-1, 361, 0.5]:
|
||
|
with pytest.raises(ValueError):
|
||
|
dev.set_hsv(invalid_hue, 0, 0)
|
||
|
|
||
|
for invalid_saturation in [-1, 101, 0.5]:
|
||
|
with pytest.raises(ValueError):
|
||
|
dev.set_hsv(0, invalid_saturation, 0)
|
||
|
|
||
|
for invalid_brightness in [-1, 101, 0.5]:
|
||
|
with pytest.raises(ValueError):
|
||
|
dev.set_hsv(0, 0, invalid_brightness)
|
||
|
|
||
|
|
||
|
@non_color_bulb
|
||
|
def test_hsv_on_non_color(dev):
|
||
|
assert not dev.is_color
|
||
|
|
||
|
with pytest.raises(SmartDeviceException):
|
||
|
dev.set_hsv(0, 0, 0)
|
||
|
with pytest.raises(SmartDeviceException):
|
||
|
print(dev.hsv)
|
||
|
|
||
|
|
||
|
@variable_temp
|
||
|
@turn_on
|
||
|
def test_try_set_colortemp(dev, turn_on):
|
||
|
handle_turn_on(dev, turn_on)
|
||
|
|
||
|
dev.set_color_temp(2700)
|
||
|
assert dev.color_temp == 2700
|
||
|
|
||
|
|
||
|
@variable_temp
|
||
|
@turn_on
|
||
|
def test_deprecated_colortemp(dev, turn_on):
|
||
|
handle_turn_on(dev, turn_on)
|
||
|
with pytest.deprecated_call():
|
||
|
dev.color_temp = 2700
|
||
|
|
||
|
|
||
|
@dimmable
|
||
|
def test_deprecated_brightness(dev):
|
||
|
with pytest.deprecated_call():
|
||
|
dev.brightness = 10
|
||
|
|
||
|
|
||
|
@non_variable_temp
|
||
|
def test_non_variable_temp(dev):
|
||
|
with pytest.raises(SmartDeviceException):
|
||
|
dev.set_color_temp(2700)
|
||
|
|
||
|
|
||
|
@color_bulb
|
||
|
@turn_on
|
||
|
def test_deprecated_hsv(dev, turn_on):
|
||
|
handle_turn_on(dev, turn_on)
|
||
|
with pytest.deprecated_call():
|
||
|
dev.hsv = (1, 1, 1)
|
||
|
|
||
|
|
||
|
@strip
|
||
|
def test_children_is_on(dev):
|
||
|
is_on = dev.get_is_on()
|
||
|
for i in range(dev.num_children):
|
||
|
assert is_on[i] == dev.get_is_on(index=i)
|
||
|
|
||
|
|
||
|
@strip
|
||
|
@turn_on
|
||
|
def test_children_change_state(dev, turn_on):
|
||
|
handle_turn_on(dev, turn_on)
|
||
|
for i in range(dev.num_children):
|
||
|
orig_state = dev.get_is_on(index=i)
|
||
|
if orig_state:
|
||
|
dev.turn_off(index=i)
|
||
|
assert not dev.get_is_on(index=i)
|
||
|
assert dev.get_is_off(index=i)
|
||
|
|
||
|
dev.turn_on(index=i)
|
||
|
assert dev.get_is_on(index=i)
|
||
|
assert not dev.get_is_off(index=i)
|
||
|
else:
|
||
|
dev.turn_on(index=i)
|
||
|
assert dev.get_is_on(index=i)
|
||
|
assert not dev.get_is_off(index=i)
|
||
|
dev.turn_off(index=i)
|
||
|
assert not dev.get_is_on(index=i)
|
||
|
assert dev.get_is_off(index=i)
|
||
|
|
||
|
|
||
|
@strip
|
||
|
def test_children_bounds(dev):
|
||
|
out_of_bounds = dev.num_children + 100
|
||
|
|
||
|
with pytest.raises(SmartDeviceException):
|
||
|
dev.turn_off(index=out_of_bounds)
|
||
|
with pytest.raises(SmartDeviceException):
|
||
|
dev.turn_on(index=out_of_bounds)
|
||
|
with pytest.raises(SmartDeviceException):
|
||
|
dev.get_is_on(index=out_of_bounds)
|
||
|
with pytest.raises(SmartDeviceException):
|
||
|
dev.get_alias(index=out_of_bounds)
|
||
|
with pytest.raises(SmartDeviceException):
|
||
|
dev.get_on_since(index=out_of_bounds)
|
||
|
|
||
|
|
||
|
@strip
|
||
|
def test_children_alias(dev):
|
||
|
original = dev.get_alias()
|
||
|
test_alias = "TEST1234"
|
||
|
for idx in range(dev.num_children):
|
||
|
dev.set_alias(alias=test_alias, index=idx)
|
||
|
assert dev.get_alias(index=idx) == test_alias
|
||
|
dev.set_alias(alias=original[idx], index=idx)
|
||
|
assert dev.get_alias(index=idx) == original[idx]
|
||
|
|
||
|
|
||
|
@strip
|
||
|
def test_children_on_since(dev):
|
||
|
for idx in range(dev.num_children):
|
||
|
assert dev.get_on_since(index=idx)
|
||
|
|
||
|
|
||
|
@pytest.mark.skip("this test will wear out your relays")
|
||
|
def test_all_binary_states(dev):
|
||
|
# test every binary state
|
||
|
for state in range(2 ** dev.num_children):
|
||
|
# create binary state map
|
||
|
state_map = {}
|
||
|
for plug_index in range(dev.num_children):
|
||
|
state_map[plug_index] = bool((state >> plug_index) & 1)
|
||
|
|
||
|
if state_map[plug_index]:
|
||
|
dev.turn_on(index=plug_index)
|
||
|
else:
|
||
|
dev.turn_off(index=plug_index)
|
||
|
|
||
|
# check state map applied
|
||
|
for index, state in dev.get_is_on().items():
|
||
|
assert state_map[index] == state
|
||
|
|
||
|
# toggle each outlet with state map applied
|
||
|
for plug_index in range(dev.num_children):
|
||
|
|
||
|
# toggle state
|
||
|
if state_map[plug_index]:
|
||
|
dev.turn_off(index=plug_index)
|
||
|
else:
|
||
|
dev.turn_on(index=plug_index)
|
||
|
|
||
|
# only target outlet should have state changed
|
||
|
for index, state in dev.get_is_on().items():
|
||
|
if index == plug_index:
|
||
|
assert state != state_map[index]
|
||
|
else:
|
||
|
assert state == state_map[index]
|
||
|
|
||
|
# reset state
|
||
|
if state_map[plug_index]:
|
||
|
dev.turn_on(index=plug_index)
|
||
|
else:
|
||
|
dev.turn_off(index=plug_index)
|
||
|
|
||
|
# original state map should be restored
|
||
|
for index, state in dev.get_is_on().items():
|
||
|
assert state == state_map[index]
|
||
|
|
||
|
|
||
|
@strip
|
||
|
def test_children_get_emeter_realtime(dev):
|
||
|
assert dev.has_emeter
|
||
|
# test with index
|
||
|
for plug_index in range(dev.num_children):
|
||
|
emeter = dev.get_emeter_realtime(index=plug_index)
|
||
|
CURRENT_CONSUMPTION_SCHEMA(emeter)
|
||
|
|
||
|
# test without index
|
||
|
for index, emeter in dev.get_emeter_realtime().items():
|
||
|
CURRENT_CONSUMPTION_SCHEMA(emeter)
|
||
|
|
||
|
# out of bounds
|
||
|
with pytest.raises(SmartStripException):
|
||
|
dev.get_emeter_realtime(index=dev.num_children + 100)
|
||
|
|
||
|
|
||
|
@strip
|
||
|
def test_children_get_emeter_daily(dev):
|
||
|
assert dev.has_emeter
|
||
|
# test with index
|
||
|
for plug_index in range(dev.num_children):
|
||
|
emeter = dev.get_emeter_daily(year=1900, month=1, index=plug_index)
|
||
|
assert emeter == {}
|
||
|
|
||
|
emeter = dev.get_emeter_daily(index=plug_index)
|
||
|
assert len(emeter) > 0
|
||
|
|
||
|
k, v = emeter.popitem()
|
||
|
assert isinstance(k, int)
|
||
|
assert isinstance(v, float)
|
||
|
|
||
|
# test without index
|
||
|
all_emeter = dev.get_emeter_daily(year=1900, month=1)
|
||
|
for plug_index, emeter in all_emeter.items():
|
||
|
assert emeter == {}
|
||
|
|
||
|
emeter = dev.get_emeter_daily(index=plug_index)
|
||
|
|
||
|
k, v = emeter.popitem()
|
||
|
assert isinstance(k, int)
|
||
|
assert isinstance(v, float)
|
||
|
|
||
|
# out of bounds
|
||
|
with pytest.raises(SmartStripException):
|
||
|
dev.get_emeter_daily(year=1900, month=1, index=dev.num_children + 100)
|
||
|
|
||
|
|
||
|
@strip
|
||
|
def test_children_get_emeter_monthly(dev):
|
||
|
assert dev.has_emeter
|
||
|
# test with index
|
||
|
for plug_index in range(dev.num_children):
|
||
|
emeter = dev.get_emeter_monthly(year=1900, index=plug_index)
|
||
|
assert emeter == {}
|
||
|
|
||
|
emeter = dev.get_emeter_monthly()
|
||
|
assert len(emeter) > 0
|
||
|
|
||
|
k, v = emeter.popitem()
|
||
|
assert isinstance(k, int)
|
||
|
assert isinstance(v, float)
|
||
|
|
||
|
# test without index
|
||
|
all_emeter = dev.get_emeter_monthly(year=1900)
|
||
|
for index, emeter in all_emeter.items():
|
||
|
assert emeter == {}
|
||
|
assert len(emeter) > 0
|
||
|
|
||
|
k, v = emeter.popitem()
|
||
|
assert isinstance(k, int)
|
||
|
assert isinstance(v, float)
|
||
|
|
||
|
# out of bounds
|
||
|
with pytest.raises(SmartStripException):
|
||
|
dev.get_emeter_monthly(year=1900, index=dev.num_children + 100)
|
||
|
|
||
|
|
||
|
def test_cache(dev):
|
||
|
from datetime import timedelta
|
||
|
|
||
|
dev.cache_ttl = timedelta(seconds=3)
|
||
|
with patch.object(
|
||
|
FakeTransportProtocol, "query", wraps=dev.protocol.query
|
||
|
) as query_mock:
|
||
|
CHECK_COUNT = 1
|
||
|
# Smartstrip calls sysinfo in its __init__ to request children, so
|
||
|
# the even first get call here will get its results from the cache.
|
||
|
if dev.is_strip:
|
||
|
CHECK_COUNT = 0
|
||
|
|
||
|
dev.get_sysinfo()
|
||
|
assert query_mock.call_count == CHECK_COUNT
|
||
|
dev.get_sysinfo()
|
||
|
assert query_mock.call_count == CHECK_COUNT
|
||
|
|
||
|
|
||
|
def test_cache_invalidates(dev):
|
||
|
from datetime import timedelta
|
||
|
|
||
|
dev.cache_ttl = timedelta(seconds=0)
|
||
|
|
||
|
with patch.object(
|
||
|
FakeTransportProtocol, "query", wraps=dev.protocol.query
|
||
|
) as query_mock:
|
||
|
dev.get_sysinfo()
|
||
|
assert query_mock.call_count == 1
|
||
|
dev.get_sysinfo()
|
||
|
assert query_mock.call_count == 2
|
||
|
# assert query_mock.called_once()
|