Clean up, add test

zigpy · Feb 16, 2025 · 24d39e9 · 24d39e9
1 parent ab3a9e7
commit 24d39e9
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/tests/test_tuya_trv.py b/tests/test_tuya_trv.py
@@ -87,6 +87,14 @@
             zha.DeviceType.THERMOSTAT,  # quirk replaces device type with THERMOSTAT
             True,  # Enusure schedule is turned off
         ),
+        (
+            "_TZE200_ne4pikwm",
+            "TS0601",
+            TUYA_TEST_PLAN_V02,
+            TUYA_SP_V02,
+            TUYA_SYS_MODE_V02,
+            None,  # test device has specific device type, real one has SMART_PLUG
+        ),
     ),
 )
 async def test_handle_get_data(

diff --git a/zhaquirks/tuya/tuya_trv.py b/zhaquirks/tuya/tuya_trv.py
@@ -431,7 +431,7 @@ async def write_attributes(
         dp_id=3,
         ep_attribute=TuyaThermostatV2.ep_attribute,
         attribute_name=TuyaThermostatV2.AttributeDefs.running_state.name,
-        converter=lambda x: 0x01 if x else 0x00,  # Heat, Idle
+        converter=lambda x: RunningState.Heat_State_On if not x else RunningState.Idle,
     )
     .tuya_switch(
         dp_id=8,