404 test

Author: Kilo59

tests/test_routes.py

@@ -120,21 +120,23 @@ def tearDown(self):
 
 
 @pytest.mark.parametrize(
-    "query_params",
+    "query_params,expected_status",
     [
-        {"source": "csbs"},
-        {"source": "jhu"},
-        {"timelines": True},
-        {"timelines": "true"},
-        {"timelines": 1},
-        {"source": "jhu", "timelines": True},
+        ({"source": "csbs"}, 200),
+        ({"source": "jhu"}, 200),
+        ({"timelines": True}, 200),
+        ({"timelines": "true"}, 200),
+        ({"timelines": 1}, 200),
+        ({"source": "jhu", "timelines": True}, 200),
+        ({"source": "csbs", "country_code": "US"}, 200),
+        ({"source": "jhu", "country_code": "US"}, 404)
     ],
 )
-def test_locations_status_code(api_client, query_params):
+def test_locations_status_code(api_client, query_params, expected_status):
     response = api_client.get("/v2/locations", params=query_params)
     print(f"GET {response.url}\n{response}")
-    assert response.status_code == 200
-
+    print(f"\tjson:\n{pf(response.json())[:1000]}\n\t...")
+    assert response.status_code == expected_status
 
 @pytest.mark.parametrize(
     "query_params",