Created and tested Star Wars character DOB function

Created, type hinted, and tested a Star Wars character DOB function
using the SWAPI api. The function searches for DOB using the name
(or substring of name) and returns the DOB if name is found. I tested
the function including mocking the exceptions and requests.get. I had
to add a pragma to the next call on the generator expression as
coverage though that variable prior to exit immediately. Seems to be a
bug related to this issue
nedbat/coveragepy#515.

Author: codeaucafe

pyproject.toml

@@ -45,7 +45,7 @@ force_grid_wrap = 0
 use_parentheses = true
 line_length = 100
 indent = '    '
-known_third_party = ["pytest"]
+known_third_party = ["pytest", "requests"]
 known_first_party = ["sandbox_pytest"]
 
 [tool.pytest.ini_options]

sandbox_pytest/star_wars_info.py

@@ -0,0 +1,29 @@
+from typing import Any, Dict, List, Union
+
+import requests
+from requests.exceptions import ConnectionError, HTTPError
+
+
+def get_person_dob(name: str) -> Union[None, str]:
+    try:
+        response: requests.models.Response
+        with requests.get("https://swapi.dev/api/people/") as response:
+            response.raise_for_status()
+            resp_dict: Dict[str, Any] = response.json()
+
+        resp_results: List[Dict[str, Any]] = resp_dict["results"]
+        resp_person: Dict[str, Any] = next(  # pragma: no cover
+            item for item in resp_results if name.lower() in item["name"].lower()
+        )
+        dob: str = resp_person["birth_year"]
+    except HTTPError as http_err:
+        raise http_err
+    except ConnectionError as ce:
+        print(ce)
+        raise ce
+    except Exception as err:
+        print(f"Other error occurred: {err}")
+        raise err
+    else:
+        print(f"successfully retrieved the DOB of {name}")
+        return dob

tests/test_sandbox_pytest/test_star_wars_info.py

@@ -0,0 +1,40 @@
+from unittest.mock import patch
+
+import pytest
+
+from sandbox_pytest.star_wars_info import get_person_dob, requests
+
+
+@pytest.mark.parametrize(
+    "arg_names, add_exp_result", [("Luke Skywalker", "19BBY"), ("Darth Vader", "41.9BBY")]
+)
+def test_get_dob(arg_names, add_exp_result) -> None:
+    assert get_person_dob(arg_names) == add_exp_result
+
+
+@patch.object(
+    requests,
+    "get",
+    side_effect=[requests.exceptions.HTTPError, requests.exceptions.ConnectionError, Exception],
+)
+def test_dob_erros(mock_requests) -> None:
+    with pytest.raises(requests.exceptions.HTTPError):
+        get_person_dob("Darth Vader")
+    with pytest.raises(requests.exceptions.ConnectionError):
+        get_person_dob("Darth Vader")
+    with pytest.raises(Exception):
+        get_person_dob("Darth Vader")
+
+
+def test_mock_dob() -> None:
+    with patch.object(requests, "get") as mock_response:
+        mock_response.return_value.__enter__.return_value.json.return_value = {
+            "results": [
+                {"name": "Luke Skywalker", "birth_year": "19BBY"},
+                {"name": "Darth Vader", "birth_year": "41.9BBY"},
+            ]
+        }
+        assert get_person_dob("Luke Skywalker") == "19BBY"
+        mock_response.assert_called_once()
+        assert get_person_dob("Darth Vader") == "41.9BBY"
+        mock_response.assert_called()