jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-05-02 21:19:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Save migrated store data (#82523)

Browse Source
This commit is contained in:
Erik Montnemery 2022-11-28 15:42:08 +01:00 committed by GitHub
parent 69d519418e
commit effa9940ff
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
homeassistant/helpers/storage.py
View File

@ -177,6 +177,7 @@ class Store(Generic[_T]):
if data["version"] != self.version: if data["version"] != self.version:
raise raise
stored = data["data"] stored = data["data"]
await self.async_save(stored)
return stored return stored

10
tests/helpers/test_storage.py
View File

@ -394,13 +394,13 @@ async def test_migration(hass, hass_storage, store_v_1_2):
} }
assert calls == 0 assert calls == 0
legacy_store = CustomStore(hass, 2, store_v_1_2.key, minor_version=1) custom_store = CustomStore(hass, 2, store_v_1_2.key, minor_version=1)
data = await legacy_store.async_load() data = await custom_store.async_load()
assert calls == 1 assert calls == 1
assert hass_storage[store_v_1_2.key]["data"] == data assert hass_storage[store_v_1_2.key]["data"] == data
await legacy_store.async_save(MOCK_DATA) # Assert the migrated data has been saved
assert hass_storage[legacy_store.key] == { assert hass_storage[custom_store.key] == {
"key": MOCK_KEY, "key": MOCK_KEY,
"version": 2, "version": 2,
"minor_version": 1, "minor_version": 1,
@ -433,7 +433,7 @@ async def test_legacy_migration(hass, hass_storage, store_v_1_2):
assert calls == 1 assert calls == 1
assert hass_storage[store_v_1_2.key]["data"] == data assert hass_storage[store_v_1_2.key]["data"] == data
await legacy_store.async_save(MOCK_DATA) # Assert the migrated data has been saved
assert hass_storage[legacy_store.key] == { assert hass_storage[legacy_store.key] == {
"key": MOCK_KEY, "key": MOCK_KEY,
"version": 2, "version": 2,