fix(device): ZHA parser for WXKG01LMLightController

related to #241
xaviml · Feb 7, 2021 · 945efbd · 945efbd
1 parent 55611fb
commit 945efbd
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/apps/controllerx/cx_devices/aqara.py b/apps/controllerx/cx_devices/aqara.py
@@ -111,7 +111,10 @@ def get_zha_actions_mapping(self) -> DefaultActionsMapping:
         }
 
     def get_zha_action(self, data: EventData) -> str:
-        return data["args"]["click_type"]
+        args = data["args"]
+        if "click_type" in args:
+            return args["click_type"]
+        return data["command"]
 
 
 class WXKG11LMRemoteLightController(LightController):

diff --git a/tests/unit_tests/cx_devices/aqara_test.py b/tests/unit_tests/cx_devices/aqara_test.py
@@ -34,6 +34,10 @@ def test_zha_action_MFKZQ01LMLightController(data: EventData, expected_action: s
         ({"command": "click", "args": {"click_type": "triple"}}, "triple"),
         ({"command": "click", "args": {"click_type": "quadruple"}}, "quadruple"),
         ({"command": "click", "args": {"click_type": "furious"}}, "furious"),
+        (
+            {"command": "attribute_updated", "args": {"value": True}},
+            "attribute_updated",
+        ),
     ],
 )
 def test_zha_action_WXKG01LMLightController(data: EventData, expected_action: str):