evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Refactor recorder states tests (#114073)

Browse source 
https://github.com/home-assistant/core/pull/113985#discussion_r1536596573
This commit is contained in:
J. Nick Koston 2024-03-23 09:41:12 -10:00 • committed by GitHub
parent 4f18f0d902
commit 82016ff528
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 36 additions and 32 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
tests/components/humidifier/test_recorder.py
View file

@ -36,12 +36,13 @@ async def test_exclude_attributes(recorder_mock: Recorder, hass: HomeAssistant)
get_significant_states, hass, now, None, hass.states.async_entity_ids()
)
assert len(states) >= 1
for entity_states in states.values():
for state in filter(
lambda state: split_entity_id(state.entity_id)[0] == humidifier.DOMAIN,
entity_states,
):
assert ATTR_MIN_HUMIDITY not in state.attributes
assert ATTR_MAX_HUMIDITY not in state.attributes
assert ATTR_AVAILABLE_MODES not in state.attributes
assert ATTR_FRIENDLY_NAME in state.attributes
for state in (
state
for entity_states in states.values()
for state in entity_states
if split_entity_id(state.entity_id)[0] == humidifier.DOMAIN
):
assert ATTR_MIN_HUMIDITY not in state.attributes
assert ATTR_MAX_HUMIDITY not in state.attributes
assert ATTR_AVAILABLE_MODES not in state.attributes
assert ATTR_FRIENDLY_NAME in state.attributes

15
tests/components/siren/test_recorder.py
View file

@ -32,10 +32,11 @@ async def test_exclude_attributes(recorder_mock: Recorder, hass: HomeAssistant)
get_significant_states, hass, now, None, hass.states.async_entity_ids()
)
assert len(states) >= 1
for entity_states in states.values():
for state in filter(
lambda state: split_entity_id(state.entity_id)[0] == siren.DOMAIN,
entity_states,
):
assert ATTR_AVAILABLE_TONES not in state.attributes
assert ATTR_FRIENDLY_NAME in state.attributes
for state in (
state
for entity_states in states.values()
for state in entity_states
if split_entity_id(state.entity_id)[0] == siren.DOMAIN
):
assert ATTR_AVAILABLE_TONES not in state.attributes
assert ATTR_FRIENDLY_NAME in state.attributes

15
tests/components/vacuum/test_recorder.py
View file

@ -32,10 +32,11 @@ async def test_exclude_attributes(recorder_mock: Recorder, hass: HomeAssistant)
get_significant_states, hass, now, None, hass.states.async_entity_ids()
)
assert len(states) >= 1
for entity_states in states.values():
for state in filter(
lambda state: split_entity_id(state.entity_id)[0] == vacuum.DOMAIN,
entity_states,
):
assert ATTR_FAN_SPEED_LIST not in state.attributes
assert ATTR_FRIENDLY_NAME in state.attributes
for state in (
state
for entity_states in states.values()
for state in entity_states
if split_entity_id(state.entity_id)[0] == vacuum.DOMAIN
):
assert ATTR_FAN_SPEED_LIST not in state.attributes
assert ATTR_FRIENDLY_NAME in state.attributes

19
tests/components/water_heater/test_recorder.py
View file

@ -36,12 +36,13 @@ async def test_exclude_attributes(recorder_mock: Recorder, hass: HomeAssistant)
get_significant_states, hass, now, None, hass.states.async_entity_ids()
)
assert len(states) >= 1
for entity_states in states.values():
for state in filter(
lambda state: split_entity_id(state.entity_id)[0] == water_heater.DOMAIN,
entity_states,
):
assert ATTR_OPERATION_LIST not in state.attributes
assert ATTR_MIN_TEMP not in state.attributes
assert ATTR_MAX_TEMP not in state.attributes
assert ATTR_FRIENDLY_NAME in state.attributes
for state in (
state
for entity_states in states.values()
for state in entity_states
if split_entity_id(state.entity_id)[0] == water_heater.DOMAIN
):
assert ATTR_OPERATION_LIST not in state.attributes
assert ATTR_MIN_TEMP not in state.attributes
assert ATTR_MAX_TEMP not in state.attributes
assert ATTR_FRIENDLY_NAME in state.attributes