Add switchbot_cloud climate TURN_OFF, TURN_ON support. (home-assistant#154017)

Author: danada

homeassistant/components/switchbot_cloud/climate.py

@@ -66,7 +66,10 @@ class SwitchBotCloudAirConditioner(SwitchBotCloudEntity, ClimateEntity, RestoreE
 
     _attr_assumed_state = True
     _attr_supported_features = (
-        ClimateEntityFeature.FAN_MODE | ClimateEntityFeature.TARGET_TEMPERATURE
+        ClimateEntityFeature.FAN_MODE
+        | ClimateEntityFeature.TARGET_TEMPERATURE
+        | ClimateEntityFeature.TURN_OFF
+        | ClimateEntityFeature.TURN_ON
     )
     _attr_fan_modes = [
         FanState.FAN_AUTO,
@@ -161,3 +164,17 @@ async def async_set_temperature(self, **kwargs: Any) -> None:
         await self._do_send_command(temperature=temperature)
         self._attr_target_temperature = temperature
         self.async_write_ha_state()
+
+    async def async_turn_off(self) -> None:
+        """Turn climate entity off."""
+        await self.async_set_hvac_mode(HVACMode.OFF)
+
+    async def async_turn_on(self) -> None:
+        """Turn climate entity on.
+
+        Uses the last known hvac_mode (if not OFF), otherwise defaults to FAN_ONLY.
+        """
+        hvac_mode = self._attr_hvac_mode
+        if hvac_mode == HVACMode.OFF:
+            hvac_mode = HVACMode.FAN_ONLY
+        await self.async_set_hvac_mode(hvac_mode)

tests/components/switchbot_cloud/test_climate.py

@@ -12,6 +12,8 @@
     SERVICE_SET_FAN_MODE,
     SERVICE_SET_HVAC_MODE,
     SERVICE_SET_TEMPERATURE,
+    SERVICE_TURN_OFF,
+    SERVICE_TURN_ON,
 )
 from homeassistant.components.switchbot_cloud import SwitchBotAPI
 from homeassistant.config_entries import ConfigEntryState
@@ -180,3 +182,117 @@ async def test_air_conditioner_no_last_state(
     assert state.state == "fan_only"
     assert state.attributes[ATTR_FAN_MODE] == "auto"
     assert state.attributes[ATTR_TEMPERATURE] == 21
+
+
+async def test_air_conditioner_turn_off(
+    hass: HomeAssistant, mock_list_devices, mock_get_status
+) -> None:
+    """Test the climate.turn_off service."""
+    mock_list_devices.return_value = [
+        Remote(
+            deviceId="ac-device-id-1",
+            deviceName="climate-1",
+            remoteType="DIY Air Conditioner",
+            hubDeviceId="test-hub-id",
+        ),
+    ]
+
+    entry = await configure_integration(hass)
+    assert entry.state is ConfigEntryState.LOADED
+
+    entity_id = "climate.climate_1"
+    assert hass.states.get(entity_id).state == "fan_only"
+
+    with patch.object(SwitchBotAPI, "send_command") as mock_send_command:
+        await hass.services.async_call(
+            CLIMATE_DOMAIN,
+            SERVICE_TURN_OFF,
+            {ATTR_ENTITY_ID: entity_id},
+            blocking=True,
+        )
+        mock_send_command.assert_called_once()
+        assert "21,4,1,off" in str(mock_send_command.call_args)
+
+    assert hass.states.get(entity_id).state == "off"
+
+
+async def test_air_conditioner_turn_on(
+    hass: HomeAssistant, mock_list_devices, mock_get_status
+) -> None:
+    """Test turning on a climate entity that has a non-off HVAC_STATE."""
+    mock_list_devices.return_value = [
+        Remote(
+            deviceId="ac-device-id-1",
+            deviceName="climate-1",
+            remoteType="DIY Air Conditioner",
+            hubDeviceId="test-hub-id",
+        ),
+    ]
+
+    mock_state = State(
+        "climate.climate_1",
+        "cool",
+        {
+            ATTR_FAN_MODE: "high",
+            ATTR_TEMPERATURE: 25,
+        },
+    )
+    mock_restore_cache(hass, (mock_state,))
+    entry = await configure_integration(hass)
+    assert entry.state is ConfigEntryState.LOADED
+
+    entity_id = "climate.climate_1"
+    assert hass.states.get(entity_id).state == "cool"
+
+    with patch.object(SwitchBotAPI, "send_command") as mock_turn_on_command:
+        await hass.services.async_call(
+            CLIMATE_DOMAIN,
+            SERVICE_TURN_ON,
+            {ATTR_ENTITY_ID: entity_id},
+            blocking=True,
+        )
+        mock_turn_on_command.assert_called_once()
+        assert "25,2,4,on" in str(mock_turn_on_command.call_args)
+
+    assert hass.states.get(entity_id).state == "cool"
+
+
+async def test_air_conditioner_turn_on_from_hvac_mode_off(
+    hass: HomeAssistant, mock_list_devices, mock_get_status
+) -> None:
+    """Test turning on a climate entity that has an off HVAC_STATE."""
+    mock_list_devices.return_value = [
+        Remote(
+            deviceId="ac-device-id-1",
+            deviceName="climate-1",
+            remoteType="DIY Air Conditioner",
+            hubDeviceId="test-hub-id",
+        ),
+    ]
+
+    mock_state = State(
+        "climate.climate_1",
+        "off",
+        {
+            ATTR_FAN_MODE: "high",
+            ATTR_TEMPERATURE: 25,
+        },
+    )
+    mock_restore_cache(hass, (mock_state,))
+    entry = await configure_integration(hass)
+    assert entry.state is ConfigEntryState.LOADED
+
+    entity_id = "climate.climate_1"
+    assert hass.states.get(entity_id).state == "off"
+
+    with patch.object(SwitchBotAPI, "send_command") as mock_turn_on_command:
+        await hass.services.async_call(
+            CLIMATE_DOMAIN,
+            SERVICE_TURN_ON,
+            {ATTR_ENTITY_ID: entity_id},
+            blocking=True,
+        )
+        mock_turn_on_command.assert_called_once()
+        assert "25,4,4,on" in str(mock_turn_on_command.call_args)
+
+    assert hass.states.get(entity_id).state == "fan_only"