feedc0de/homeassistant-core
1
0
Fork 0
forked from home-assistant/core
Code Pull Requests Activity

Tweak

Browse Source
This commit is contained in:
Erik
2025-04-30 15:51:23 +02:00
parent 71af693569
commit 4d5e809e9b
1 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
tests/common.py
View File

@@ -642,7 +642,8 @@ def mock_registry(
registry.entities[key] = entry
hass.data[er.DATA_REGISTRY] = registry
er.async_get.cache_clear()
with suppress(AttributeError):
er.async_get.cache_clear()
return registry
@@ -694,7 +695,8 @@ def mock_area_registry(
registry.areas[key] = entry
hass.data[ar.DATA_REGISTRY] = registry
ar.async_get.cache_clear()
with suppress(AttributeError):
ar.async_get.cache_clear()
return registry
@@ -723,7 +725,8 @@ def mock_device_registry(
registry.deleted_devices = dr.DeviceRegistryItems()
hass.data[dr.DATA_REGISTRY] = registry
dr.async_get.cache_clear()
with suppress(AttributeError):
dr.async_get.cache_clear()
return registry
@@ -1307,7 +1310,8 @@ def mock_restore_cache(hass: HomeAssistant, states: Sequence[State]) -> None:
_LOGGER.debug("Restore cache: %s", data.last_states)
assert len(data.last_states) == len(states), f"Duplicate entity_id? {states}"
rs.async_get.cache_clear()
with suppress(AttributeError):
rs.async_get.cache_clear()
hass.data[key] = data
@@ -1335,7 +1339,8 @@ def mock_restore_cache_with_extra_data(
_LOGGER.debug("Restore cache: %s", data.last_states)
assert len(data.last_states) == len(states), f"Duplicate entity_id? {states}"
rs.async_get.cache_clear()
with suppress(AttributeError):
rs.async_get.cache_clear()
hass.data[key] = data