Mock timeout scenario for default_location

src/api.py

@@ -50,7 +50,11 @@ def default_location():
     If no location specified in cli, find user's location
     Make a GET request to the API endpoint
     """
-    response = requests.get("https://ipinfo.io/json", timeout=10)
+    try:
+        response = requests.get("https://ipinfo.io/json", timeout=10)
+        response.raise_for_status()
+    except requests.exceptions.Timeout:
+        return "No data"
 
     if response.status_code == HTTPStatus.OK:
         data = response.json()

tests/test_api.py

@@ -9,6 +9,7 @@
 
 import pytest
 from openmeteo_requests.Client import OpenMeteoRequestsError
+from requests.exceptions import Timeout
 
 from src.api import (
     default_location,
@@ -33,6 +34,7 @@
             ["43.03", "-72.001", "New York"],
         ),
         (HTTPStatus.BAD_REQUEST, {}, "No data"),
+        (Timeout, {}, "No data"),
     ],
 )
 def test_default_location_mocked(
@@ -46,6 +48,10 @@ def test_default_location_mocked(
     # Mock the 'requests.get' method
     mock_requests = mocker.patch("requests.get", return_value=mock_response)
 
+    # Check for Timeout and set side effect
+    if status_code == Timeout:
+        mock_requests.side_effect = Timeout("Test Timeout")
+
     # Act: Call the function
     result = default_location()