Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Avoid nesting sessions in recorder statistics tests (#122582)

pull/122568/head
Erik Montnemery 2024-07-25 12:26:44 +02:00 committed by GitHub
parent 78e24be1e7
commit 33d5ed52e6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/components/recorder/test_statistics.py
View File

@ -24,6 +24,7 @@ from homeassistant.components.recorder.statistics import (
get_last_statistics, get_last_statistics,
get_latest_short_term_statistics_with_session, get_latest_short_term_statistics_with_session,
get_metadata, get_metadata,
get_metadata_with_session,
get_short_term_statistics_run_cache, get_short_term_statistics_run_cache,
list_statistic_ids, list_statistic_ids,
) )
@ -293,14 +294,17 @@ def mock_sensor_statistics():
} }
def get_fake_stats(_hass, session, start, _end): def get_fake_stats(_hass, session, start, _end):
instance = recorder.get_instance(_hass)
return statistics.PlatformCompiledStatistics( return statistics.PlatformCompiledStatistics(
[ [
sensor_stats("sensor.test1", start), sensor_stats("sensor.test1", start),
sensor_stats("sensor.test2", start), sensor_stats("sensor.test2", start),
sensor_stats("sensor.test3", start), sensor_stats("sensor.test3", start),
], ],
get_metadata( get_metadata_with_session(
_hass, statistic_ids={"sensor.test1", "sensor.test2", "sensor.test3"} instance,
session,
statistic_ids={"sensor.test1", "sensor.test2", "sensor.test3"},
), ),
) )