From 4d5e809e9ba76db57b9f4ac51cd741f3da27af6c Mon Sep 17 00:00:00 2001 From: Erik Date: Wed, 30 Apr 2025 15:51:23 +0200 Subject: [PATCH] Tweak --- tests/common.py | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/tests/common.py b/tests/common.py index d439021a9df..49ad20c0e2c 100644 --- a/tests/common.py +++ b/tests/common.py @@ -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