Fix SharkIQ token expiration

homeassistant/components/sharkiq/update_coordinator.py

@@ -60,6 +60,9 @@ async def _async_update_vacuum(sharkiq: SharkIqVacuum) -> None:
     async def _async_update_data(self) -> bool:
         """Update data device by device."""
         try:
+            if self.ayla_api.token_expiring_soon:
+                await self.ayla_api.async_refresh_auth()
+
             all_vacuums = await self.ayla_api.async_list_devices()
             self._online_dsns = {
                 v["dsn"]
@@ -76,9 +79,11 @@ async def _async_update_data(self) -> bool:
             SharkIqAuthExpiringError,
         ) as err:
             LOGGER.debug("Bad auth state.  Attempting re-auth", exc_info=err)
+            await self.ayla_api.async_sign_in()
             raise ConfigEntryAuthFailed from err
         except Exception as err:
-            LOGGER.exception("Unexpected error updating SharkIQ")
+            LOGGER.exception("Unexpected error updating SharkIQ.  Attempting re-auth")
+            await self.ayla_api.async_sign_in()
             raise UpdateFailed(err) from err
 
         return True

tests/components/sharkiq/test_vacuum.py

@@ -75,6 +75,9 @@ class MockAyla(AylaApi):
     async def async_sign_in(self):
         """Instead of signing in, just return."""
 
+    async def async_refresh_auth(self):
+        """Instead of refreshing auth, just return."""
+
     async def async_list_devices(self) -> list[dict]:
         """Return the device list."""
         return [SHARK_DEVICE_DICT]