4f4dac0ff2
Breaking : migrate your tests that use the TestDomain.client method following the instructions here https://github.com/Kludex/bump-testclient Squashed commit of the following: commit0417f27b3fAuthor: Maxime Alves LIRMM <maxime.alves@lirmm.fr> Date: Sat Jan 14 11:08:44 2023 +0100 [deps] starlette 0.23 commit552f00a65bAuthor: Maxime Alves LIRMM <maxime.alves@lirmm.fr> Date: Sat Jan 14 10:59:42 2023 +0100 [deps] starlette 0.22 commitaefe448717Author: Maxime Alves LIRMM <maxime.alves@lirmm.fr> Date: Sat Jan 14 10:55:45 2023 +0100 [tests][fix] compares the json interpreted value instead of the string commit01333a200cAuthor: Maxime Alves LIRMM <maxime.alves@lirmm.fr> Date: Sat Jan 14 10:55:20 2023 +0100 [testing] changes from requests to httpx for Starlette TestClient (breaks) commitf3784fab7fAuthor: Maxime Alves LIRMM <maxime.alves@lirmm.fr> Date: Sat Jan 14 10:54:10 2023 +0100 [deps][breaking] starlette 0.21 commit717d3f8bd6Author: Maxime Alves LIRMM <maxime.alves@lirmm.fr> Date: Sat Jan 14 10:26:31 2023 +0100 [responses] use a wrapper function for exception handling (fix starlette 0.20) commitd0876e45daAuthor: Maxime Alves LIRMM <maxime.alves@lirmm.fr> Date: Sat Jan 14 10:25:21 2023 +0100 [deps][breaking] starlette 0.20 commit6504191c53Author: Maxime Alves LIRMM <maxime.alves@lirmm.fr> Date: Sat Jan 14 10:12:51 2023 +0100 [deps] starlette 0.19 commit7b639a8dc2Author: Maxime Alves LIRMM <maxime.alves@lirmm.fr> Date: Sat Jan 14 10:11:14 2023 +0100 [deps] starlette 0.18 commit20bd9077a4Author: Maxime Alves LIRMM <maxime.alves@lirmm.fr> Date: Sat Jan 14 10:07:48 2023 +0100 pipenv update
108 lines
3.0 KiB
Python
108 lines
3.0 KiB
Python
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.request('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.request('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.request('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.request('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.request('get', '/test',
|
|
headers={
|
|
'Authorization': token_debug_true_builder
|
|
})
|
|
assert resp.status_code == 200
|