mirror of
https://github.com/fusion44/blitz_api.git
synced 2026-08-20 12:57:23 +02:00
26 lines
956 B
Python
26 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
|