From c0f5d5b5941a19e839ab61f57bf46acd8a182ab7 Mon Sep 17 00:00:00 2001
From: Teemu Rytilahti <tpr@iki.fi>
Date: Thu, 5 Dec 2024 17:52:48 +0100
Subject: [PATCH] Add tests

---
 tests/smart/modules/test_powerprotection.py | 76 +++++++++++++++++++++
 1 file changed, 76 insertions(+)
 create mode 100644 tests/smart/modules/test_powerprotection.py

diff --git a/tests/smart/modules/test_powerprotection.py b/tests/smart/modules/test_powerprotection.py
new file mode 100644
index 00000000..8db89721
--- /dev/null
+++ b/tests/smart/modules/test_powerprotection.py
@@ -0,0 +1,76 @@
+import pytest
+from pytest_mock import MockerFixture
+
+from kasa import Module, SmartDevice
+from kasa.smart.modules import PowerProtection
+
+from ...device_fixtures import parametrize
+
+powerprotection = parametrize(
+    "has powerprotection",
+    component_filter="power_protection",
+    protocol_filter={"SMART"},
+)
+
+
+def _skip_on_unavailable(dev: SmartDevice):
+    if Module.PowerProtection not in dev.modules:
+        pytest.skip(f"No powerprotection module on {dev}, maybe a strip parent?")
+
+
+@powerprotection
+@pytest.mark.parametrize(
+    ("feature", "prop_name", "type"),
+    [
+        ("overloaded", "overloaded", bool | None),
+        ("power_protection_enabled", "enabled", bool),
+        ("power_protection_threshold", "protection_threshold", int),
+    ],
+)
+async def test_features(dev, feature, prop_name, type):
+    """Test that features are registered and work as expected."""
+    _skip_on_unavailable(dev)
+
+    powerprot: PowerProtection = dev.modules[Module.PowerProtection]
+
+    prop = getattr(powerprot, prop_name)
+    assert isinstance(prop, type)
+
+    feat = dev.features[feature]
+    assert feat.value == prop
+    assert isinstance(feat.value, type)
+
+
+@powerprotection
+async def test_set_enable(dev: SmartDevice, mocker: MockerFixture):
+    """Test enable."""
+    _skip_on_unavailable(dev)
+
+    powerprot: PowerProtection = dev.modules[Module.PowerProtection]
+
+    call_spy = mocker.spy(powerprot, "call")
+    await powerprot.set_enabled(True)
+    params = {
+        "enabled": mocker.ANY,
+        "protection_power": mocker.ANY,
+    }
+    call_spy.assert_called_with("set_protection_power", params)
+
+
+@powerprotection
+async def test_set_threshold(dev: SmartDevice, mocker: MockerFixture):
+    """Test enable."""
+    _skip_on_unavailable(dev)
+
+    powerprot: PowerProtection = dev.modules[Module.PowerProtection]
+
+    call_spy = mocker.spy(powerprot, "call")
+    await powerprot.set_protection_threshold(123)
+    params = {
+        "enabled": mocker.ANY,
+        "protection_power": 123,
+    }
+    call_spy.assert_called_with("set_protection_power", params)
+
+    with pytest.raises(ValueError, match="Threshold out of range"):
+        await powerprot.set_protection_threshold(-10)