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

Fix unclean shutdown of recorder test (#47791)

Browse Source
This commit is contained in:
Erik Montnemery 2021-03-12 23:17:27 +01:00 committed by GitHub
parent 07aeb8d160
commit 2178e27fb4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
tests/components/recorder/test_migrate.py
View File

@ -8,7 +8,8 @@ from sqlalchemy.exc import InternalError, OperationalError, ProgrammingError
from sqlalchemy.pool import StaticPool from sqlalchemy.pool import StaticPool
from homeassistant.bootstrap import async_setup_component from homeassistant.bootstrap import async_setup_component
from homeassistant.components.recorder import const, migration, models from homeassistant.components.recorder import RecorderRuns, const, migration, models
import homeassistant.util.dt as dt_util
from tests.components.recorder import models_original from tests.components.recorder import models_original
@ -51,8 +52,16 @@ async def test_schema_migrate(hass):
throwing exceptions. Maintaining a set of assertions based on schema throwing exceptions. Maintaining a set of assertions based on schema
inspection could quickly become quite cumbersome. inspection could quickly become quite cumbersome.
""" """
def _mock_setup_run(self):
self.run_info = RecorderRuns(
start=self.recording_start, created=dt_util.utcnow()
)
with patch("sqlalchemy.create_engine", new=create_engine_test), patch( with patch("sqlalchemy.create_engine", new=create_engine_test), patch(
"homeassistant.components.recorder.Recorder._setup_run" "homeassistant.components.recorder.Recorder._setup_run",
side_effect=_mock_setup_run,
autospec=True,
) as setup_run: ) as setup_run:
await async_setup_component( await async_setup_component(
hass, "recorder", {"recorder": {"db_url": "sqlite://"}} hass, "recorder", {"recorder": {"db_url": "sqlite://"}}