feat(device): change some things from AUA1ZBR2GW controller

xaviml · Mar 6, 2021 · 0f85d67 · 0f85d67
1 parent d5b45c4
commit 0f85d67
Show file tree

Hide file tree

Showing 4 changed files with 24 additions and 4 deletions.
diff --git a/apps/controllerx/cx_devices/aurora.py b/apps/controllerx/cx_devices/aurora.py
@@ -26,11 +26,11 @@ def get_zha_actions_mapping(self) -> DefaultActionsMapping:
     def get_zha_action(self, data: EventData) -> str:
         endpoint_id = data.get("endpoint_id", 1)
         command = action = data["command"]
-        args = data.get("args", {})
-        args_mapping = {0: "up", 1: "down", 3: "up"}
+        args = data.get("args", [])
         if command == "step" or command == "step_color_temp":
+            args_mapping = {0: "up", 1: "down", 3: "up"}
             action = "_".join((action, args_mapping[args[0]]))
         if command == "on" or command == "off":
             action = "toggle"
-        action = "_".join((str(endpoint_id), action))
+        action = f"{endpoint_id}_{action}"
         return action
diff --git a/docs/_data/controllers/AU-A1ZBR2GW.yaml → docs/_data/controllers/AU-A1ZBR2GW.yml b/docs/_data/controllers/AU-A1ZBR2GW.yaml → docs/_data/controllers/AU-A1ZBR2GW.yml
@@ -3,7 +3,7 @@ device_support:
   - type: Light
     domain: light
     controller: AUA1ZBR2GWLightController
-    delay: 100
+    delay: 350
     mapping:
       - "Click left knob → Toggle 2"
       - "Rotate left knob clockwise → Increase brightness 2"

diff --git a/docs/assets/img/AU-A1ZBR2GW.jpeg b/docs/assets/img/AU-A1ZBR2GW.jpeg
diff --git a/tests/unit_tests/cx_devices/aurora_test.py b/tests/unit_tests/cx_devices/aurora_test.py
@@ -0,0 +1,20 @@
+import pytest
+from cx_core.integration import EventData
+from cx_devices.aurora import AUA1ZBR2GWLightController
+
+
+@pytest.mark.parametrize(
+    "data, expected_action",
+    [
+        ({"command": "on"}, "1_toggle"),
+        ({"command": "on", "endpoint_id": 2}, "2_toggle"),
+        ({"command": "off"}, "1_toggle"),
+        ({"command": "step", "args": [0]}, "1_step_up"),
+        ({"command": "step", "args": [3]}, "1_step_up"),
+        ({"command": "step_color_temp", "args": [1]}, "1_step_color_temp_down"),
+    ],
+)
+def test_zha_action_WXKG01LMLightController(data: EventData, expected_action: str):
+    sut = AUA1ZBR2GWLightController()  # type: ignore
+    action = sut.get_zha_action(data)
+    assert action == expected_action