Update unit tests with new queries

Add unit tests for details query and parser.
Add new test case for parsing search output with no data.

Author: Electronic-Mango

test/simplejustwatchapi/test_parser.py

@@ -1,4 +1,6 @@
-from simplejustwatchapi.query import MediaEntry, Offer, parse_search_response, OfferPackage
+from pytest import mark
+
+from simplejustwatchapi.query import MediaEntry, Offer, parse_search_response, OfferPackage, parse_details_response
 
 DETAILS_URL = "https://justwatch.com"
 IMAGES_URL = "https://images.justwatch.com"
@@ -261,7 +263,7 @@
     [],
 )
 
-API_RESPONSE_JSON = {
+API_SEARCH_RESPONSE_JSON = {
     "data": {
         "popularTitles": {
             "edges": [
@@ -273,7 +275,30 @@
     }
 }
 
+API_SEARCH_RESPONSE_NO_DATA = {"data": {"popularTitles": {"edges": []}}}
+
+
+@mark.parametrize(
+    argnames=["response_json", "expected_output"],
+    argvalues=[
+        (API_SEARCH_RESPONSE_JSON, [PARSED_NODE_1, PARSED_NODE_2, PARSED_NODE_3]),
+        (API_SEARCH_RESPONSE_NO_DATA, [])
+    ]
+)
+def test_parse_search_response(response_json, expected_output) -> None:
+    parsed_entries = parse_search_response(response_json)
+    assert parsed_entries == expected_output
+
 
-def test_parse() -> None:
-    parsed_entries = parse_search_response(API_RESPONSE_JSON)
-    assert [PARSED_NODE_1, PARSED_NODE_2, PARSED_NODE_3] == parsed_entries
+@mark.parametrize(
+    argnames=["response_json", "expected_output"],
+    argvalues=[
+        ({"data": {"node": RESPONSE_NODE_1}}, PARSED_NODE_1),
+        ({"data": {"node": RESPONSE_NODE_2}}, PARSED_NODE_2),
+        ({"data": {"node": RESPONSE_NODE_3}}, PARSED_NODE_3),
+        ({"errors": [], "data": {"node": None}}, None)
+    ]
+)
+def test_parse_details_response(response_json, expected_output) -> None:
+    parsed_entries = parse_details_response(response_json)
+    assert parsed_entries == expected_output

test/simplejustwatchapi/test_request.py

@@ -1,6 +1,6 @@
 from pytest import mark, raises
 
-from simplejustwatchapi.query import prepare_search_request
+from simplejustwatchapi.query import prepare_search_request, prepare_details_request
 
 GRAPHQL_DETAILS_QUERY = """
 query GetTitleNode(
@@ -161,3 +161,46 @@ def test_prepare_search_request_asserts_on_invalid_country_code(invalid_code: st
     with raises(AssertionError) as error:
         prepare_search_request("", invalid_code, "", 1, True)
         assert str(error.value) == expected_error_message
+
+
+@mark.parametrize(
+    argnames=["node_id", "country", "language", "best_only"],
+    argvalues=[
+        ("NODE ID 1", "US", "language 1", True),
+        ("NODE ID 1", "gb", "language 2", False),
+    ],
+)
+def test_prepare_details_request(
+    node_id: str, country: str, language: str, best_only: bool
+) -> None:
+    expected_request = {
+        "operationName": "GetTitleNode",
+        "variables": {
+            "nodeId": node_id,
+            "language": language,
+            "country": country.upper(),
+            "formatPoster": "JPG",
+            "formatOfferIcon": "PNG",
+            "profile": "S718",
+            "backdropProfile": "S1920",
+            "filter": {"bestOnly": best_only},
+        },
+        "query": GRAPHQL_DETAILS_QUERY + GRAPHQL_DETAILS_FRAGMENT + GRAPHQL_OFFER_FRAGMENT,
+    }
+    request = prepare_details_request(node_id, country, language, best_only)
+    assert expected_request == request
+
+
+@mark.parametrize(
+    argnames=["invalid_code"],
+    argvalues=[
+        ("United Stated of America",),  # too long
+        ("usa",),  # too long
+        ("u",),  # too short
+    ],
+)
+def test_prepare_details_request_asserts_on_invalid_country_code(invalid_code: str) -> None:
+    expected_error_message = f"Invalid country code: {invalid_code}, code must be 2 characters long"
+    with raises(AssertionError) as error:
+        prepare_details_request("", invalid_code, "", True)
+        assert str(error.value) == expected_error_message