import os import jwt from requests import Request import pytest from unittest.mock import patch import json from json.decoder import JSONDecodeError import sys from hashlib import sha256 from base64 import b64decode from uuid import uuid4, UUID from starlette.testclient import TestClient from starlette.authentication import ( AuthenticationBackend, AuthenticationError, BaseUser, AuthCredentials, UnauthenticatedUser) from starlette.responses import PlainTextResponse from halfapi.lib.jwt_middleware import ( JWTUser, JWTAuthenticationBackend, JWTWebSocketAuthenticationBackend) def test_JWTUser(): uid = uuid4() token = '{}' payload = {} user = JWTUser(uid, token, payload) assert user.id == uid assert user.token == token assert user.payload == payload assert user.is_authenticated == True def test_jwt_NoToken(dummy_app): async def test_route(request): assert isinstance(request.user, UnauthenticatedUser) return PlainTextResponse('ok') dummy_app.add_route('/test', test_route) test_client = TestClient(dummy_app) resp = test_client.get('/test') assert resp.status_code == 200 def test_jwt_Token(dummy_app, token_builder): async def test_route(request): assert isinstance(request.user, JWTUser) print(request.scope['app'].debug) return PlainTextResponse('ok') dummy_app.add_route('/test', test_route) test_client = TestClient(dummy_app) resp = test_client.get('/test', headers={ 'Authorization': token_builder }) assert resp.status_code == 200 def test_jwt_DebugFalse(dummy_app, token_debug_false_builder): async def test_route(request): assert isinstance(request.user, JWTUser) return PlainTextResponse('ok') dummy_app.add_route('/test', test_route) test_client = TestClient(dummy_app) resp = test_client.get('/test', headers={ 'Authorization': token_debug_false_builder }) assert resp.status_code == 200 def test_jwt_DebugTrue(dummy_app, token_debug_true_builder): """ A debug token should return a 400 status code with a non debug app """ async def test_route(request): return PlainTextResponse('ok') dummy_app.add_route('/test', test_route) test_client = TestClient(dummy_app) resp = test_client.get('/test', headers={ 'Authorization': token_debug_true_builder }) assert resp.status_code == 400 def test_jwt_DebugTrue_DebugApp(dummy_debug_app, token_debug_true_builder): """ A debug token should return a 200 status code with a debug app """ async def test_route(request): assert isinstance(request.user, JWTUser) return PlainTextResponse('ok') dummy_debug_app.add_route('/test', test_route) test_client = TestClient(dummy_debug_app) resp = test_client.get('/test', headers={ 'Authorization': token_debug_true_builder }) assert resp.status_code == 200