jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-08-02 01:58:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve Tado water heater tests (#149806)

Browse Source
This commit is contained in:
Joost Lekkerkerker 2025-08-01 14:38:02 +02:00 committed by GitHub
parent 37579440e6
commit 506431c75f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 163 additions and 38 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

139
tests/components/tado/snapshots/test_water_heater.ambr Normal file
View File

@ -0,0 +1,139 @@
# serializer version: 1
# name: test_entities[water_heater.second_water_heater-entry]
EntityRegistryEntrySnapshot({
'aliases': set({
}),
'area_id': None,
'capabilities': dict({
'max_temp': 31.0,
'min_temp': 16.0,
'operation_list': list([
'auto',
'heat',
'off',
]),
}),
'config_entry_id': <ANY>,
'config_subentry_id': <ANY>,
'device_class': None,
'device_id': <ANY>,
'disabled_by': None,
'domain': 'water_heater',
'entity_category': None,
'entity_id': 'water_heater.second_water_heater',
'has_entity_name': True,
'hidden_by': None,
'icon': None,
'id': <ANY>,
'labels': set({
}),
'name': None,
'options': dict({
}),
'original_device_class': None,
'original_icon': None,
'original_name': None,
'platform': 'tado',
'previous_unique_id': None,
'suggested_object_id': None,
'supported_features': <WaterHeaterEntityFeature: 3>,
'translation_key': None,
'unique_id': '4 1',
'unit_of_measurement': None,
})
# ---
# name: test_entities[water_heater.second_water_heater-state]
StateSnapshot({
'attributes': ReadOnlyDict({
'current_temperature': None,
'friendly_name': 'Second Water Heater',
'max_temp': 31.0,
'min_temp': 16.0,
'operation_list': list([
'auto',
'heat',
'off',
]),
'operation_mode': 'heat',
'supported_features': <WaterHeaterEntityFeature: 3>,
'target_temp_high': None,
'target_temp_low': None,
'temperature': 30.0,
}),
'context': <ANY>,
'entity_id': 'water_heater.second_water_heater',
'last_changed': <ANY>,
'last_reported': <ANY>,
'last_updated': <ANY>,
'state': 'heat',
})
# ---
# name: test_entities[water_heater.water_heater-entry]
EntityRegistryEntrySnapshot({
'aliases': set({
}),
'area_id': None,
'capabilities': dict({
'max_temp': 31.0,
'min_temp': 16.0,
'operation_list': list([
'auto',
'heat',
'off',
]),
}),
'config_entry_id': <ANY>,
'config_subentry_id': <ANY>,
'device_class': None,
'device_id': <ANY>,
'disabled_by': None,
'domain': 'water_heater',
'entity_category': None,
'entity_id': 'water_heater.water_heater',
'has_entity_name': True,
'hidden_by': None,
'icon': None,
'id': <ANY>,
'labels': set({
}),
'name': None,
'options': dict({
}),
'original_device_class': None,
'original_icon': None,
'original_name': None,
'platform': 'tado',
'previous_unique_id': None,
'suggested_object_id': None,
'supported_features': <WaterHeaterEntityFeature: 3>,
'translation_key': None,
'unique_id': '2 1',
'unit_of_measurement': None,
})
# ---
# name: test_entities[water_heater.water_heater-state]
StateSnapshot({
'attributes': ReadOnlyDict({
'current_temperature': None,
'friendly_name': 'Water Heater',
'max_temp': 31.0,
'min_temp': 16.0,
'operation_list': list([
'auto',
'heat',
'off',
]),
'operation_mode': 'auto',
'supported_features': <WaterHeaterEntityFeature: 3>,
'target_temp_high': None,
'target_temp_low': None,
'temperature': 65.0,
}),
'context': <ANY>,
'entity_id': 'water_heater.water_heater',
'last_changed': <ANY>,
'last_reported': <ANY>,
'last_updated': <ANY>,
'state': 'auto',
})
# ---

62
tests/components/tado/test_water_heater.py
View File

@ -1,49 +1,35 @@
"""The sensor tests for the tado platform."""
"""The water heater tests for the tado platform."""
from collections.abc import AsyncGenerator
from unittest.mock import patch
import pytest
from syrupy.assertion import SnapshotAssertion
from homeassistant.components.tado import DOMAIN
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
from .util import async_init_integration
from tests.common import MockConfigEntry, snapshot_platform
async def test_water_heater_create_sensors(hass: HomeAssistant) -> None:
@pytest.fixture(autouse=True)
def setup_platforms() -> AsyncGenerator[None]:
"""Set up the platforms for the tests."""
with patch("homeassistant.components.tado.PLATFORMS", [Platform.WATER_HEATER]):
yield
async def test_entities(
hass: HomeAssistant, entity_registry: er.EntityRegistry, snapshot: SnapshotAssertion
) -> None:
"""Test creation of water heater."""
await async_init_integration(hass)
state = hass.states.get("water_heater.water_heater")
assert state.state == "auto"
config_entry: MockConfigEntry = hass.config_entries.async_entries(DOMAIN)[0]
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())
await snapshot_platform(hass, entity_registry, snapshot, config_entry.entry_id)