Add guard against non-200 HTTP responses (#39)

aiotankerkoenig/aiotankerkoenig.py

@@ -4,13 +4,18 @@
 import asyncio
 from dataclasses import dataclass
 from importlib import metadata
+import socket
 from typing import Any, Self
 
-from aiohttp import ClientSession
+from aiohttp import ClientError, ClientSession
 from yarl import URL
 
 from .const import GasType, Sort
-from .exceptions import TankerkoenigConnectionTimeoutError, TankerkoenigError
+from .exceptions import (
+    TankerkoenigConnectionError,
+    TankerkoenigConnectionTimeoutError,
+    TankerkoenigError,
+)
 from .models import (
     PriceInfo,
     PriceInfoResponse,
@@ -66,6 +71,12 @@ async def _request(self, path: str, params: dict[str, Any]) -> str:
             raise TankerkoenigConnectionTimeoutError(
                 msg,
             ) from exception
+        except (
+            ClientError,
+            socket.gaierror,
+        ) as exception:
+            msg = "Error occurred while communicating with the tankerkoenig.de API"
+            raise TankerkoenigConnectionError(msg) from exception
 
         content_type = response.headers.get("Content-Type", "")
         text = await response.text()

tests/test_aiotankerkoenig.py

@@ -11,6 +11,7 @@
     GasType,
     Sort,
     Tankerkoenig,
+    TankerkoenigConnectionError,
     TankerkoenigConnectionTimeoutError,
     TankerkoenigError,
     TankerkoenigInvalidKeyError,
@@ -57,6 +58,21 @@ async def test_unexpected_server_response(
         await tankerkoenig_client.station_details(station_id="1")
 
 
+async def test_non_200_response(
+    responses: aioresponses,
+    tankerkoenig_client: Tankerkoenig,
+) -> None:
+    """Test handling unexpected response."""
+    responses.get(
+        f"{TANKERKOENIG_ENDPOINT}/json/detail.php?apikey=abc123&id=1",
+        status=500,
+        headers={"Content-Type": "plain/text"},
+        body="Yes",
+    )
+    with pytest.raises(TankerkoenigConnectionError):
+        await tankerkoenig_client.station_details(station_id="1")
+
+
 @pytest.mark.parametrize(
     "message",
     [