python/fastapi-users
1
0
Fork 0
mirror of https://github.com/fastapi-users/fastapi-users.git synced 2026-03-13 07:49:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

Improve generic typing

Browse Source
This commit is contained in:
François Voron
2021-09-14 11:53:43 +02:00
parent 90aee2d487
commit fdc8e54253
15 changed files with 60 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/test_authentication.py
View File

@@ -4,9 +4,9 @@ import pytest
from fastapi import Request, status
from fastapi.security.base import SecurityBase
from fastapi_users import models
from fastapi_users.authentication import BaseAuthentication, DuplicateBackendNamesError
from fastapi_users.manager import UserManager
from fastapi_users.models import BaseUserDB
class MockSecurityScheme(SecurityBase):
@@ -20,20 +20,20 @@ class BackendNone(BaseAuthentication[str]):
self.scheme = MockSecurityScheme()
async def __call__(
self, credentials: Optional[str], user_manager: UserManager
) -> Optional[BaseUserDB]:
self, credentials: Optional[str], user_manager: UserManager[models.UD]
) -> Optional[models.UD]:
return None
class BackendUser(BaseAuthentication[str]):
def __init__(self, user: BaseUserDB, name="user"):
def __init__(self, user: models.UD, name="user"):
super().__init__(name, logout=False)
self.scheme = MockSecurityScheme()
self.user = user
async def __call__(
self, credentials: Optional[str], user_manager: UserManager
) -> Optional[BaseUserDB]:
self, credentials: Optional[str], user_manager: UserManager[models.UD]
) -> Optional[models.UD]:
return self.user

2
tests/test_fastapi_users.py
View File

@@ -18,7 +18,7 @@ async def test_app_client(
get_test_client,
validate_password,
) -> AsyncGenerator[httpx.AsyncClient, None]:
fastapi_users = FastAPIUsers(
fastapi_users = FastAPIUsers[User, UserCreate, UserUpdate, UserDB](
get_mock_user_db,
[mock_authentication],
User,