fix: fix test for resetting login for relogin

closes #756
alandtse · May 31, 2020 · 2388345 · 2388345
1 parent 7095691
commit 2388345
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 8 deletions.
diff --git a/custom_components/alexa_media/__init__.py b/custom_components/alexa_media/__init__.py
@@ -313,14 +313,13 @@ async def async_update_data():
                 hide_email(email),
                 login_obj.status,
             )
-            if login_obj.status and not await test_login_status(
-                hass, config_entry, login_obj, setup_alexa
-            ):
-                login_obj.status = {}
+            if login_obj.status:
                 await login_obj.reset()
                 await login_obj.login()
+                await test_login_status(hass, config_entry, login_obj, setup_alexa)
             return
         except BaseException as err:
+            raise
             raise UpdateFailed(f"Error communicating with API: {err}")
 
         await process_notifications(login_obj, raw_notifications)

diff --git a/custom_components/alexa_media/helpers.py b/custom_components/alexa_media/helpers.py
@@ -180,12 +180,10 @@ async def wrapper(*args, **kwargs) -> Any:
                         "%s: Alexa API disconnected; attempting to relogin",
                         hide_email(email),
                     )
-                    if login.status and not await test_login_status(
-                        hass, config_entry, login, setup_alexa
-                    ):
-                        login.status = {}
+                    if login.status:
                         await login.reset()
                         await login.login()
+                        await test_login_status(hass, config_entry, login, setup_alexa)
             return None
         return result