Fix API connection error handling (#13)

Fixes #12
Fixes home-assistant/core/issues/61687

Author: Cougar

luftdaten/__init__.py

@@ -29,6 +29,10 @@ async def get_data(self):
                 response = await client.get(str(url))
         except httpx.ConnectError:
             raise exceptions.LuftdatenConnectionError(f"Connection to {url} failed")
+        except httpx.ConnectTimeout:
+            raise exceptions.LuftdatenConnectionError(f"Connection to {url} timed out")
+        except httpx.ReadTimeout:
+            raise exceptions.LuftdatenConnectionError(f"Read timeout from {url}")
 
         if response.status_code == httpx.codes.OK:
             try:

tests/test_timeout.py

@@ -4,12 +4,29 @@
 from pytest_httpx import HTTPXMock
 
 from luftdaten import Luftdaten
+from luftdaten.exceptions import LuftdatenConnectionError
 
 SENSOR_ID = 1
 
 @pytest.mark.asyncio
-async def test_timeout(httpx_mock: HTTPXMock):
-    """Test if the connection is hitting the timeout."""
+async def test_connect_timeout(httpx_mock: HTTPXMock):
+    """Test if the connection is hitting the timeout during connect."""
+
+    def raise_timeout(request):
+        """Set the timeout for the requests."""
+        raise httpx.ConnectTimeout(
+            f"Unable to connect within {request.extensions['timeout']}", request=request
+        )
+
+    httpx_mock.add_callback(raise_timeout)
+
+    with pytest.raises(LuftdatenConnectionError):
+        client = Luftdaten(SENSOR_ID)
+        await client.get_data()
+
+@pytest.mark.asyncio
+async def test_read_timeout(httpx_mock: HTTPXMock):
+    """Test if the connection is hitting the timeout during data reading."""
 
     def raise_timeout(request):
         """Set the timeout for the requests."""
@@ -19,6 +36,6 @@ def raise_timeout(request):
 
     httpx_mock.add_callback(raise_timeout)
 
-    with pytest.raises(httpx.ReadTimeout):
+    with pytest.raises(LuftdatenConnectionError):
         client = Luftdaten(SENSOR_ID)
         await client.get_data()