Mirrored_Repos/python-kasa
1
0
Fork 0
mirror of https://github.com/python-kasa/python-kasa.git synced 2025-08-06 10:44:04 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Allow getting Annotated features from modules (#1018)

Browse Source 
Co-authored-by: Teemu R. <tpr@iki.fi>
This commit is contained in:
Steven B.
2024-11-22 07:52:23 +00:00
committed by GitHub
parent cae9decb02
commit 37cc4da7b6
3 changed files with 119 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
tests/smart/modules/test_fan.py
View File

@@ -1,8 +1,9 @@
import pytest
from pytest_mock import MockerFixture
from kasa import Module
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
@@ -77,8 +78,42 @@ async def test_fan_module(dev: SmartDevice, mocker: MockerFixture):
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(-1)
await fan.set_fan_speed_level(min_level - 1)
with pytest.raises(ValueError, match="Invalid level"):
await fan.set_fan_speed_level(5)
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")