mirror of
https://github.com/python-kasa/python-kasa.git
synced 2024-12-25 04:33:33 +00:00
46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
from datetime import datetime
|
|
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", "prop_name", "type"),
|
|
[
|
|
("water_alert", "alert", int),
|
|
# Can be converted to 'datetime | None' after py3.9 support is dropped
|
|
("water_alert_timestamp", "alert_timestamp", (datetime, type(None))),
|
|
("water_leak", "status", Enum),
|
|
],
|
|
)
|
|
async def test_waterleak_properties(dev, feature, prop_name, type):
|
|
"""Test that features are registered and work as expected."""
|
|
waterleak: WaterleakSensor = dev.modules["WaterleakSensor"]
|
|
|
|
prop = getattr(waterleak, prop_name)
|
|
assert isinstance(prop, type)
|
|
|
|
feat = dev.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
|