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

Migrate restore_state to use the singleton helper (#117385)

Browse Source
This commit is contained in:
J. Nick Koston 2024-05-14 09:13:44 +09:00 committed by GitHub
parent 728e1a2223
commit 9381462877
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
homeassistant/helpers/restore_state.py
View File

@ -19,6 +19,7 @@ from .entity import Entity
from .event import async_track_time_interval from .event import async_track_time_interval
from .frame import report from .frame import report
from .json import JSONEncoder from .json import JSONEncoder
from .singleton import singleton
from .storage import Store from .storage import Store
DATA_RESTORE_STATE: HassKey[RestoreStateData] = HassKey("restore_state") DATA_RESTORE_STATE: HassKey[RestoreStateData] = HassKey("restore_state")
@ -97,15 +98,14 @@ class StoredState:
async def async_load(hass: HomeAssistant) -> None: async def async_load(hass: HomeAssistant) -> None:
"""Load the restore state task.""" """Load the restore state task."""
restore_state = RestoreStateData(hass) await async_get(hass).async_setup()
await restore_state.async_setup()
hass.data[DATA_RESTORE_STATE] = restore_state
@callback @callback
@singleton(DATA_RESTORE_STATE)
def async_get(hass: HomeAssistant) -> RestoreStateData: def async_get(hass: HomeAssistant) -> RestoreStateData:
"""Get the restore state data helper.""" """Get the restore state data helper."""
return hass.data[DATA_RESTORE_STATE] return RestoreStateData(hass)
class RestoreStateData: class RestoreStateData:

2
tests/common.py
View File

@ -1175,6 +1175,7 @@ def mock_restore_cache(hass: HomeAssistant, states: Sequence[State]) -> None:
_LOGGER.debug("Restore cache: %s", data.last_states) _LOGGER.debug("Restore cache: %s", data.last_states)
assert len(data.last_states) == len(states), f"Duplicate entity_id? {states}" assert len(data.last_states) == len(states), f"Duplicate entity_id? {states}"
restore_state.async_get.cache_clear()
hass.data[key] = data hass.data[key] = data
@ -1202,6 +1203,7 @@ def mock_restore_cache_with_extra_data(
_LOGGER.debug("Restore cache: %s", data.last_states) _LOGGER.debug("Restore cache: %s", data.last_states)
assert len(data.last_states) == len(states), f"Duplicate entity_id? {states}" assert len(data.last_states) == len(states), f"Duplicate entity_id? {states}"
restore_state.async_get.cache_clear()
hass.data[key] = data hass.data[key] = data