Test setting operating mode while cooling.

zigpy · Jan 29, 2025 · b8d6339 · b8d6339
1 parent 384764d
commit b8d6339
Showing 1 changed file with 24 additions and 4 deletions.
diff --git a/tests/test_bosch.py b/tests/test_bosch.py
@@ -2,15 +2,14 @@
 
 from unittest import mock
 
-from zigpy.zcl import foundation
-from zigpy.zcl.clusters.hvac import ControlSequenceOfOperation, Thermostat
-from zigpy.zcl.foundation import WriteAttributesStatusRecord
-
 import zhaquirks
 from zhaquirks.bosch.rbsh_trv0_zb_eu import (
     BoschOperatingMode,
     BoschThermostatCluster as BoschTrvThermostatCluster,
 )
+from zigpy.zcl import foundation
+from zigpy.zcl.clusters.hvac import ControlSequenceOfOperation, Thermostat
+from zigpy.zcl.foundation import WriteAttributesStatusRecord
 
 zhaquirks.setup()
 
@@ -347,6 +346,27 @@ def mock_read(attributes, manufacturer=None):
             == ControlSequenceOfOperation.Cooling_Only
         )
 
+        # -- operating_mode (by-id) in cooling mode
+        success, fail = await bosch_thermostat_cluster.write_attributes(
+            {
+                BoschTrvThermostatCluster.AttributeDefs.operating_mode.id: BoschOperatingMode.Manual,
+            }
+        )
+        assert success
+        assert not fail
+        assert (
+            bosch_thermostat_cluster._attr_cache[
+                BoschTrvThermostatCluster.AttributeDefs.operating_mode.id
+            ]
+            == BoschOperatingMode.Manual
+        )
+        assert (
+            bosch_thermostat_cluster._attr_cache[
+                Thermostat.AttributeDefs.system_mode.id
+            ]
+            == Thermostat.SystemMode.Cool
+        )
+
         # -- operating_mode (by-id) gets ignored when system_mode is written
         success, fail = await bosch_thermostat_cluster.write_attributes(
             {