mirror of
https://github.com/fusion44/blitz_api.git
synced 2026-08-14 12:02:46 +02:00
25 lines
956 B
Python
25 lines
956 B
Python
from fastapi import status
|
|
from starlette.testclient import TestClient
|
|
|
|
invalid_auth_header = {"Authorization": "Bearer dsgfdg"}
|
|
|
|
|
|
def call_route(test_client: TestClient, route: str, params={}, method="g"):
|
|
# Test with no "Authentication" header at all
|
|
# Should return a 403 by FastAPI's HTTPBearer
|
|
response = None
|
|
if method == "g":
|
|
response = test_client.get(route, params=params)
|
|
elif method == "p":
|
|
response = test_client.post(route, params=params)
|
|
assert response.status_code == status.HTTP_403_FORBIDDEN
|
|
|
|
# Test with an invalid "Authentication" header
|
|
# Should return a 401 by BlitzAPI's AuthBearer
|
|
response = None
|
|
if method == "g":
|
|
response = test_client.get(route, params=params, headers=invalid_auth_header)
|
|
elif method == "p":
|
|
response = test_client.post(route, params=params, headers=invalid_auth_header)
|
|
|
|
assert response.status_code == status.HTTP_401_UNAUTHORIZED
|