Home-Assistant/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-12-24 21:06:19 +00:00
Code Activity

Move recorder data migration implementation to migrators (#122045)

Browse Source
This commit is contained in:
Erik Montnemery
2024-07-17 10:59:58 +02:00
committed by GitHub
parent 7f76de2c7a
commit 6921e053e4
2 changed files with 214 additions and 227 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
tests/conftest.py
View File

@@ -1451,16 +1451,22 @@ async def async_test_recorder(
else None
)
migrate_states_context_ids = (
migration.migrate_states_context_ids if enable_migrate_context_ids else None
migration.StatesContextIDMigration.migrate_data
if enable_migrate_context_ids
else None
)
migrate_events_context_ids = (
migration.migrate_events_context_ids if enable_migrate_context_ids else None
migration.EventsContextIDMigration.migrate_data
if enable_migrate_context_ids
else None
)
migrate_event_type_ids = (
migration.migrate_event_type_ids if enable_migrate_event_type_ids else None
migration.EventTypeIDMigration.migrate_data
if enable_migrate_event_type_ids
else None
)
migrate_entity_ids = (
migration.migrate_entity_ids if enable_migrate_entity_ids else None
migration.EntityIDMigration.migrate_data if enable_migrate_entity_ids else None
)
legacy_event_id_foreign_key_exists = (
recorder.Recorder._legacy_event_id_foreign_key_exists
@@ -1484,22 +1490,22 @@ async def async_test_recorder(
autospec=True,
),
patch(
"homeassistant.components.recorder.migration.migrate_events_context_ids",
"homeassistant.components.recorder.migration.EventsContextIDMigration.migrate_data",
side_effect=migrate_events_context_ids,
autospec=True,
),
patch(
"homeassistant.components.recorder.migration.migrate_states_context_ids",
"homeassistant.components.recorder.migration.StatesContextIDMigration.migrate_data",
side_effect=migrate_states_context_ids,
autospec=True,
),
patch(
"homeassistant.components.recorder.migration.migrate_event_type_ids",
"homeassistant.components.recorder.migration.EventTypeIDMigration.migrate_data",
side_effect=migrate_event_type_ids,
autospec=True,
),
patch(
"homeassistant.components.recorder.migration.migrate_entity_ids",
"homeassistant.components.recorder.migration.EntityIDMigration.migrate_data",
side_effect=migrate_entity_ids,
autospec=True,
),