Change error again

agittins · Nov 11, 2024 · 7b746c9 · 7b746c9
1 parent b8acc4a
commit 7b746c9
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/custom_components/bermuda/__init__.py b/custom_components/bermuda/__init__.py
@@ -41,13 +41,19 @@ async def async_setup_entry(hass: HomeAssistant, entry: BermudaConfigEntry):
     coordinator = BermudaDataUpdateCoordinator(hass, entry)
     entry.runtime_data = BermudaData(coordinator)
 
-    await coordinator.async_refresh()
-
-    if not coordinator.last_update_success:
+    async def on_failure():
         _LOGGER.debug("Coordinator last update failed, rasing ConfigEntryNotReady")
         await coordinator.stop_purging()
         raise ConfigEntryNotReady
 
+    try:
+        await coordinator.async_refresh()
+    except Exception as ex:  # noqa: BLE001
+        _LOGGER.exception(ex)
+        await on_failure()
+    if not coordinator.last_update_success:
+        await on_failure()
+
     await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
 
     entry.async_on_unload(entry.add_update_listener(async_reload_entry))

diff --git a/tests/conftest.py b/tests/conftest.py
@@ -70,7 +70,7 @@ def error_get_data_fixture():
     """Simulate error when retrieving data from API."""
     with patch(
         "custom_components.bermuda.BermudaDataUpdateCoordinator.async_refresh",
-        side_effect=UpdateFailed,
+        side_effect=Exception,
     ):
         yield