mirror of
https://github.com/home-assistant/core.git
synced 2026-08-29 10:21:09 -05:00
f29e4a6465
Co-authored-by: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com> Co-authored-by: balloob <1444314+balloob@users.noreply.github.com>
603 lines
18 KiB
Python
603 lines
18 KiB
Python
"""Test for the switchbot_cloud climate."""
|
|
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
import pytest
|
|
from switchbot_api import Device, Remote, SmartRadiatorThermostatCommands, SwitchBotAPI
|
|
|
|
from homeassistant.components.climate import (
|
|
ATTR_FAN_MODE,
|
|
ATTR_HVAC_MODE,
|
|
ATTR_TEMPERATURE,
|
|
DOMAIN as CLIMATE_DOMAIN,
|
|
SERVICE_SET_FAN_MODE,
|
|
SERVICE_SET_HVAC_MODE,
|
|
SERVICE_SET_PRESET_MODE,
|
|
SERVICE_SET_TEMPERATURE,
|
|
SERVICE_TURN_OFF,
|
|
SERVICE_TURN_ON,
|
|
HVACMode,
|
|
)
|
|
from homeassistant.config_entries import ConfigEntryState
|
|
from homeassistant.const import ATTR_ENTITY_ID
|
|
from homeassistant.core import HomeAssistant, State
|
|
from homeassistant.exceptions import ServiceValidationError
|
|
from homeassistant.util.unit_system import US_CUSTOMARY_SYSTEM
|
|
|
|
from . import configure_integration
|
|
|
|
from tests.common import mock_restore_cache
|
|
|
|
|
|
async def test_air_conditioner_set_hvac_mode(
|
|
hass: HomeAssistant, mock_list_devices, mock_get_status
|
|
) -> None:
|
|
"""Test setting HVAC mode for air conditioner."""
|
|
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"
|
|
|
|
with patch.object(SwitchBotAPI, "send_command") as mock_send_command:
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN,
|
|
SERVICE_SET_HVAC_MODE,
|
|
{ATTR_ENTITY_ID: entity_id, ATTR_HVAC_MODE: "cool"},
|
|
blocking=True,
|
|
)
|
|
mock_send_command.assert_called_once()
|
|
assert "21,2,1,on" in str(mock_send_command.call_args)
|
|
|
|
assert hass.states.get(entity_id).state == "cool"
|
|
|
|
# Test turning off
|
|
with patch.object(SwitchBotAPI, "send_command") as mock_send_command:
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN,
|
|
SERVICE_SET_HVAC_MODE,
|
|
{ATTR_ENTITY_ID: entity_id, ATTR_HVAC_MODE: "off"},
|
|
blocking=True,
|
|
)
|
|
mock_send_command.assert_called_once()
|
|
assert "21,2,1,off" in str(mock_send_command.call_args)
|
|
|
|
assert hass.states.get(entity_id).state == "off"
|
|
|
|
|
|
async def test_air_conditioner_set_fan_mode(
|
|
hass: HomeAssistant, mock_list_devices, mock_get_status
|
|
) -> None:
|
|
"""Test setting fan mode for air conditioner."""
|
|
mock_list_devices.return_value = [
|
|
Remote(
|
|
deviceId="ac-device-id-1",
|
|
deviceName="climate-1",
|
|
remoteType="Air Conditioner",
|
|
hubDeviceId="test-hub-id",
|
|
),
|
|
]
|
|
|
|
entry = await configure_integration(hass)
|
|
assert entry.state is ConfigEntryState.LOADED
|
|
entity_id = "climate.climate_1"
|
|
|
|
with patch.object(SwitchBotAPI, "send_command") as mock_send_command:
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN,
|
|
SERVICE_SET_FAN_MODE,
|
|
{ATTR_ENTITY_ID: entity_id, ATTR_FAN_MODE: "high"},
|
|
blocking=True,
|
|
)
|
|
mock_send_command.assert_called_once()
|
|
assert "21,4,4,on" in str(mock_send_command.call_args)
|
|
|
|
assert hass.states.get(entity_id).attributes[ATTR_FAN_MODE] == "high"
|
|
|
|
|
|
async def test_air_conditioner_set_temperature(
|
|
hass: HomeAssistant, mock_list_devices, mock_get_status
|
|
) -> None:
|
|
"""Test setting temperature for air conditioner."""
|
|
mock_list_devices.return_value = [
|
|
Remote(
|
|
deviceId="ac-device-id-1",
|
|
deviceName="climate-1",
|
|
remoteType="Air Conditioner",
|
|
hubDeviceId="test-hub-id",
|
|
),
|
|
]
|
|
|
|
entry = await configure_integration(hass)
|
|
assert entry.state is ConfigEntryState.LOADED
|
|
entity_id = "climate.climate_1"
|
|
|
|
with patch.object(SwitchBotAPI, "send_command") as mock_send_command:
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN,
|
|
SERVICE_SET_TEMPERATURE,
|
|
{ATTR_ENTITY_ID: entity_id, ATTR_TEMPERATURE: 25},
|
|
blocking=True,
|
|
)
|
|
mock_send_command.assert_called_once()
|
|
assert "25,4,1,on" in str(mock_send_command.call_args)
|
|
|
|
assert hass.states.get(entity_id).attributes[ATTR_TEMPERATURE] == 25
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("service_data", "expected_command", "expected_state"),
|
|
[
|
|
pytest.param(
|
|
{ATTR_TEMPERATURE: 27, ATTR_HVAC_MODE: HVACMode.DRY},
|
|
"27,3,1,on",
|
|
HVACMode.DRY,
|
|
id="with_hvac_mode",
|
|
),
|
|
pytest.param(
|
|
{ATTR_TEMPERATURE: 27},
|
|
"27,4,1,off",
|
|
HVACMode.OFF,
|
|
id="without_hvac_mode",
|
|
),
|
|
],
|
|
)
|
|
async def test_air_conditioner_set_temperature_while_off(
|
|
hass: HomeAssistant,
|
|
mock_list_devices: AsyncMock,
|
|
mock_get_status: AsyncMock,
|
|
service_data: dict[str, float | HVACMode],
|
|
expected_command: str,
|
|
expected_state: HVACMode,
|
|
) -> None:
|
|
"""Test setting the temperature of an air conditioner that is off."""
|
|
mock_list_devices.return_value = [
|
|
Remote(
|
|
deviceId="ac-device-id-1",
|
|
deviceName="climate-1",
|
|
remoteType="Air Conditioner",
|
|
hubDeviceId="test-hub-id",
|
|
),
|
|
]
|
|
|
|
entity_id = "climate.climate_1"
|
|
mock_restore_cache(hass, (State(entity_id, HVACMode.OFF),))
|
|
|
|
entry = await configure_integration(hass)
|
|
assert entry.state is ConfigEntryState.LOADED
|
|
assert hass.states.get(entity_id).state == HVACMode.OFF
|
|
|
|
with patch.object(SwitchBotAPI, "send_command") as mock_send_command:
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN,
|
|
SERVICE_SET_TEMPERATURE,
|
|
{ATTR_ENTITY_ID: entity_id, **service_data},
|
|
blocking=True,
|
|
)
|
|
mock_send_command.assert_called_once()
|
|
assert expected_command in str(mock_send_command.call_args)
|
|
|
|
state = hass.states.get(entity_id)
|
|
assert state.state == expected_state
|
|
assert state.attributes[ATTR_TEMPERATURE] == 27
|
|
|
|
|
|
async def test_air_conditioner_set_temperature_unsupported_hvac_mode(
|
|
hass: HomeAssistant,
|
|
mock_list_devices: AsyncMock,
|
|
mock_get_status: AsyncMock,
|
|
) -> None:
|
|
"""Test setting the temperature with an hvac mode the air conditioner lacks."""
|
|
mock_list_devices.return_value = [
|
|
Remote(
|
|
deviceId="ac-device-id-1",
|
|
deviceName="climate-1",
|
|
remoteType="Air Conditioner",
|
|
hubDeviceId="test-hub-id",
|
|
),
|
|
]
|
|
|
|
entry = await configure_integration(hass)
|
|
assert entry.state is ConfigEntryState.LOADED
|
|
entity_id = "climate.climate_1"
|
|
|
|
with (
|
|
patch.object(SwitchBotAPI, "send_command") as mock_send_command,
|
|
pytest.raises(ServiceValidationError),
|
|
):
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN,
|
|
SERVICE_SET_TEMPERATURE,
|
|
{
|
|
ATTR_ENTITY_ID: entity_id,
|
|
ATTR_TEMPERATURE: 27,
|
|
ATTR_HVAC_MODE: HVACMode.AUTO,
|
|
},
|
|
blocking=True,
|
|
)
|
|
|
|
mock_send_command.assert_not_called()
|
|
assert hass.states.get(entity_id).state == HVACMode.FAN_ONLY
|
|
|
|
|
|
async def test_air_conditioner_restore_state(
|
|
hass: HomeAssistant, mock_list_devices, mock_get_status
|
|
) -> None:
|
|
"""Test restoring state for air conditioner."""
|
|
mock_list_devices.return_value = [
|
|
Remote(
|
|
deviceId="ac-device-id-1",
|
|
deviceName="climate-1",
|
|
remoteType="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"
|
|
state = hass.states.get(entity_id)
|
|
assert state.state == "cool"
|
|
assert state.attributes[ATTR_FAN_MODE] == "high"
|
|
assert state.attributes[ATTR_TEMPERATURE] == 25
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("restored_temperature", "expected_temperature"),
|
|
[
|
|
pytest.param(75, 75, id="restored_as_published"),
|
|
pytest.param(19381607032619749376, 70, id="corrupted_falls_back_to_default"),
|
|
],
|
|
)
|
|
async def test_air_conditioner_restore_state_in_fahrenheit(
|
|
hass: HomeAssistant,
|
|
mock_list_devices: AsyncMock,
|
|
mock_get_status: AsyncMock,
|
|
restored_temperature: float,
|
|
expected_temperature: float,
|
|
) -> None:
|
|
"""Test the target temperature is restored in the unit it was published in."""
|
|
hass.config.units = US_CUSTOMARY_SYSTEM
|
|
mock_list_devices.return_value = [
|
|
Remote(
|
|
deviceId="ac-device-id-1",
|
|
deviceName="climate-1",
|
|
remoteType="Air Conditioner",
|
|
hubDeviceId="test-hub-id",
|
|
),
|
|
]
|
|
|
|
entity_id = "climate.climate_1"
|
|
mock_restore_cache(
|
|
hass,
|
|
(State(entity_id, HVACMode.COOL, {ATTR_TEMPERATURE: restored_temperature}),),
|
|
)
|
|
|
|
entry = await configure_integration(hass)
|
|
assert entry.state is ConfigEntryState.LOADED
|
|
|
|
state = hass.states.get(entity_id)
|
|
assert state.attributes[ATTR_TEMPERATURE] == expected_temperature
|
|
|
|
|
|
async def test_air_conditioner_no_last_state(
|
|
hass: HomeAssistant, mock_list_devices, mock_get_status
|
|
) -> None:
|
|
"""Test behavior when no previous state exists."""
|
|
mock_list_devices.return_value = [
|
|
Remote(
|
|
deviceId="ac-device-id-1",
|
|
deviceName="climate-1",
|
|
remoteType="Air Conditioner",
|
|
hubDeviceId="test-hub-id",
|
|
),
|
|
]
|
|
|
|
entry = await configure_integration(hass)
|
|
assert entry.state is ConfigEntryState.LOADED
|
|
|
|
entity_id = "climate.climate_1"
|
|
state = hass.states.get(entity_id)
|
|
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"
|
|
|
|
|
|
async def test_smart_radiator_thermostat_set_temperature(
|
|
hass: HomeAssistant, mock_list_devices, mock_get_status
|
|
) -> None:
|
|
"""Test smart radiator thermostat set temperature."""
|
|
mock_list_devices.return_value = [
|
|
Device(
|
|
version="V1.0",
|
|
deviceId="ac-device-id-1",
|
|
deviceName="climate-1",
|
|
deviceType="Smart Radiator Thermostat",
|
|
hubDeviceId="test-hub-id",
|
|
),
|
|
]
|
|
|
|
mock_get_status.side_effect = [
|
|
{
|
|
"mode": 1,
|
|
"temperature": 27.5,
|
|
},
|
|
{
|
|
"mode": 1,
|
|
"temperature": 27.5,
|
|
},
|
|
{
|
|
"mode": 2,
|
|
"temperature": 27.5,
|
|
},
|
|
]
|
|
entry = await configure_integration(hass)
|
|
assert entry.state is ConfigEntryState.LOADED
|
|
|
|
entity_id = "climate.climate_1"
|
|
|
|
with patch.object(SwitchBotAPI, "send_command") as mock_send_command:
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN,
|
|
SERVICE_SET_TEMPERATURE,
|
|
{ATTR_ENTITY_ID: entity_id, "temperature": 27},
|
|
)
|
|
mock_send_command.assert_called_once_with(
|
|
"ac-device-id-1",
|
|
SmartRadiatorThermostatCommands.SET_MANUAL_MODE_TEMPERATURE,
|
|
"command",
|
|
"27.0",
|
|
)
|
|
|
|
|
|
async def test_smart_radiator_thermostat_set_preset_mode(
|
|
hass: HomeAssistant, mock_list_devices, mock_get_status
|
|
) -> None:
|
|
"""Test smart radiator thermostat set preset mode."""
|
|
mock_list_devices.return_value = [
|
|
Device(
|
|
version="V1.0",
|
|
deviceId="ac-device-id-1",
|
|
deviceName="climate-1",
|
|
deviceType="Smart Radiator Thermostat",
|
|
hubDeviceId="test-hub-id",
|
|
),
|
|
]
|
|
|
|
mock_get_status.side_effect = [
|
|
{
|
|
"mode": 1,
|
|
"temperature": 27.5,
|
|
},
|
|
{
|
|
"mode": 1,
|
|
"temperature": 27.5,
|
|
},
|
|
{
|
|
"mode": 2,
|
|
"temperature": 27.5,
|
|
},
|
|
]
|
|
entry = await configure_integration(hass)
|
|
assert entry.state is ConfigEntryState.LOADED
|
|
|
|
entity_id = "climate.climate_1"
|
|
|
|
with patch.object(SwitchBotAPI, "send_command") as mock_send_command:
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN,
|
|
SERVICE_SET_PRESET_MODE,
|
|
{ATTR_ENTITY_ID: entity_id, "preset_mode": "none"},
|
|
)
|
|
mock_send_command.assert_called_once_with(
|
|
"ac-device-id-1",
|
|
SmartRadiatorThermostatCommands.SET_MODE,
|
|
"command",
|
|
2,
|
|
)
|
|
|
|
with patch.object(SwitchBotAPI, "send_command") as mock_send_command:
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN,
|
|
SERVICE_SET_PRESET_MODE,
|
|
{ATTR_ENTITY_ID: entity_id, "preset_mode": "home"},
|
|
)
|
|
mock_send_command.assert_called_once_with(
|
|
"ac-device-id-1",
|
|
SmartRadiatorThermostatCommands.SET_MODE,
|
|
"command",
|
|
1,
|
|
)
|
|
|
|
|
|
async def test_smart_radiator_thermostat_set_hvac_mode(
|
|
hass: HomeAssistant, mock_list_devices, mock_get_status
|
|
) -> None:
|
|
"""Test smart radiator thermostat set hvac mode."""
|
|
mock_list_devices.return_value = [
|
|
Device(
|
|
version="V1.0",
|
|
deviceId="ac-device-id-1",
|
|
deviceName="climate-1",
|
|
deviceType="Smart Radiator Thermostat",
|
|
hubDeviceId="test-hub-id",
|
|
),
|
|
]
|
|
|
|
mock_get_status.side_effect = [
|
|
{
|
|
"mode": 2,
|
|
"temperature": 27.5,
|
|
},
|
|
{
|
|
"mode": 2,
|
|
"temperature": 27.5,
|
|
},
|
|
{
|
|
"mode": 2,
|
|
"temperature": 27.5,
|
|
},
|
|
]
|
|
entry = await configure_integration(hass)
|
|
assert entry.state is ConfigEntryState.LOADED
|
|
|
|
entity_id = "climate.climate_1"
|
|
|
|
with patch.object(SwitchBotAPI, "send_command") as mock_send_command:
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN,
|
|
SERVICE_SET_HVAC_MODE,
|
|
{ATTR_ENTITY_ID: entity_id, "hvac_mode": HVACMode.OFF},
|
|
)
|
|
mock_send_command.assert_called_once_with(
|
|
"ac-device-id-1",
|
|
SmartRadiatorThermostatCommands.SET_MODE,
|
|
"command",
|
|
2,
|
|
)
|
|
|
|
with patch.object(SwitchBotAPI, "send_command") as mock_send_command:
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN,
|
|
SERVICE_SET_HVAC_MODE,
|
|
{ATTR_ENTITY_ID: entity_id, "hvac_mode": HVACMode.HEAT},
|
|
)
|
|
mock_send_command.assert_called_once_with(
|
|
"ac-device-id-1",
|
|
SmartRadiatorThermostatCommands.SET_MODE,
|
|
"command",
|
|
5,
|
|
)
|