wrap httpx.TimeoutException (#12)

This allows to catch it via the generic WRCError or with the new WRCTimeoutError

src/pywaze/route_calculator.py

@@ -9,6 +9,10 @@ class WRCError(Exception):
     """Waze Route Calculator Error."""
 
 
+class WRCTimeoutError(WRCError):
+    """Waze Route Calculator Timeout Error."""
+
+
 class WazeRouteCalculator:
     """Calculate actual route time and distance with Waze API."""
 
@@ -85,9 +89,12 @@ async def address_to_coords(self, address: str) -> dict[str, Any]:
             "lon": base_coords["lon"],
         }
 
-        response: httpx.Response = await self.client.get(
-            self.WAZE_URL + get_cord, params=url_options, headers=self.HEADERS
-        )
+        try:
+            response: httpx.Response = await self.client.get(
+                self.WAZE_URL + get_cord, params=url_options, headers=self.HEADERS
+            )
+        except httpx.TimeoutException as e:
+            raise WRCTimeoutError("Timeout getting coords for %s" % address) from e
         for response_json in response.json():
             if response_json.get("city"):
                 lat = response_json["location"]["lat"]
@@ -145,9 +152,12 @@ async def get_route(
         if avoid_subscription_roads is False:
             url_options["subscription"] = "*"
 
-        response: httpx.Response = await self.client.get(
-            self.WAZE_URL + routing_server, params=url_options, headers=self.HEADERS
-        )
+        try:
+            response: httpx.Response = await self.client.get(
+                self.WAZE_URL + routing_server, params=url_options, headers=self.HEADERS
+            )
+        except httpx.TimeoutException as e:
+            raise WRCTimeoutError("Timeout getting route") from e
         response_json = self._check_response(response)
         if response_json:
             if "error" in response_json:

tests/conftest.py

@@ -2,6 +2,7 @@
 
 from typing import Any
 from httpx import Response
+import httpx
 import pytest
 from respx import MockRouter
 
@@ -22,6 +23,14 @@ def get_route_mock(get_route_response: dict[str, Any], respx_mock: MockRouter):
     )
 
 
+@pytest.fixture
+def timeout_mock(respx_mock: MockRouter):
+    """Throw a httpx.TimeoutException when calculating routes."""
+    yield respx_mock.get("https://www.waze.com/row-RoutingManager/routingRequest").mock(
+        side_effect=httpx.TimeoutException("Timeout")
+    )
+
+
 @pytest.fixture()
 def wiesbaden_to_coords_mock(respx_mock: MockRouter):
     """Return the provided json response when converting this address to coordinates."""

tests/test_route_calculator.py

@@ -92,3 +92,14 @@ async def test_calc_all_routes_info(
         route_time, route_distance = list(results.values())[0]
         assert route_time == expected_route_time
         assert route_distance == expected_route_distance
+
+
+async def test_calc_route_info_timeout(timeout_mock):
+    """Test calc_route_info with timeout."""
+
+    async with route_calculator.WazeRouteCalculator() as client:
+        with pytest.raises(route_calculator.WRCTimeoutError):
+            await client.calc_route_info(
+                "50.00332659227126,8.262322651915843",
+                "50.08414976707619,8.247836017342934",
+            )