Add support for waterleak sensor (T300) (#876)

This commit is contained in:
Teemu R
2024-04-30 17:31:47 +02:00
committed by GitHub
parent 300d823895
commit 5599756d28
4 changed files with 107 additions and 0 deletions

View File

@@ -0,0 +1,42 @@
from enum import Enum
import pytest
from kasa.smart.modules import WaterleakSensor
from kasa.tests.device_fixtures import parametrize
waterleak = parametrize(
"has waterleak", component_filter="sensor_alarm", protocol_filter={"SMART.CHILD"}
)
@waterleak
@pytest.mark.parametrize(
"feature, type",
[
("alert", int),
("status", Enum),
],
)
async def test_waterleak_properties(dev, feature, type):
"""Test that features are registered and work as expected."""
waterleak: WaterleakSensor = dev.modules["WaterleakSensor"]
prop = getattr(waterleak, feature)
assert isinstance(prop, type)
feat = waterleak._module_features[feature]
assert feat.value == prop
assert isinstance(feat.value, type)
@waterleak
async def test_waterleak_features(dev):
"""Test waterleak features."""
waterleak: WaterleakSensor = dev.modules["WaterleakSensor"]
assert "water_leak" in dev.features
assert dev.features["water_leak"].value == waterleak.status
assert "water_alert" in dev.features
assert dev.features["water_alert"].value == waterleak.alert