evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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:
raise
stored = data["data"]
await self.async_save(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
legacy_store = CustomStore(hass, 2, store_v_1_2.key, minor_version=1)
data = await legacy_store.async_load()
custom_store = CustomStore(hass, 2, store_v_1_2.key, minor_version=1)
data = await custom_store.async_load()
assert calls == 1
assert hass_storage[store_v_1_2.key]["data"] == data
await legacy_store.async_save(MOCK_DATA)
assert hass_storage[legacy_store.key] == {
# Assert the migrated data has been saved
assert hass_storage[custom_store.key] == {
"key": MOCK_KEY,
"version": 2,
"minor_version": 1,
@ -433,7 +433,7 @@ async def test_legacy_migration(hass, hass_storage, store_v_1_2):
assert calls == 1
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] == {
"key": MOCK_KEY,
"version": 2,