mirror of
https://github.com/fastapi-users/fastapi-users.git
synced 2026-03-13 07:49:55 +08:00
40 lines
901 B
Python
40 lines
901 B
Python
import pytest
|
|
|
|
from fastapi_users.db import UserDBInterface
|
|
from fastapi_users.models import UserDB
|
|
from fastapi_users.password import get_password_hash
|
|
|
|
active_user = UserDB(
|
|
email='king.arthur@camelot.bt',
|
|
hashed_password=get_password_hash('guinevere'),
|
|
)
|
|
|
|
inactive_user = UserDB(
|
|
email='percival@camelot.bt',
|
|
hashed_password=get_password_hash('angharad'),
|
|
is_active=False
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def user() -> UserDB:
|
|
return active_user
|
|
|
|
|
|
class MockUserDBInterface(UserDBInterface):
|
|
|
|
async def get_by_email(self, email: str) -> UserDB:
|
|
if email == active_user.email:
|
|
return active_user
|
|
elif email == inactive_user.email:
|
|
return inactive_user
|
|
return None
|
|
|
|
async def create(self, user: UserDB) -> UserDB:
|
|
return user
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_db_interface() -> MockUserDBInterface:
|
|
return MockUserDBInterface()
|