mirror of https://github.com/home-assistant/core
50 lines
1.6 KiB
Python
50 lines
1.6 KiB
Python
"""The sensor tests for the tado platform."""
|
|
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from .util import async_init_integration
|
|
|
|
|
|
async def test_water_heater_create_sensors(hass: HomeAssistant) -> None:
|
|
"""Test creation of water heater."""
|
|
|
|
await async_init_integration(hass)
|
|
|
|
state = hass.states.get("water_heater.water_heater")
|
|
assert state.state == "auto"
|
|
|
|
expected_attributes = {
|
|
"current_temperature": None,
|
|
"friendly_name": "Water Heater",
|
|
"max_temp": 31.0,
|
|
"min_temp": 16.0,
|
|
"operation_list": ["auto", "heat", "off"],
|
|
"operation_mode": "auto",
|
|
"supported_features": 3,
|
|
"target_temp_high": None,
|
|
"target_temp_low": None,
|
|
"temperature": 65.0,
|
|
}
|
|
# Only test for a subset of attributes in case
|
|
# HA changes the implementation and a new one appears
|
|
assert all(item in state.attributes.items() for item in expected_attributes.items())
|
|
|
|
state = hass.states.get("water_heater.second_water_heater")
|
|
assert state.state == "heat"
|
|
|
|
expected_attributes = {
|
|
"current_temperature": None,
|
|
"friendly_name": "Second Water Heater",
|
|
"max_temp": 31.0,
|
|
"min_temp": 16.0,
|
|
"operation_list": ["auto", "heat", "off"],
|
|
"operation_mode": "heat",
|
|
"supported_features": 3,
|
|
"target_temp_high": None,
|
|
"target_temp_low": None,
|
|
"temperature": 30.0,
|
|
}
|
|
# Only test for a subset of attributes in case
|
|
# HA changes the implementation and a new one appears
|
|
assert all(item in state.attributes.items() for item in expected_attributes.items())
|