mirror of
https://github.com/fastapi-users/fastapi-users.git
synced 2025-11-04 14:45:50 +08:00
c4de66b81c
* Implement Transport classes * Implement authentication strategy classes * Revamp authentication with Transport and Strategy * Revamp strategy and OAuth so that they can use a callable dependency * Update docstring * Make ErrorCode a proper Enum and cleanup unused OpenAPI utils * Remove useless check * Tweak typing in authenticator * Update docs * Improve logout/destroy token logic * Update docs * Update docs * Update docs and full examples * Apply formatting to examples * Update OAuth doc and examples * Add migration doc * Implement Redis session token * Add Redis Session documentation * RedisSession -> Redis * Fix links in docs
48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
import pytest
|
|
from fastapi import Response, status
|
|
|
|
from fastapi_users.authentication.transport import (
|
|
BearerTransport,
|
|
TransportLogoutNotSupportedError,
|
|
)
|
|
from fastapi_users.authentication.transport.bearer import BearerResponse
|
|
|
|
|
|
@pytest.fixture()
|
|
def bearer_transport() -> BearerTransport:
|
|
return BearerTransport(tokenUrl="/login")
|
|
|
|
|
|
@pytest.mark.authentication
|
|
@pytest.mark.asyncio
|
|
async def test_get_login_response(bearer_transport: BearerTransport):
|
|
response = Response()
|
|
login_response = await bearer_transport.get_login_response("TOKEN", response)
|
|
|
|
assert isinstance(login_response, BearerResponse)
|
|
|
|
assert login_response.access_token == "TOKEN"
|
|
assert login_response.token_type == "bearer"
|
|
|
|
|
|
@pytest.mark.authentication
|
|
@pytest.mark.asyncio
|
|
async def test_get_logout_response(bearer_transport: BearerTransport):
|
|
response = Response()
|
|
with pytest.raises(TransportLogoutNotSupportedError):
|
|
await bearer_transport.get_logout_response(response)
|
|
|
|
|
|
@pytest.mark.authentication
|
|
@pytest.mark.openapi
|
|
def test_get_openapi_login_responses_success(bearer_transport: BearerTransport):
|
|
openapi_responses = bearer_transport.get_openapi_login_responses_success()
|
|
assert openapi_responses[status.HTTP_200_OK]["model"] == BearerResponse
|
|
|
|
|
|
@pytest.mark.authentication
|
|
@pytest.mark.openapi
|
|
def test_get_openapi_logout_responses_success(bearer_transport: BearerTransport):
|
|
openapi_responses = bearer_transport.get_openapi_logout_responses_success()
|
|
assert openapi_responses == {}
|