96 lines
2.8 KiB
Python
96 lines
2.8 KiB
Python
"""Test the SmartTub climate platform."""
|
|
|
|
import smarttub
|
|
|
|
from homeassistant.components.climate.const import (
|
|
ATTR_CURRENT_TEMPERATURE,
|
|
ATTR_HVAC_ACTION,
|
|
ATTR_HVAC_MODE,
|
|
ATTR_HVAC_MODES,
|
|
ATTR_MAX_TEMP,
|
|
ATTR_MIN_TEMP,
|
|
ATTR_PRESET_MODE,
|
|
ATTR_PRESET_MODES,
|
|
CURRENT_HVAC_HEAT,
|
|
CURRENT_HVAC_IDLE,
|
|
DOMAIN as CLIMATE_DOMAIN,
|
|
HVAC_MODE_HEAT,
|
|
PRESET_ECO,
|
|
PRESET_NONE,
|
|
SERVICE_SET_HVAC_MODE,
|
|
SERVICE_SET_PRESET_MODE,
|
|
SERVICE_SET_TEMPERATURE,
|
|
SUPPORT_PRESET_MODE,
|
|
SUPPORT_TARGET_TEMPERATURE,
|
|
)
|
|
from homeassistant.components.smarttub.const import DEFAULT_MAX_TEMP, DEFAULT_MIN_TEMP
|
|
from homeassistant.const import (
|
|
ATTR_ENTITY_ID,
|
|
ATTR_SUPPORTED_FEATURES,
|
|
ATTR_TEMPERATURE,
|
|
)
|
|
|
|
from . import trigger_update
|
|
|
|
|
|
async def test_thermostat_update(spa, spa_state, setup_entry, hass):
|
|
"""Test the thermostat entity."""
|
|
|
|
entity_id = f"climate.{spa.brand}_{spa.model}_thermostat"
|
|
state = hass.states.get(entity_id)
|
|
assert state
|
|
|
|
assert state.attributes[ATTR_HVAC_ACTION] == CURRENT_HVAC_HEAT
|
|
|
|
spa_state.heater = "OFF"
|
|
await trigger_update(hass)
|
|
state = hass.states.get(entity_id)
|
|
|
|
assert state.attributes[ATTR_HVAC_ACTION] == CURRENT_HVAC_IDLE
|
|
|
|
assert set(state.attributes[ATTR_HVAC_MODES]) == {HVAC_MODE_HEAT}
|
|
assert state.state == HVAC_MODE_HEAT
|
|
assert (
|
|
state.attributes[ATTR_SUPPORTED_FEATURES]
|
|
== SUPPORT_PRESET_MODE | SUPPORT_TARGET_TEMPERATURE
|
|
)
|
|
assert state.attributes[ATTR_CURRENT_TEMPERATURE] == 38
|
|
assert state.attributes[ATTR_TEMPERATURE] == 39
|
|
assert state.attributes[ATTR_MAX_TEMP] == DEFAULT_MAX_TEMP
|
|
assert state.attributes[ATTR_MIN_TEMP] == DEFAULT_MIN_TEMP
|
|
assert state.attributes[ATTR_PRESET_MODES] == ["none", "eco", "day"]
|
|
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN,
|
|
SERVICE_SET_TEMPERATURE,
|
|
{ATTR_ENTITY_ID: entity_id, ATTR_TEMPERATURE: 37},
|
|
blocking=True,
|
|
)
|
|
spa.set_temperature.assert_called_with(37)
|
|
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN,
|
|
SERVICE_SET_HVAC_MODE,
|
|
{ATTR_ENTITY_ID: entity_id, ATTR_HVAC_MODE: HVAC_MODE_HEAT},
|
|
blocking=True,
|
|
)
|
|
# does nothing
|
|
|
|
assert state.attributes.get(ATTR_PRESET_MODE) == PRESET_NONE
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN,
|
|
SERVICE_SET_PRESET_MODE,
|
|
{ATTR_ENTITY_ID: entity_id, ATTR_PRESET_MODE: PRESET_ECO},
|
|
blocking=True,
|
|
)
|
|
spa.set_heat_mode.assert_called_with(smarttub.Spa.HeatMode.ECONOMY)
|
|
|
|
spa_state.heat_mode = smarttub.Spa.HeatMode.ECONOMY
|
|
await trigger_update(hass)
|
|
state = hass.states.get(entity_id)
|
|
assert state.attributes.get(ATTR_PRESET_MODE) == PRESET_ECO
|
|
|
|
spa.get_status_full.side_effect = smarttub.APIError
|
|
await trigger_update(hass)
|
|
# should not fail
|