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

Fix flaky recorder migration tests (#123971)

pull/123968/head^2
Erik Montnemery 2024-08-15 10:32:40 +02:00 committed by GitHub
parent 5836f8edb5
commit 81c4bb5f72
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/components/recorder/conftest.py
View File

@ -68,7 +68,7 @@ class InstrumentedMigration:
non_live_migration_done: threading.Event
non_live_migration_done_stall: threading.Event
apply_update_mock: Mock
stall_on_schema_version: int
stall_on_schema_version: int | None
apply_update_stalled: threading.Event
@ -147,7 +147,8 @@ def instrument_migration(
old_version: int,
):
"""Control migration progress."""
if new_version == instrumented_migration.stall_on_schema_version:
stall_version = instrumented_migration.stall_on_schema_version
if stall_version is None or stall_version == new_version:
instrumented_migration.apply_update_stalled.set()
instrumented_migration.migration_stall.wait()
real_apply_update(
@ -179,7 +180,7 @@ def instrument_migration(
non_live_migration_done=threading.Event(),
non_live_migration_done_stall=threading.Event(),
apply_update_mock=apply_update_mock,
stall_on_schema_version=1,
stall_on_schema_version=None,
apply_update_stalled=threading.Event(),
)