mirror of https://github.com/home-assistant/core
145 lines
5.5 KiB
Python
145 lines
5.5 KiB
Python
"""The tests for the demo water_heater component."""
|
|
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.components import water_heater
|
|
from homeassistant.const import Platform
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.setup import async_setup_component
|
|
from homeassistant.util.unit_system import US_CUSTOMARY_SYSTEM
|
|
|
|
from tests.components.water_heater import common
|
|
|
|
ENTITY_WATER_HEATER = "water_heater.demo_water_heater"
|
|
ENTITY_WATER_HEATER_CELSIUS = "water_heater.demo_water_heater_celsius"
|
|
|
|
|
|
@pytest.fixture
|
|
async def water_heater_only() -> None:
|
|
"""Enable only the datetime platform."""
|
|
with patch(
|
|
"homeassistant.components.demo.COMPONENTS_WITH_CONFIG_ENTRY_DEMO_PLATFORM",
|
|
[Platform.WATER_HEATER],
|
|
):
|
|
yield
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
async def setup_comp(hass: HomeAssistant, water_heater_only: None):
|
|
"""Set up demo component."""
|
|
hass.config.units = US_CUSTOMARY_SYSTEM
|
|
assert await async_setup_component(
|
|
hass, water_heater.DOMAIN, {"water_heater": {"platform": "demo"}}
|
|
)
|
|
await hass.async_block_till_done()
|
|
|
|
|
|
async def test_setup_params(hass: HomeAssistant) -> None:
|
|
"""Test the initial parameters."""
|
|
state = hass.states.get(ENTITY_WATER_HEATER)
|
|
assert state.attributes.get("temperature") == 119
|
|
assert state.attributes.get("away_mode") == "off"
|
|
assert state.attributes.get("operation_mode") == "eco"
|
|
|
|
|
|
async def test_default_setup_params(hass: HomeAssistant) -> None:
|
|
"""Test the setup with default parameters."""
|
|
state = hass.states.get(ENTITY_WATER_HEATER)
|
|
assert state.attributes.get("min_temp") == 110
|
|
assert state.attributes.get("max_temp") == 140
|
|
|
|
|
|
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_WATER_HEATER)
|
|
assert state.attributes.get("temperature") == 119
|
|
with pytest.raises(vol.Invalid):
|
|
await common.async_set_temperature(hass, None, ENTITY_WATER_HEATER)
|
|
assert state.attributes.get("temperature") == 119
|
|
|
|
|
|
async def test_set_only_target_temp(hass: HomeAssistant) -> None:
|
|
"""Test the setting of the target temperature."""
|
|
state = hass.states.get(ENTITY_WATER_HEATER)
|
|
assert state.attributes.get("temperature") == 119
|
|
await common.async_set_temperature(hass, 110, ENTITY_WATER_HEATER)
|
|
state = hass.states.get(ENTITY_WATER_HEATER)
|
|
assert state.attributes.get("temperature") == 110
|
|
|
|
|
|
async def test_set_operation_bad_attr_and_state(hass: HomeAssistant) -> None:
|
|
"""Test setting operation mode without required attribute.
|
|
|
|
Also check the state.
|
|
"""
|
|
state = hass.states.get(ENTITY_WATER_HEATER)
|
|
assert state.attributes.get("operation_mode") == "eco"
|
|
assert state.state == "eco"
|
|
with pytest.raises(vol.Invalid):
|
|
await common.async_set_operation_mode(hass, None, ENTITY_WATER_HEATER)
|
|
state = hass.states.get(ENTITY_WATER_HEATER)
|
|
assert state.attributes.get("operation_mode") == "eco"
|
|
assert state.state == "eco"
|
|
|
|
|
|
async def test_set_operation(hass: HomeAssistant) -> None:
|
|
"""Test setting of new operation mode."""
|
|
state = hass.states.get(ENTITY_WATER_HEATER)
|
|
assert state.attributes.get("operation_mode") == "eco"
|
|
assert state.state == "eco"
|
|
await common.async_set_operation_mode(hass, "electric", ENTITY_WATER_HEATER)
|
|
state = hass.states.get(ENTITY_WATER_HEATER)
|
|
assert state.attributes.get("operation_mode") == "electric"
|
|
assert state.state == "electric"
|
|
|
|
|
|
async def test_set_away_mode_bad_attr(hass: HomeAssistant) -> None:
|
|
"""Test setting the away mode without required attribute."""
|
|
state = hass.states.get(ENTITY_WATER_HEATER)
|
|
assert state.attributes.get("away_mode") == "off"
|
|
with pytest.raises(vol.Invalid):
|
|
await common.async_set_away_mode(hass, None, ENTITY_WATER_HEATER)
|
|
assert state.attributes.get("away_mode") == "off"
|
|
|
|
|
|
async def test_set_away_mode_on(hass: HomeAssistant) -> None:
|
|
"""Test setting the away mode on/true."""
|
|
await common.async_set_away_mode(hass, True, ENTITY_WATER_HEATER)
|
|
state = hass.states.get(ENTITY_WATER_HEATER)
|
|
assert state.attributes.get("away_mode") == "on"
|
|
|
|
|
|
async def test_set_away_mode_off(hass: HomeAssistant) -> None:
|
|
"""Test setting the away mode off/false."""
|
|
await common.async_set_away_mode(hass, False, ENTITY_WATER_HEATER_CELSIUS)
|
|
state = hass.states.get(ENTITY_WATER_HEATER_CELSIUS)
|
|
assert state.attributes.get("away_mode") == "off"
|
|
|
|
|
|
async def test_set_only_target_temp_with_convert(hass: HomeAssistant) -> None:
|
|
"""Test the setting of the target temperature."""
|
|
state = hass.states.get(ENTITY_WATER_HEATER_CELSIUS)
|
|
assert state.attributes.get("temperature") == 113
|
|
await common.async_set_temperature(hass, 114, ENTITY_WATER_HEATER_CELSIUS)
|
|
state = hass.states.get(ENTITY_WATER_HEATER_CELSIUS)
|
|
assert state.attributes.get("temperature") == 114
|
|
|
|
|
|
async def test_turn_on_off(hass: HomeAssistant) -> None:
|
|
"""Test turn on and off."""
|
|
state = hass.states.get(ENTITY_WATER_HEATER)
|
|
assert state.attributes.get("temperature") == 119
|
|
assert state.attributes.get("away_mode") == "off"
|
|
assert state.attributes.get("operation_mode") == "eco"
|
|
|
|
await common.async_turn_off(hass, ENTITY_WATER_HEATER)
|
|
state = hass.states.get(ENTITY_WATER_HEATER)
|
|
assert state.attributes.get("operation_mode") == "off"
|
|
|
|
await common.async_turn_on(hass, ENTITY_WATER_HEATER)
|
|
state = hass.states.get(ENTITY_WATER_HEATER)
|
|
assert state.attributes.get("operation_mode") == "eco"
|