Expand example to demonstrate refresh token works

Author: erwindouna

examples/example.py

@@ -1,5 +1,7 @@
 """Asynchronous Python client for the Tado API. This is an example file."""
 
+from __future__ import annotations
+
 import asyncio
 import logging
 
@@ -10,13 +12,26 @@
 
 async def main() -> None:
     """Show example on how to use aiohttp.ClientSession."""
+    refresh_token: str | None = None
     async with Tado(debug=True) as tado:
         print("Device activation status: ", tado.device_activation_status)  # noqa: T201
         print("Device verification URL: ", tado.device_verification_url)  # noqa: T201
 
         print("Starting device activation")  # noqa: T201
         await tado.device_activation()
+        refresh_token = tado.refresh_token
+
+        print("Device activation status: ", tado.device_activation_status)  # noqa: T201
+
+        devices = await tado.get_devices()
+
+        print("Devices: ", devices)  # noqa: T201
+
+    print("Trying to use the stored refresh token for another run...")  # noqa: T201
+    await asyncio.sleep(1)
 
+    async with Tado(debug=True, refresh_token=refresh_token) as tado:
+        print("Refresh token: ", tado.refresh_token)  # noqa: T201
         print("Device activation status: ", tado.device_activation_status)  # noqa: T201
 
         devices = await tado.get_devices()

src/tadoasync/tadoasync.py

@@ -123,6 +123,8 @@ async def async_init(self) -> None:
             self._device_activation_status = await self.login_device_flow()
         else:
             self._device_ready()
+            get_me = await self.get_me()
+            self._home_id = get_me.homes[0].id
 
     @property
     def device_activation_status(self) -> DeviceActivationStatus:
@@ -134,6 +136,11 @@ def device_verification_url(self) -> str | None:
         """Return the device verification URL."""
         return self._device_verification_url
 
+    @property
+    def refresh_token(self) -> str | None:
+        """Return the refresh token."""
+        return self._refresh_token
+
     async def login_device_flow(self) -> DeviceActivationStatus:
         """Login using device flow."""
         if self._device_activation_status != DeviceActivationStatus.NOT_STARTED:
@@ -340,17 +347,15 @@ async def _refresh_auth(self) -> None:
         data = {
             "client_id": CLIENT_ID,
             "grant_type": "refresh_token",
-            "scope": "home.user",
             "refresh_token": self._refresh_token,
         }
 
-        if self._session is None:
-            self._session = ClientSession()
-            self._close_session = True
+        _LOGGER.debug("Refreshing Tado token")
 
         try:
             async with asyncio.timeout(self._request_timeout):
-                request = await self._session.post(url=TOKEN_URL, data=data)
+                session = self._ensure_session()
+                request = await session.post(url=TOKEN_URL, data=data)
                 request.raise_for_status()
         except asyncio.TimeoutError as err:
             raise TadoConnectionError(
@@ -364,6 +369,8 @@ async def _refresh_auth(self) -> None:
         self._token_expiry = time.time() + float(response["expires_in"])
         self._refresh_token = response["refresh_token"]
 
+        _LOGGER.debug("Tado token refreshed")
+
     async def get_me(self) -> GetMe:
         """Get the user information."""
         if self._me is None:

tests/test_tado.py

@@ -188,6 +188,27 @@ async def mock_post(*args: Any, **kwargs: Any) -> ClientResponse:  # noqa: ARG00
             await tado.async_init()
 
 
+async def test_login_client_response_still_pending() -> None:
+    """Test login client response error."""
+    mock_request_info = MagicMock(spec=RequestInfo)
+    mock_response = MagicMock(spec=ClientResponse)
+    mock_response.raise_for_status.side_effect = ClientResponseError(
+        mock_request_info, (mock_response,), status=400
+    )
+    mock_response.status = 400
+    mock_response.text = AsyncMock(return_value="authorization_pending")
+
+    async def mock_post(*args: Any, **kwargs: Any) -> ClientResponse:  # noqa: ARG001 # pylint: disable=unused-argument
+        return mock_response
+
+    async with aiohttp.ClientSession() as session:
+        tado = Tado(session=session)
+        with patch("aiohttp.ClientSession.post", new=mock_post), pytest.raises(
+            TadoAuthenticationError
+        ):
+            await tado.async_init()
+
+
 async def test_refresh_auth_success(responses: aioresponses) -> None:
     """Test successful refresh of auth token."""
     responses.post(