jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-25 22:27:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix lingering timers in sensor tests (#90856)

Browse Source
This commit is contained in:
epenet 2023-04-05 19:52:30 +02:00 committed by GitHub
parent 04cfd7b41d
commit 19ab76dad9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 18 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
tests/components/sensor/test_recorder.py
View File

@ -32,14 +32,13 @@ from homeassistant.components.recorder.statistics import (
list_statistic_ids, list_statistic_ids,
) )
from homeassistant.components.recorder.util import get_instance, session_scope from homeassistant.components.recorder.util import get_instance, session_scope
from homeassistant.components.sensor import ATTR_OPTIONS, DOMAIN from homeassistant.components.sensor import ATTR_OPTIONS, SensorDeviceClass
from homeassistant.const import ATTR_FRIENDLY_NAME, STATE_UNAVAILABLE from homeassistant.const import ATTR_FRIENDLY_NAME, STATE_UNAVAILABLE
from homeassistant.core import HomeAssistant, State from homeassistant.core import HomeAssistant, State
from homeassistant.setup import async_setup_component, setup_component from homeassistant.setup import async_setup_component, setup_component
import homeassistant.util.dt as dt_util import homeassistant.util.dt as dt_util
from homeassistant.util.unit_system import METRIC_SYSTEM, US_CUSTOMARY_SYSTEM from homeassistant.util.unit_system import METRIC_SYSTEM, US_CUSTOMARY_SYSTEM
from tests.common import async_fire_time_changed
from tests.components.recorder.common import ( from tests.components.recorder.common import (
assert_dict_of_states_equal_without_context_and_last_changed, assert_dict_of_states_equal_without_context_and_last_changed,
assert_multiple_states_equal_without_context_and_last_changed, assert_multiple_states_equal_without_context_and_last_changed,
@ -5059,11 +5058,21 @@ def record_states_partially_unavailable(hass, zero, entity_id, attributes):
return four, states return four, states
async def test_exclude_attributes(recorder_mock: Recorder, hass: HomeAssistant) -> None: async def test_exclude_attributes(
recorder_mock: Recorder, hass: HomeAssistant, enable_custom_integrations: None
) -> None:
"""Test sensor attributes to be excluded.""" """Test sensor attributes to be excluded."""
await async_setup_component(hass, DOMAIN, {DOMAIN: {"platform": "demo"}}) platform = getattr(hass.components, "test.sensor")
await hass.async_block_till_done() platform.init(empty=True)
async_fire_time_changed(hass, dt_util.utcnow() + timedelta(minutes=5)) platform.ENTITIES["0"] = platform.MockSensor(
has_entity_name=True,
unique_id="test",
name="Test",
native_value="option1",
device_class=SensorDeviceClass.ENUM,
options=["option1", "option2"],
)
assert await async_setup_component(hass, "sensor", {"sensor": {"platform": "test"}})
await hass.async_block_till_done() await hass.async_block_till_done()
await async_wait_recording_done(hass) await async_wait_recording_done(hass)
@ -5085,9 +5094,6 @@ async def test_exclude_attributes(recorder_mock: Recorder, hass: HomeAssistant)
return native_states return native_states
states: list[State] = await hass.async_add_executor_job(_fetch_states) states: list[State] = await hass.async_add_executor_job(_fetch_states)
assert len(states) > 1 assert len(states) == 1
for state in states: assert ATTR_OPTIONS not in states[0].attributes
if state.domain != DOMAIN: assert ATTR_FRIENDLY_NAME in states[0].attributes
continue
assert ATTR_OPTIONS not in state.attributes
assert ATTR_FRIENDLY_NAME in state.attributes