Create test_dependencies.py

Author: Kanchan-Microsoft

src/ContentProcessorAPI/app/tests/test_dependencies.py

@@ -0,0 +1,36 @@
+import pytest
+from fastapi import FastAPI, Depends
+from fastapi.testclient import TestClient
+from src.ContentProcessorAPI.app.dependencies import get_token_header, get_query_token
+# from starlette.status import HTTP_400_BAD_REQUEST
+
+
+@pytest.fixture
+def test_app():
+    app = FastAPI()
+
+    @app.get("/header-protected")
+    async def protected_route_header(dep=Depends(get_token_header)):
+        return {"message": "Success"}
+
+    @app.get("/query-protected")
+    async def protected_route_query(dep=Depends(get_query_token)):
+        return {"message": "Success"}
+
+    return app
+
+
+def test_get_token_header_fails(test_app):
+    client = TestClient(test_app)
+    # Provide the required header so FastAPI doesn't return 422
+    response = client.get("/header-protected", headers={"x-token": "fake"})
+    assert response.status_code == 400
+    assert response.json() == {"detail": "X-Token header invalid"}
+
+
+def test_get_query_token_fails(test_app):
+    client = TestClient(test_app)
+    # Provide the required query param so FastAPI doesn't return 422
+    response = client.get("/query-protected?token=fake")
+    assert response.status_code == 400
+    assert response.json() == {"detail": "No ... token provided"}