Refactor auth test

test_service.py

@@ -72,18 +72,18 @@ def test_similarity(client: TestClient, image1_path, image2_path, exp_score):
         assert response.json()["score"] == pytest.approx(exp_score, abs=1e-4)
 
 
-def test_authentication(client: TestClient):
-    response = client.post(
-        "/embed",
-        files=[("image", ("face_pic.jpeg", b""))],
-    )
-    assert response.status_code == 401
-    assert response.json()["detail"] == "No token provided"
-
+@pytest.mark.parametrize(
+    "token, status_code, detail",
+    [
+        pytest.param(None, 401, "No token provided", id="no token"),
+        pytest.param("wrong_token", 401, "Invalid token", id="wrong token"),
+    ],
+)
+def test_authentication(client: TestClient, token: str, status_code: int, detail: str):
     response = client.post(
         "/embed",
         files=[("image", ("face_pic.jpeg", b""))],
-        params={"token": "wrong_token"},
+        params={"token": token} if token else {},
     )
-    assert response.status_code == 401
-    assert response.json()["detail"] == "Invalid token"
+    assert response.status_code == status_code
+    assert response.json()["detail"] == detail