Handle partial V02 authentication failures

Co-authored-by: Copilot App <223556219+Copilot@users.noreply.github.com>

Author: hugo-brito

custom_components/bestway/aws_iot/api.py

@@ -684,7 +684,7 @@ async def fetch_data(self) -> Any:  # Returns BestwayApiResults
 
         refreshed, auth_failed = await self._poll_all_devices()
 
-        if self.devices and refreshed == 0 and auth_failed:
+        if self.devices and auth_failed:
             _LOGGER.info("Re-authenticating after auth failure during poll")
             try:
                 token = await self.authenticate(

tests/test_aws_iot_api.py

@@ -374,6 +374,33 @@ async def test_fetch_data_reauthenticates_and_propagates_token(aws_api):
     token_updated.assert_called_once_with("fresh_token")
 
 
+@pytest.mark.asyncio
+async def test_fetch_data_reauthenticates_after_partial_auth_failure(aws_api):
+    """Any auth rejection refreshes the shared account token."""
+    aws_api.devices = {
+        "device1": _make_aws_device("device1"),
+        "device2": _make_aws_device("device2"),
+    }
+    shadow = {"code": 0, "data": {"state": {"reported": {"power_state": 1}}}}
+    aws_api._do_post = AsyncMock(
+        side_effect=[
+            shadow,
+            AwsIotAuthException("expired"),
+            shadow,
+            shadow,
+        ]
+    )
+
+    with patch.object(
+        AwsIotApi, "authenticate", new=AsyncMock(return_value="fresh_token")
+    ) as authenticate:
+        results = await aws_api.fetch_data()
+
+    authenticate.assert_awaited_once()
+    assert set(results.devices) == {"device1", "device2"}
+    assert aws_api._do_post.await_count == 4
+
+
 @pytest.mark.asyncio
 async def test_fetch_data_raises_auth_failed_when_reauth_rejected(aws_api):
     """A rejected runtime reauth starts Home Assistant's reauth handling."""