mirror of https://github.com/home-assistant/core
424 lines
13 KiB
Python
424 lines
13 KiB
Python
"""The tests for the demo climate component."""
|
|
|
|
from collections.abc import Generator
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.components.climate import (
|
|
ATTR_CURRENT_HUMIDITY,
|
|
ATTR_CURRENT_TEMPERATURE,
|
|
ATTR_FAN_MODE,
|
|
ATTR_HUMIDITY,
|
|
ATTR_HVAC_ACTION,
|
|
ATTR_HVAC_MODE,
|
|
ATTR_HVAC_MODES,
|
|
ATTR_MAX_HUMIDITY,
|
|
ATTR_MAX_TEMP,
|
|
ATTR_MIN_HUMIDITY,
|
|
ATTR_MIN_TEMP,
|
|
ATTR_PRESET_MODE,
|
|
ATTR_SWING_MODE,
|
|
ATTR_TARGET_TEMP_HIGH,
|
|
ATTR_TARGET_TEMP_LOW,
|
|
DOMAIN as CLIMATE_DOMAIN,
|
|
PRESET_AWAY,
|
|
PRESET_ECO,
|
|
SERVICE_SET_FAN_MODE,
|
|
SERVICE_SET_HUMIDITY,
|
|
SERVICE_SET_HVAC_MODE,
|
|
SERVICE_SET_PRESET_MODE,
|
|
SERVICE_SET_SWING_MODE,
|
|
SERVICE_SET_TEMPERATURE,
|
|
HVACAction,
|
|
HVACMode,
|
|
)
|
|
from homeassistant.const import (
|
|
ATTR_ENTITY_ID,
|
|
ATTR_TEMPERATURE,
|
|
SERVICE_TURN_OFF,
|
|
SERVICE_TURN_ON,
|
|
Platform,
|
|
)
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.setup import async_setup_component
|
|
from homeassistant.util.unit_system import METRIC_SYSTEM
|
|
|
|
ENTITY_CLIMATE = "climate.hvac"
|
|
ENTITY_ECOBEE = "climate.ecobee"
|
|
ENTITY_HEATPUMP = "climate.heatpump"
|
|
|
|
|
|
@pytest.fixture
|
|
def climate_only() -> Generator[None]:
|
|
"""Enable only the climate platform."""
|
|
with patch(
|
|
"homeassistant.components.demo.COMPONENTS_WITH_CONFIG_ENTRY_DEMO_PLATFORM",
|
|
[Platform.CLIMATE],
|
|
):
|
|
yield
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
async def setup_demo_climate(hass: HomeAssistant, climate_only: None) -> None:
|
|
"""Initialize setup demo climate."""
|
|
hass.config.units = METRIC_SYSTEM
|
|
assert await async_setup_component(
|
|
hass, CLIMATE_DOMAIN, {"climate": {"platform": "demo"}}
|
|
)
|
|
await hass.async_block_till_done()
|
|
|
|
|
|
def test_setup_params(hass: HomeAssistant) -> None:
|
|
"""Test the initial parameters."""
|
|
state = hass.states.get(ENTITY_CLIMATE)
|
|
assert state.state == HVACMode.COOL
|
|
assert state.attributes.get(ATTR_TEMPERATURE) == 21
|
|
assert state.attributes.get(ATTR_CURRENT_TEMPERATURE) == 22
|
|
assert state.attributes.get(ATTR_FAN_MODE) == "on_high"
|
|
assert state.attributes.get(ATTR_HUMIDITY) == 67.4
|
|
assert state.attributes.get(ATTR_CURRENT_HUMIDITY) == 54.2
|
|
assert state.attributes.get(ATTR_SWING_MODE) == "off"
|
|
assert state.attributes.get(ATTR_HVAC_MODES) == [
|
|
HVACMode.OFF,
|
|
HVACMode.HEAT,
|
|
HVACMode.COOL,
|
|
HVACMode.AUTO,
|
|
HVACMode.DRY,
|
|
HVACMode.FAN_ONLY,
|
|
]
|
|
|
|
|
|
def test_default_setup_params(hass: HomeAssistant) -> None:
|
|
"""Test the setup with default parameters."""
|
|
state = hass.states.get(ENTITY_CLIMATE)
|
|
assert state.attributes.get(ATTR_MIN_TEMP) == 7
|
|
assert state.attributes.get(ATTR_MAX_TEMP) == 35
|
|
assert state.attributes.get(ATTR_MIN_HUMIDITY) == 30
|
|
assert state.attributes.get(ATTR_MAX_HUMIDITY) == 99
|
|
|
|
|
|
async def test_set_only_target_temp_bad_attr(hass: HomeAssistant) -> None:
|
|
"""Test setting the target temperature without required attribute."""
|
|
state = hass.states.get(ENTITY_CLIMATE)
|
|
assert state.attributes.get(ATTR_TEMPERATURE) == 21
|
|
|
|
with pytest.raises(vol.Invalid):
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN,
|
|
SERVICE_SET_TEMPERATURE,
|
|
{ATTR_ENTITY_ID: ENTITY_CLIMATE, ATTR_TEMPERATURE: None},
|
|
blocking=True,
|
|
)
|
|
|
|
state = hass.states.get(ENTITY_CLIMATE)
|
|
assert state.attributes.get(ATTR_TEMPERATURE) == 21
|
|
|
|
|
|
async def test_set_only_target_temp(hass: HomeAssistant) -> None:
|
|
"""Test the setting of the target temperature."""
|
|
state = hass.states.get(ENTITY_CLIMATE)
|
|
assert state.attributes.get(ATTR_TEMPERATURE) == 21
|
|
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN,
|
|
SERVICE_SET_TEMPERATURE,
|
|
{ATTR_ENTITY_ID: ENTITY_CLIMATE, ATTR_TEMPERATURE: 30},
|
|
blocking=True,
|
|
)
|
|
|
|
state = hass.states.get(ENTITY_CLIMATE)
|
|
assert state.attributes.get(ATTR_TEMPERATURE) == 30.0
|
|
|
|
|
|
async def test_set_only_target_temp_with_convert(hass: HomeAssistant) -> None:
|
|
"""Test the setting of the target temperature."""
|
|
state = hass.states.get(ENTITY_HEATPUMP)
|
|
assert state.attributes.get(ATTR_TEMPERATURE) == 20
|
|
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN,
|
|
SERVICE_SET_TEMPERATURE,
|
|
{ATTR_ENTITY_ID: ENTITY_HEATPUMP, ATTR_TEMPERATURE: 21},
|
|
blocking=True,
|
|
)
|
|
|
|
state = hass.states.get(ENTITY_HEATPUMP)
|
|
assert state.attributes.get(ATTR_TEMPERATURE) == 21.0
|
|
|
|
|
|
async def test_set_target_temp_range(hass: HomeAssistant) -> None:
|
|
"""Test the setting of the target temperature with range."""
|
|
state = hass.states.get(ENTITY_ECOBEE)
|
|
assert state.attributes.get(ATTR_TEMPERATURE) is None
|
|
assert state.attributes.get(ATTR_TARGET_TEMP_LOW) == 21.0
|
|
assert state.attributes.get(ATTR_TARGET_TEMP_HIGH) == 24.0
|
|
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN,
|
|
SERVICE_SET_TEMPERATURE,
|
|
{
|
|
ATTR_ENTITY_ID: ENTITY_ECOBEE,
|
|
ATTR_TARGET_TEMP_LOW: 20,
|
|
ATTR_TARGET_TEMP_HIGH: 25,
|
|
},
|
|
blocking=True,
|
|
)
|
|
|
|
state = hass.states.get(ENTITY_ECOBEE)
|
|
assert state.attributes.get(ATTR_TEMPERATURE) is None
|
|
assert state.attributes.get(ATTR_TARGET_TEMP_LOW) == 20.0
|
|
assert state.attributes.get(ATTR_TARGET_TEMP_HIGH) == 25.0
|
|
|
|
|
|
async def test_set_target_temp_range_bad_attr(hass: HomeAssistant) -> None:
|
|
"""Test setting the target temperature range without attribute."""
|
|
state = hass.states.get(ENTITY_ECOBEE)
|
|
assert state.attributes.get(ATTR_TEMPERATURE) is None
|
|
assert state.attributes.get(ATTR_TARGET_TEMP_LOW) == 21.0
|
|
assert state.attributes.get(ATTR_TARGET_TEMP_HIGH) == 24.0
|
|
|
|
with pytest.raises(vol.Invalid):
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN,
|
|
SERVICE_SET_TEMPERATURE,
|
|
{
|
|
ATTR_ENTITY_ID: ENTITY_ECOBEE,
|
|
ATTR_TARGET_TEMP_LOW: None,
|
|
ATTR_TARGET_TEMP_HIGH: None,
|
|
},
|
|
blocking=True,
|
|
)
|
|
|
|
state = hass.states.get(ENTITY_ECOBEE)
|
|
assert state.attributes.get(ATTR_TEMPERATURE) is None
|
|
assert state.attributes.get(ATTR_TARGET_TEMP_LOW) == 21.0
|
|
assert state.attributes.get(ATTR_TARGET_TEMP_HIGH) == 24.0
|
|
|
|
|
|
async def test_set_temp_with_hvac_mode(hass: HomeAssistant) -> None:
|
|
"""Test the setting of the hvac_mode in set_temperature."""
|
|
state = hass.states.get(ENTITY_CLIMATE)
|
|
assert state.attributes.get(ATTR_TEMPERATURE) == 21
|
|
assert state.state == HVACMode.COOL
|
|
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN,
|
|
SERVICE_SET_TEMPERATURE,
|
|
{
|
|
ATTR_ENTITY_ID: ENTITY_CLIMATE,
|
|
ATTR_TEMPERATURE: 23,
|
|
ATTR_HVAC_MODE: HVACMode.OFF,
|
|
},
|
|
blocking=True,
|
|
)
|
|
|
|
state = hass.states.get(ENTITY_CLIMATE)
|
|
assert state.state == HVACMode.OFF
|
|
assert state.attributes.get(ATTR_TEMPERATURE) == 23
|
|
|
|
|
|
async def test_set_target_humidity_bad_attr(hass: HomeAssistant) -> None:
|
|
"""Test setting the target humidity without required attribute."""
|
|
state = hass.states.get(ENTITY_CLIMATE)
|
|
assert state.attributes.get(ATTR_HUMIDITY) == 67.4
|
|
|
|
with pytest.raises(vol.Invalid):
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN,
|
|
SERVICE_SET_HUMIDITY,
|
|
{ATTR_ENTITY_ID: ENTITY_CLIMATE, ATTR_HUMIDITY: None},
|
|
blocking=True,
|
|
)
|
|
|
|
state = hass.states.get(ENTITY_CLIMATE)
|
|
assert state.attributes.get(ATTR_HUMIDITY) == 67.4
|
|
|
|
|
|
async def test_set_target_humidity(hass: HomeAssistant) -> None:
|
|
"""Test the setting of the target humidity."""
|
|
state = hass.states.get(ENTITY_CLIMATE)
|
|
assert state.attributes.get(ATTR_HUMIDITY) == 67.4
|
|
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN,
|
|
SERVICE_SET_HUMIDITY,
|
|
{ATTR_ENTITY_ID: ENTITY_CLIMATE, ATTR_HUMIDITY: 64},
|
|
blocking=True,
|
|
)
|
|
|
|
state = hass.states.get(ENTITY_CLIMATE)
|
|
assert state.attributes.get(ATTR_HUMIDITY) == 64.0
|
|
|
|
|
|
async def test_set_fan_mode_bad_attr(hass: HomeAssistant) -> None:
|
|
"""Test setting fan mode without required attribute."""
|
|
state = hass.states.get(ENTITY_CLIMATE)
|
|
assert state.attributes.get(ATTR_FAN_MODE) == "on_high"
|
|
|
|
with pytest.raises(vol.Invalid):
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN,
|
|
SERVICE_SET_FAN_MODE,
|
|
{ATTR_ENTITY_ID: ENTITY_CLIMATE, ATTR_FAN_MODE: None},
|
|
blocking=True,
|
|
)
|
|
|
|
state = hass.states.get(ENTITY_CLIMATE)
|
|
assert state.attributes.get(ATTR_FAN_MODE) == "on_high"
|
|
|
|
|
|
async def test_set_fan_mode(hass: HomeAssistant) -> None:
|
|
"""Test setting of new fan mode."""
|
|
state = hass.states.get(ENTITY_CLIMATE)
|
|
assert state.attributes.get(ATTR_FAN_MODE) == "on_high"
|
|
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN,
|
|
SERVICE_SET_FAN_MODE,
|
|
{ATTR_ENTITY_ID: ENTITY_CLIMATE, ATTR_FAN_MODE: "on_low"},
|
|
blocking=True,
|
|
)
|
|
|
|
state = hass.states.get(ENTITY_CLIMATE)
|
|
assert state.attributes.get(ATTR_FAN_MODE) == "on_low"
|
|
|
|
|
|
async def test_set_swing_mode_bad_attr(hass: HomeAssistant) -> None:
|
|
"""Test setting swing mode without required attribute."""
|
|
state = hass.states.get(ENTITY_CLIMATE)
|
|
assert state.attributes.get(ATTR_SWING_MODE) == "off"
|
|
|
|
with pytest.raises(vol.Invalid):
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN,
|
|
SERVICE_SET_SWING_MODE,
|
|
{ATTR_ENTITY_ID: ENTITY_CLIMATE, ATTR_SWING_MODE: None},
|
|
blocking=True,
|
|
)
|
|
|
|
state = hass.states.get(ENTITY_CLIMATE)
|
|
assert state.attributes.get(ATTR_SWING_MODE) == "off"
|
|
|
|
|
|
async def test_set_swing(hass: HomeAssistant) -> None:
|
|
"""Test setting of new swing mode."""
|
|
state = hass.states.get(ENTITY_CLIMATE)
|
|
assert state.attributes.get(ATTR_SWING_MODE) == "off"
|
|
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN,
|
|
SERVICE_SET_SWING_MODE,
|
|
{ATTR_ENTITY_ID: ENTITY_CLIMATE, ATTR_SWING_MODE: "auto"},
|
|
blocking=True,
|
|
)
|
|
|
|
state = hass.states.get(ENTITY_CLIMATE)
|
|
assert state.attributes.get(ATTR_SWING_MODE) == "auto"
|
|
|
|
|
|
async def test_set_hvac_bad_attr_and_state(hass: HomeAssistant) -> None:
|
|
"""Test setting hvac mode without required attribute.
|
|
|
|
Also check the state.
|
|
"""
|
|
state = hass.states.get(ENTITY_CLIMATE)
|
|
assert state.attributes.get(ATTR_HVAC_ACTION) == HVACAction.COOLING
|
|
assert state.state == HVACMode.COOL
|
|
|
|
with pytest.raises(vol.Invalid):
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN,
|
|
SERVICE_SET_HVAC_MODE,
|
|
{ATTR_ENTITY_ID: ENTITY_CLIMATE, ATTR_HVAC_MODE: None},
|
|
blocking=True,
|
|
)
|
|
|
|
state = hass.states.get(ENTITY_CLIMATE)
|
|
assert state.attributes.get(ATTR_HVAC_ACTION) == HVACAction.COOLING
|
|
assert state.state == HVACMode.COOL
|
|
|
|
|
|
async def test_set_hvac(hass: HomeAssistant) -> None:
|
|
"""Test setting of new hvac mode."""
|
|
state = hass.states.get(ENTITY_CLIMATE)
|
|
assert state.state == HVACMode.COOL
|
|
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN,
|
|
SERVICE_SET_HVAC_MODE,
|
|
{ATTR_ENTITY_ID: ENTITY_CLIMATE, ATTR_HVAC_MODE: HVACMode.HEAT},
|
|
blocking=True,
|
|
)
|
|
|
|
state = hass.states.get(ENTITY_CLIMATE)
|
|
assert state.state == HVACMode.HEAT
|
|
|
|
|
|
async def test_set_hold_mode_away(hass: HomeAssistant) -> None:
|
|
"""Test setting the hold mode away."""
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN,
|
|
SERVICE_SET_PRESET_MODE,
|
|
{ATTR_ENTITY_ID: ENTITY_ECOBEE, ATTR_PRESET_MODE: PRESET_AWAY},
|
|
blocking=True,
|
|
)
|
|
|
|
state = hass.states.get(ENTITY_ECOBEE)
|
|
assert state.attributes.get(ATTR_PRESET_MODE) == PRESET_AWAY
|
|
|
|
|
|
async def test_set_hold_mode_eco(hass: HomeAssistant) -> None:
|
|
"""Test setting the hold mode eco."""
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN,
|
|
SERVICE_SET_PRESET_MODE,
|
|
{ATTR_ENTITY_ID: ENTITY_ECOBEE, ATTR_PRESET_MODE: PRESET_ECO},
|
|
blocking=True,
|
|
)
|
|
|
|
state = hass.states.get(ENTITY_ECOBEE)
|
|
assert state.attributes.get(ATTR_PRESET_MODE) == PRESET_ECO
|
|
|
|
|
|
async def test_turn_on(hass: HomeAssistant) -> None:
|
|
"""Test turn on device."""
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN,
|
|
SERVICE_SET_HVAC_MODE,
|
|
{ATTR_ENTITY_ID: ENTITY_CLIMATE, ATTR_HVAC_MODE: HVACMode.OFF},
|
|
blocking=True,
|
|
)
|
|
|
|
state = hass.states.get(ENTITY_CLIMATE)
|
|
assert state.state == HVACMode.OFF
|
|
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN, SERVICE_TURN_ON, {ATTR_ENTITY_ID: ENTITY_CLIMATE}, blocking=True
|
|
)
|
|
state = hass.states.get(ENTITY_CLIMATE)
|
|
assert state.state == HVACMode.HEAT
|
|
|
|
|
|
async def test_turn_off(hass: HomeAssistant) -> None:
|
|
"""Test turn on device."""
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN,
|
|
SERVICE_SET_HVAC_MODE,
|
|
{ATTR_ENTITY_ID: ENTITY_CLIMATE, ATTR_HVAC_MODE: HVACMode.HEAT},
|
|
blocking=True,
|
|
)
|
|
|
|
state = hass.states.get(ENTITY_CLIMATE)
|
|
assert state.state == HVACMode.HEAT
|
|
|
|
await hass.services.async_call(
|
|
CLIMATE_DOMAIN,
|
|
SERVICE_TURN_OFF,
|
|
{ATTR_ENTITY_ID: ENTITY_CLIMATE},
|
|
blocking=True,
|
|
)
|
|
state = hass.states.get(ENTITY_CLIMATE)
|
|
assert state.state == HVACMode.OFF
|