home-assistant/core
1
0
Fork 1
mirror of https://github.com/home-assistant/core.git synced 2025-08-13 17:45:19 +02:00
Code Issues Packages Projects Releases Wiki Activity

adjust

Browse Source
This commit is contained in:
J. Nick Koston
2023-03-09 15:41:30 -10:00
parent 860b92610b
commit 357da784fa
2 changed files with 1 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
homeassistant/components/recorder/migration.py
View File

@@ -1324,6 +1324,7 @@ def migrate_event_type_ids(instance: Recorder) -> bool:
[ [
{ {
"event_id": event_id, "event_id": event_id,
"event_type": None,
"event_type_id": event_type_to_id[event_type], "event_type_id": event_type_to_id[event_type],
} }
for event_id, event_type in events for event_id, event_type in events

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

@@ -703,51 +703,24 @@ async def test_migrate_event_type_ids(
instance = await async_setup_recorder_instance(hass) instance = await async_setup_recorder_instance(hass)
await async_wait_recording_done(hass) await async_wait_recording_done(hass)
test_uuid = uuid.uuid4()
uuid_hex = test_uuid.hex
def _insert_events(): def _insert_events():
with session_scope(hass=hass) as session: with session_scope(hass=hass) as session:
session.add_all( session.add_all(
( (
Events( Events(
event_type="event_type_one", event_type="event_type_one",
event_data=None,
origin_idx=0, origin_idx=0,
time_fired=None,
time_fired_ts=1677721632.452529, time_fired_ts=1677721632.452529,
context_id=uuid_hex,
context_id_bin=None,
context_user_id=None,
context_user_id_bin=None,
context_parent_id=None,
context_parent_id_bin=None,
), ),
Events( Events(
event_type="event_type_one", event_type="event_type_one",
event_data=None,
origin_idx=0, origin_idx=0,
time_fired=None,
time_fired_ts=1677721632.552529, time_fired_ts=1677721632.552529,
context_id=None,
context_id_bin=None,
context_user_id=None,
context_user_id_bin=None,
context_parent_id=None,
context_parent_id_bin=None,
), ),
Events( Events(
event_type="event_type_two", event_type="event_type_two",
event_data=None,
origin_idx=0, origin_idx=0,
time_fired=None,
time_fired_ts=1677721632.552529, time_fired_ts=1677721632.552529,
context_id=None,
context_id_bin=None,
context_user_id=None,
context_user_id_bin=None,
context_parent_id=None,
context_parent_id_bin=None,
), ),
) )
) )
@@ -759,9 +732,6 @@ async def test_migrate_event_type_ids(
instance.queue_task(EventTypeIDMigrationTask()) instance.queue_task(EventTypeIDMigrationTask())
await async_recorder_block_till_done(hass) await async_recorder_block_till_done(hass)
def _object_as_dict(obj):
return {c.key: getattr(obj, c.key) for c in inspect(obj).mapper.column_attrs}
def _fetch_migrated_events(): def _fetch_migrated_events():
with session_scope(hass=hass) as session: with session_scope(hass=hass) as session:
events = ( events = (