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 halfapi.app import app #os.environ['HALFAPI_PROD'] = 'True' os.environ['HALFAPI_SECRET'] = 'randomsecret' from halfapi.lib.jwt_middleware import (PRODUCTION, SECRET, JWTUser, JWTAuthenticationBackend, JWTWebSocketAuthenticationBackend) def test_constants(): assert isinstance(PRODUCTION, bool) assert isinstance(SECRET, str) @pytest.fixture def token(): # This fixture needs to have a running auth-lirmm on 127.0.0.1:3000 # Sets a valid token r = requests.post('http://127.0.0.1:3000/', data={'email':'maizi', 'password':'a'}) if len(r.text) <= 0: raise Exception('No result in token retrieval') try: res = json.loads(r.text) except JSONDecodeError: raise Exception('Malformed response from token retrieval') if 'token' not in res.keys(): raise Exception('Missing token in token request') return res['token'] @pytest.fixture def token_builder(): yield jwt.encode({ 'name':'xxx', 'user_id': str(uuid4())}, key=SECRET ) @pytest.fixture def token_debug_false_builder(): yield jwt.encode({ 'name':'xxx', 'user_id': str(uuid4()), 'debug': False}, key=SECRET ) @pytest.fixture def token_debug_true_builder(): yield jwt.encode({ 'name':'xxx', 'user_id': str(uuid4()), 'debug': True}, key=SECRET ) @pytest.fixture def token_dirser(): # This fixture needs to have a running auth-lirmm on 127.0.0.1:3000 # Sets a valid token r = requests.post('http://127.0.0.1:3000/', data={'email':'dhenaut', 'password':'a'}) if len(r.text) <= 0: raise Exception('No result in token retrieval') try: res = json.loads(r.text) except JSONDecodeError: raise Exception('Malformed response from token retrieval') if 'token' not in res.keys(): raise Exception('Missing token in token request') return res['token'] def test_JWTUser(): uid = uuid4() token = '{}' payload = {} user = JWTUser(uid, token, payload) assert user.identity == uid assert user.token == token assert user.payload == payload assert user.is_authenticated == True @pytest.mark.asyncio async def test_JWTAuthenticationBackend(token_builder): backend = JWTAuthenticationBackend() assert backend.secret_key == SECRET req = Request( headers={ 'Authorization': token_builder }) credentials, user = await backend.authenticate(req) assert type(user) == JWTUser assert type(credentials) == AuthCredentials @pytest.mark.asyncio async def test_JWTAuthenticationBackend_DebugFalse(token_debug_false_builder): backend = JWTAuthenticationBackend() assert backend.secret_key == SECRET req = Request( headers={ 'Authorization': token_debug_false_builder }) credentials, user = await backend.authenticate(req) assert type(user) == JWTUser assert type(credentials) == AuthCredentials @pytest.mark.asyncio async def test_JWTAuthenticationBackend_DebugTrue(token_debug_true_builder): backend = JWTAuthenticationBackend() assert backend.secret_key == SECRET req = Request( headers={ 'Authorization': token_debug_true_builder }) try: await backend.authenticate(req) except Exception as exc: assert type(exc) == AuthenticationError