fix(device): ZHA intgegration for WXKG11LMLightController

xaviml · Oct 28, 2020 · 8bf2a41 · 8bf2a41
1 parent 5a71002
commit 8bf2a41
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 2 deletions.
diff --git a/apps/controllerx/cx_devices/aqara.py b/apps/controllerx/cx_devices/aqara.py
@@ -90,7 +90,7 @@ def get_zha_action(self, data: dict) -> str:
             4: "quadruple",
         }
         clicks = data["args"]["value"]
-        return mapping[clicks]
+        return mapping.get(clicks, "")
 
 
 class WXKG12LMLightController(LightController):

diff --git a/tests/unit_tests/cx_devices/aqara_test.py b/tests/unit_tests/cx_devices/aqara_test.py
@@ -1,5 +1,9 @@
 import pytest
-from cx_devices.aqara import MFKZQ01LMLightController, WXKG01LMLightController
+from cx_devices.aqara import (
+    MFKZQ01LMLightController,
+    WXKG01LMLightController,
+    WXKG11LMLightController,
+)
 
 
 @pytest.mark.parametrize(
@@ -34,3 +38,19 @@ def test_zha_action_WXKG01LMLightController(data, expected_action):
     sut = WXKG01LMLightController()
     action = sut.get_zha_action(data)
     assert action == expected_action
+
+
+@pytest.mark.parametrize(
+    "data, expected_action",
+    [
+        ({"args": {"value": 0}}, ""),
+        ({"args": {"value": 1}}, "single"),
+        ({"args": {"value": 2}}, "double"),
+        ({"args": {"value": 3}}, "triple"),
+        ({"args": {"value": 4}}, "quadruple"),
+    ],
+)
+def test_zha_action_WXKG11LMLightController(data, expected_action):
+    sut = WXKG11LMLightController()
+    action = sut.get_zha_action(data)
+    assert action == expected_action