mirror of
https://github.com/fastapi-users/fastapi-users.git
synced 2025-11-04 06:37:51 +08:00
64 lines
2.0 KiB
Python
64 lines
2.0 KiB
Python
from typing import Optional
|
|
|
|
import pytest
|
|
from fastapi import Request, status
|
|
from fastapi.security.base import SecurityBase
|
|
|
|
from fastapi_users.authentication import BaseAuthentication, DuplicateBackendNamesError
|
|
from fastapi_users.db import BaseUserDatabase
|
|
from fastapi_users.models import BaseUserDB
|
|
|
|
|
|
class MockSecurityScheme(SecurityBase):
|
|
def __call__(self, request: Request) -> Optional[str]:
|
|
return "mock"
|
|
|
|
|
|
class BackendNone(BaseAuthentication[str]):
|
|
def __init__(self, name="none"):
|
|
super().__init__(name, logout=False)
|
|
self.scheme = MockSecurityScheme()
|
|
|
|
async def __call__(
|
|
self, credentials: Optional[str], user_db: BaseUserDatabase
|
|
) -> Optional[BaseUserDB]:
|
|
return None
|
|
|
|
|
|
class BackendUser(BaseAuthentication[str]):
|
|
def __init__(self, user: BaseUserDB, name="user"):
|
|
super().__init__(name, logout=False)
|
|
self.scheme = MockSecurityScheme()
|
|
self.user = user
|
|
|
|
async def __call__(
|
|
self, credentials: Optional[str], user_db: BaseUserDatabase
|
|
) -> Optional[BaseUserDB]:
|
|
return self.user
|
|
|
|
|
|
@pytest.mark.authentication
|
|
@pytest.mark.asyncio
|
|
async def test_authenticator(get_test_auth_client, user):
|
|
async for client in get_test_auth_client([BackendNone(), BackendUser(user)]):
|
|
response = await client.get("/test-current-user")
|
|
assert response.status_code == status.HTTP_200_OK
|
|
|
|
|
|
@pytest.mark.authentication
|
|
@pytest.mark.asyncio
|
|
async def test_authenticator_none(get_test_auth_client):
|
|
async for client in get_test_auth_client(
|
|
[BackendNone(), BackendNone(name="none-bis")]
|
|
):
|
|
response = await client.get("/test-current-user")
|
|
assert response.status_code == status.HTTP_401_UNAUTHORIZED
|
|
|
|
|
|
@pytest.mark.authentication
|
|
@pytest.mark.asyncio
|
|
async def test_authenticators_with_same_name(get_test_auth_client):
|
|
with pytest.raises(DuplicateBackendNamesError):
|
|
async for client in get_test_auth_client([BackendNone(), BackendNone()]):
|
|
pass
|