mirror of
https://github.com/python-kasa/python-kasa.git
synced 2024-12-23 19:53:34 +00:00
120 lines
3.7 KiB
Python
120 lines
3.7 KiB
Python
import pytest
|
|
from pytest_mock import MockerFixture
|
|
|
|
from kasa import KasaException, Module
|
|
from kasa.smart import SmartDevice
|
|
from kasa.smart.modules import Fan
|
|
|
|
from ...device_fixtures import get_parent_and_child_modules, parametrize
|
|
|
|
fan = parametrize("has fan", component_filter="fan_control", protocol_filter={"SMART"})
|
|
|
|
|
|
@fan
|
|
async def test_fan_speed(dev: SmartDevice, mocker: MockerFixture):
|
|
"""Test fan speed feature."""
|
|
fan = next(get_parent_and_child_modules(dev, Module.Fan))
|
|
assert fan
|
|
level_feature = fan._module_features["fan_speed_level"]
|
|
assert (
|
|
level_feature.minimum_value
|
|
<= level_feature.value
|
|
<= level_feature.maximum_value
|
|
)
|
|
|
|
call = mocker.spy(fan, "call")
|
|
await fan.set_fan_speed_level(3)
|
|
call.assert_called_with(
|
|
"set_device_info", {"device_on": True, "fan_speed_level": 3}
|
|
)
|
|
|
|
await dev.update()
|
|
|
|
assert fan.fan_speed_level == 3
|
|
assert level_feature.value == 3
|
|
|
|
|
|
@fan
|
|
async def test_sleep_mode(dev: SmartDevice, mocker: MockerFixture):
|
|
"""Test sleep mode feature."""
|
|
fan = next(get_parent_and_child_modules(dev, Module.Fan))
|
|
assert fan
|
|
sleep_feature = fan._module_features["fan_sleep_mode"]
|
|
assert isinstance(sleep_feature.value, bool)
|
|
|
|
call = mocker.spy(fan, "call")
|
|
await fan.set_sleep_mode(True)
|
|
call.assert_called_with("set_device_info", {"fan_sleep_mode_on": True})
|
|
|
|
await dev.update()
|
|
|
|
assert fan.sleep_mode is True
|
|
assert sleep_feature.value is True
|
|
|
|
|
|
@fan
|
|
async def test_fan_module(dev: SmartDevice, mocker: MockerFixture):
|
|
"""Test fan speed on device interface."""
|
|
assert isinstance(dev, SmartDevice)
|
|
fan = next(get_parent_and_child_modules(dev, Module.Fan))
|
|
assert fan
|
|
device = fan._device
|
|
|
|
await fan.set_fan_speed_level(1)
|
|
await dev.update()
|
|
assert fan.fan_speed_level == 1
|
|
assert device.is_on
|
|
|
|
# Check that if the device is off the speed level is 0.
|
|
await device.set_state(False)
|
|
await dev.update()
|
|
assert fan.fan_speed_level == 0
|
|
|
|
await fan.set_fan_speed_level(4)
|
|
await dev.update()
|
|
assert fan.fan_speed_level == 4
|
|
|
|
await fan.set_fan_speed_level(0)
|
|
await dev.update()
|
|
assert not device.is_on
|
|
|
|
fan_speed_level_feature = fan._module_features["fan_speed_level"]
|
|
max_level = fan_speed_level_feature.maximum_value
|
|
min_level = fan_speed_level_feature.minimum_value
|
|
with pytest.raises(ValueError, match="Invalid level"):
|
|
await fan.set_fan_speed_level(min_level - 1)
|
|
|
|
with pytest.raises(ValueError, match="Invalid level"):
|
|
await fan.set_fan_speed_level(max_level - 5)
|
|
|
|
|
|
@fan
|
|
async def test_fan_features(dev: SmartDevice, mocker: MockerFixture):
|
|
"""Test fan speed on device interface."""
|
|
assert isinstance(dev, SmartDevice)
|
|
fan = next(get_parent_and_child_modules(dev, Module.Fan))
|
|
assert fan
|
|
expected_feature = fan._module_features["fan_speed_level"]
|
|
|
|
fan_speed_level_feature = fan.get_feature(Fan.set_fan_speed_level)
|
|
assert expected_feature == fan_speed_level_feature
|
|
|
|
fan_speed_level_feature = fan.get_feature(fan.set_fan_speed_level)
|
|
assert expected_feature == fan_speed_level_feature
|
|
|
|
fan_speed_level_feature = fan.get_feature(Fan.fan_speed_level)
|
|
assert expected_feature == fan_speed_level_feature
|
|
|
|
fan_speed_level_feature = fan.get_feature("fan_speed_level")
|
|
assert expected_feature == fan_speed_level_feature
|
|
|
|
assert fan.has_feature(Fan.fan_speed_level)
|
|
|
|
msg = "Attribute _check_supported of module Fan is not bound to a feature"
|
|
with pytest.raises(KasaException, match=msg):
|
|
assert fan.has_feature(fan._check_supported)
|
|
|
|
msg = "No attribute named foobar in module Fan"
|
|
with pytest.raises(KasaException, match=msg):
|
|
assert fan.has_feature("foobar")
|