starred/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2026-09-27 09:58:07 -04:00
Code Issues Packages Projects Releases Wiki Activity

Update sensor recorder tests to async (#124061)

Browse Source
This commit is contained in:
Erik Montnemery
2024-08-18 13:20:58 -05:00
committed by GitHub
parent 7d5ddbf51c
commit ce2ffde22e
5 changed files with 54 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/components/recorder/test_init.py
+1 -1
View File
@@ -1366,7 +1366,7 @@ async def test_statistics_runs_initiated(
@pytest.mark.freeze_time("2022-09-13 09:00:00+02:00")
@pytest.mark.parametrize("persistent_database", [True])
@pytest.mark.parametrize("enable_statistics", [True])
@pytest.mark.parametrize("enable_missing_statistics", [True])
@pytest.mark.usefixtures("hass_storage") # Prevent test hass from writing to storage
async def test_compile_missing_statistics(
async_test_recorder: RecorderInstanceGenerator, freezer: FrozenDateTimeFactory
tests/components/sensor/test_recorder_missing_stats.py
+38 -35
View File
@@ -1,7 +1,6 @@
"""The tests for sensor recorder platform can catch up."""
from datetime import datetime, timedelta
from pathlib import Path
import threading
from unittest.mock import patch
@@ -17,11 +16,15 @@ from homeassistant.components.recorder.statistics import (
from homeassistant.components.recorder.util import session_scope
from homeassistant.core import CoreState
from homeassistant.helpers import recorder as recorder_helper
from homeassistant.setup import setup_component
from homeassistant.setup import async_setup_component
import homeassistant.util.dt as dt_util
from tests.common import get_test_home_assistant
from tests.components.recorder.common import do_adhoc_statistics, wait_recording_done
from tests.common import async_test_home_assistant
from tests.components.recorder.common import (
async_wait_recording_done,
do_adhoc_statistics,
)
from tests.typing import RecorderInstanceGenerator
POWER_SENSOR_ATTRIBUTES = {
"device_class": "energy",
@@ -40,37 +43,34 @@ def disable_db_issue_creation():
@pytest.mark.timeout(25)
def test_compile_missing_statistics(
freezer: FrozenDateTimeFactory, recorder_db_url: str, tmp_path: Path
@pytest.mark.parametrize("persistent_database", [True])
@pytest.mark.parametrize("enable_missing_statistics", [True])
@pytest.mark.usefixtures("hass_storage") # Prevent test hass from writing to storage
async def test_compile_missing_statistics(
async_test_recorder: RecorderInstanceGenerator, freezer: FrozenDateTimeFactory
) -> None:
"""Test compile missing statistics."""
if recorder_db_url == "sqlite://":
# On-disk database because we need to stop and start hass
# and have it persist.
recorder_db_url = "sqlite:///" + str(tmp_path / "pytest.db")
config = {
"db_url": recorder_db_url,
}
three_days_ago = datetime(2021, 1, 1, 0, 0, 0, tzinfo=dt_util.UTC)
start_time = three_days_ago + timedelta(days=3)
freezer.move_to(three_days_ago)
with get_test_home_assistant() as hass:
hass.set_state(CoreState.not_running)
async with (
async_test_home_assistant(initial_state=CoreState.not_running) as hass,
async_test_recorder(hass, wait_recorder=False),
):
recorder_helper.async_initialize_recorder(hass)
setup_component(hass, "sensor", {})
setup_component(hass, "recorder", {"recorder": config})
await async_setup_component(hass, "sensor", {})
get_instance(hass).recorder_and_worker_thread_ids.add(threading.get_ident())
hass.start()
wait_recording_done(hass)
wait_recording_done(hass)
await hass.async_start()
await async_wait_recording_done(hass)
await async_wait_recording_done(hass)
hass.states.set("sensor.test1", "0", POWER_SENSOR_ATTRIBUTES)
wait_recording_done(hass)
hass.states.async_set("sensor.test1", "0", POWER_SENSOR_ATTRIBUTES)
await async_wait_recording_done(hass)
two_days_ago = three_days_ago + timedelta(days=1)
freezer.move_to(two_days_ago)
do_adhoc_statistics(hass, start=two_days_ago)
wait_recording_done(hass)
await async_wait_recording_done(hass)
with session_scope(hass=hass, read_only=True) as session:
latest = get_latest_short_term_statistics_with_session(
hass, session, {"sensor.test1"}, {"state", "sum"}
@@ -82,29 +82,32 @@ def test_compile_missing_statistics(
past_time = two_days_ago
while past_time <= start_time:
freezer.move_to(past_time)
hass.states.set("sensor.test1", str(count), POWER_SENSOR_ATTRIBUTES)
hass.states.async_set("sensor.test1", str(count), POWER_SENSOR_ATTRIBUTES)
past_time += timedelta(minutes=5)
count += 1
wait_recording_done(hass)
await async_wait_recording_done(hass)
states = get_significant_states(
hass, three_days_ago, past_time, ["sensor.test1"]
)
assert len(states["sensor.test1"]) == 577
hass.stop()
await hass.async_stop()
await hass.async_block_till_done()
freezer.move_to(start_time)
with get_test_home_assistant() as hass:
hass.set_state(CoreState.not_running)
async with (
async_test_home_assistant(initial_state=CoreState.not_running) as hass,
async_test_recorder(hass, wait_recorder=False),
):
recorder_helper.async_initialize_recorder(hass)
setup_component(hass, "sensor", {})
hass.states.set("sensor.test1", "0", POWER_SENSOR_ATTRIBUTES)
setup_component(hass, "recorder", {"recorder": config})
await async_setup_component(hass, "sensor", {})
hass.states.async_set("sensor.test1", "0", POWER_SENSOR_ATTRIBUTES)
get_instance(hass).recorder_and_worker_thread_ids.add(threading.get_ident())
hass.start()
wait_recording_done(hass)
wait_recording_done(hass)
await hass.async_start()
await async_wait_recording_done(hass)
await async_wait_recording_done(hass)
with session_scope(hass=hass, read_only=True) as session:
latest = get_latest_short_term_statistics_with_session(
hass, session, {"sensor.test1"}, {"state", "sum", "max", "mean", "min"}
@@ -128,4 +131,4 @@ def test_compile_missing_statistics(
assert len(stats["sensor.test1"]) == 48
# Make sure the last mean is 570.5
assert stats["sensor.test1"][-1]["mean"] == 570.5
hass.stop()
await hass.async_stop()