Add Device.supports_module() and stop querying HomeKit for iot devices (#1808)

Some legacy iot firmware stops responding when an unsupported module follows `system` in a request, which made every update time out once the HomeKit query was added.

This stops querying HomeKit in `update()` for IOT devices and adds `Device.supports_module()` API to allow checking if some module is supported by the device.
This commit is contained in:
Teemu R.
2026-10-24 16:22:33 +02:00
committed by GitHub
parent e29864bfdc
commit 30f5e4a1b9
6 changed files with 130 additions and 18 deletions

View File

@@ -8,13 +8,23 @@ from kasa.iot.modules.homekit import HomeKit
from ...device_fixtures import device_iot
HOMEKIT = "smartlife.iot.homekit"
async def _homekit_module(dev: IotDevice) -> HomeKit:
"""Create the module with the data from the device, skip if not supported."""
if not await dev.supports_module(Module.IotHomeKit):
pytest.skip("HomeKit module not present on this device")
homekit = HomeKit(dev, HOMEKIT)
dev._last_update.update(await dev.protocol.query(homekit.query()))
homekit._initialize_features()
return homekit
@device_iot
def test_homekit_getters(dev: IotDevice) -> None:
async def test_homekit_getters(dev: IotDevice) -> None:
# HomeKit can be present on any IOT device
if Module.IotHomeKit not in dev.modules:
pytest.skip("HomeKit module not present on this device")
homekit: HomeKit = dev.modules[Module.IotHomeKit]
homekit = await _homekit_module(dev)
info = homekit.info
if not info:
pytest.skip("No HomeKit data present for this fixture")
@@ -31,10 +41,8 @@ def test_homekit_getters(dev: IotDevice) -> None:
@device_iot
def test_homekit_feature(dev: IotDevice) -> None:
if Module.IotHomeKit not in dev.modules:
pytest.skip("HomeKit module not present on this device")
homekit: HomeKit = dev.modules[Module.IotHomeKit]
async def test_homekit_feature(dev: IotDevice) -> None:
homekit = await _homekit_module(dev)
if not homekit.info:
pytest.skip("No HomeKit data present for this device")
feature = homekit._all_features.get("homekit_setup_code")
@@ -46,9 +54,7 @@ def test_homekit_feature(dev: IotDevice) -> None:
@device_iot
def test_initialize_features_skips_when_no_data(dev: IotDevice) -> None:
if Module.IotHomeKit not in dev.modules:
pytest.skip("HomeKit module not present on this device")
homekit: HomeKit = dev.modules[Module.IotHomeKit]
homekit = HomeKit(dev, HOMEKIT)
if "homekit_setup_code" in homekit._all_features:
pytest.skip("HomeKit feature already present on this device")
# Patch .data so it looks like no homekit data is present
@@ -57,3 +63,4 @@ def test_initialize_features_skips_when_no_data(dev: IotDevice) -> None:
homekit._initialize_features()
# Since there was no data, no features should be added
assert "homekit_setup_code" not in homekit._all_features
assert homekit.info == {}