Parameterize test_404()

Author: elliotwutingfeng

tests/test_norwegianblue.py

@@ -389,39 +389,29 @@ def test_all_products(self) -> None:
         assert output == expected
 
     @respx.mock
-    def test_404(self) -> None:
-        # Arrange
-        mocked_url = "https://endoflife.date/api/androd.json"
-        respx.get(mocked_url).respond(status_code=404)
-
-        mocked_url = "https://endoflife.date/api/all.json"
-        mocked_response = SAMPLE_RESPONSE_ALL_JSON
-        respx.get(mocked_url).respond(content=mocked_response)
-
-        # Act / Assert
-        with pytest.raises(
-            ValueError,
-            match=r"Product 'androd' not found, run 'eol all' for list\. "
-            r"Did you mean: 'android'?",
-        ):
-            norwegianblue.norwegianblue(product="androd")
-
-    @respx.mock
-    def test_404_no_suggestions(self) -> None:
+    @pytest.mark.parametrize(
+        "product, expected",
+        [
+            (
+                "androd",
+                r"Product 'androd' not found, run 'eol all' for list\. "
+                r"Did you mean: 'android'?",
+            ),
+            ("julia", r"Product 'julia' not found, run 'eol all' for list\."),
+        ],
+    )
+    def test_404(self, product, expected) -> None:
         # Arrange
-        mocked_url = "https://endoflife.date/api/julia.json"
+        mocked_url = f"https://endoflife.date/api/{product}.json"
         respx.get(mocked_url).respond(status_code=404)
 
         mocked_url = "https://endoflife.date/api/all.json"
         mocked_response = SAMPLE_RESPONSE_ALL_JSON
         respx.get(mocked_url).respond(content=mocked_response)
 
         # Act / Assert
-        with pytest.raises(
-            ValueError,
-            match=r"Product 'julia' not found, run 'eol all' for list\.",
-        ):
-            norwegianblue.norwegianblue(product="julia")
+        with pytest.raises(ValueError, match=expected):
+            norwegianblue.norwegianblue(product=product)
 
     def test_norwegianblue_norwegianblue(self) -> None:
         # Act