mirror of
https://github.com/python-kasa/python-kasa.git
synced 2026-07-08 14:52:03 +00:00
88e1c27bd0
Some checks are pending
CI / Perform Lint Checks (3.14) (push) Waiting to run
CI / Python 3.11 on macos-latest (push) Blocked by required conditions
CI / Python 3.12 on macos-latest (push) Blocked by required conditions
CI / Python 3.13 on macos-latest (push) Blocked by required conditions
CI / Python 3.14 on macos-latest (push) Blocked by required conditions
CI / Python 3.11 on ubuntu-latest (push) Blocked by required conditions
CI / Python 3.12 on ubuntu-latest (push) Blocked by required conditions
CI / Python 3.13 on ubuntu-latest (push) Blocked by required conditions
CI / Python 3.14 on ubuntu-latest (push) Blocked by required conditions
CI / Python 3.11 on windows-latest (push) Blocked by required conditions
CI / Python 3.12 on windows-latest (push) Blocked by required conditions
CI / Python 3.13 on windows-latest (push) Blocked by required conditions
CI / Python 3.14 on windows-latest (push) Blocked by required conditions
CodeQL Checks / Analyze (python) (push) Waiting to run
Add type annotations across all 'smart' tests, enabling mypy to check test function bodies, catching type errors that were previously hidden.
120 lines
3.8 KiB
Python
120 lines
3.8 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) -> None:
|
|
"""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) -> None:
|
|
"""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) -> None:
|
|
"""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) -> None:
|
|
"""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")
|