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

Use recorder test fixtures in tests (#70652)

Browse source
This commit is contained in:
Erik Montnemery 2022-04-25 14:23:52 +02:00 • committed by GitHub
parent 09b4b7eb37
commit d045e8678d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 35 additions and 30 deletions
Download patch file Download diff file Expand all files Collapse all files

19
tests/components/recorder/test_util.py
View file

@ -18,11 +18,12 @@ from homeassistant.components.recorder.util import (
session_scope,
)
from homeassistant.const import EVENT_HOMEASSISTANT_STOP
from homeassistant.core import HomeAssistant
from homeassistant.util import dt as dt_util
from .common import corrupt_db_file
from tests.common import async_init_recorder_component, async_test_home_assistant
from tests.common import SetupRecorderInstanceT, async_test_home_assistant
def test_session_scope_not_setup(hass_recorder):
@ -96,7 +97,9 @@ def test_validate_or_move_away_sqlite_database(hass, tmpdir, caplog):
assert util.validate_or_move_away_sqlite_database(dburl) is True
async def test_last_run_was_recently_clean(hass, tmp_path):
async def test_last_run_was_recently_clean(
loop, async_setup_recorder_instance: SetupRecorderInstanceT, tmp_path
):
"""Test we can check if the last recorder run was recently clean."""
config = {
recorder.CONF_DB_URL: "sqlite:///" + str(tmp_path / "pytest.db"),
@ -116,7 +119,7 @@ async def test_last_run_was_recently_clean(hass, tmp_path):
"homeassistant.components.recorder.util.last_run_was_recently_clean",
wraps=_last_run_was_recently_clean,
) as last_run_was_recently_clean_mock:
await async_init_recorder_component(hass, config)
await async_setup_recorder_instance(hass, config)
await hass.async_block_till_done()
last_run_was_recently_clean_mock.assert_not_called()
@ -130,7 +133,7 @@ async def test_last_run_was_recently_clean(hass, tmp_path):
wraps=_last_run_was_recently_clean,
) as last_run_was_recently_clean_mock:
hass = await async_test_home_assistant(None)
await async_init_recorder_component(hass, config)
await async_setup_recorder_instance(hass, config)
last_run_was_recently_clean_mock.assert_called_once()
assert return_values[-1] is True
@ -149,7 +152,7 @@ async def test_last_run_was_recently_clean(hass, tmp_path):
return_value=thirty_min_future_time,
):
hass = await async_test_home_assistant(None)
await async_init_recorder_component(hass, config)
await async_setup_recorder_instance(hass, config)
last_run_was_recently_clean_mock.assert_called_once()
assert return_values[-1] is False
@ -594,7 +597,9 @@ def test_periodic_db_cleanups(hass_recorder):
assert str(text_obj) == "PRAGMA wal_checkpoint(TRUNCATE);"
async def test_write_lock_db(hass, tmp_path):
async def test_write_lock_db(
hass: HomeAssistant, async_setup_recorder_instance: SetupRecorderInstanceT, tmp_path
):
"""Test database write lock."""
from sqlalchemy.exc import OperationalError
@ -602,7 +607,7 @@ async def test_write_lock_db(hass, tmp_path):
config = {
recorder.CONF_DB_URL: "sqlite:///" + str(tmp_path / "pytest.db?timeout=0.1")
}
await async_init_recorder_component(hass, config)
await async_setup_recorder_instance(hass, config)
await hass.async_block_till_done()
instance = hass.data[DATA_INSTANCE]