Fix tests after #1373

AlexxIT · Apr 16, 2024 · 4f02ea2 · 4f02ea2
1 parent fa07f4c
commit 4f02ea2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/custom_components/sonoff/sensor.py b/custom_components/sonoff/sensor.py
@@ -334,6 +334,7 @@ class XT5Action(XEventSesor):
     uid = "action"
 
     def set_state(self, params: dict):
+        # https://github.com/AlexxIT/SonoffLAN/issues/1373
         if "switches" in params and params.get("triggerType") == 2:
             self._attr_native_value = "touch"
             asyncio.create_task(self.clear_state())

diff --git a/tests/test_entity.py b/tests/test_entity.py
@@ -1641,7 +1641,9 @@ def test_t5():
     action: XT5Action = entities[4]
     assert action.state == ""
 
-    action.internal_update({"triggerType": 2})
+    action.internal_update(
+        {"switches": [{"switch": "on", "outlet": 0}], "triggerType": 2}
+    )
     assert action.state == "touch"
 
     action.internal_update({"slide": 2})