mirror of
https://github.com/fastapi-users/fastapi-users.git
synced 2026-03-13 07:49:55 +08:00
Fix #391: put user creation logic in a importable function
This commit is contained in:
47
tests/test_user.py
Normal file
47
tests/test_user.py
Normal file
@@ -0,0 +1,47 @@
|
||||
import pytest
|
||||
|
||||
from fastapi_users.user import CreateUserProtocol, UserAlreadyExists, get_create_user
|
||||
from tests.conftest import UserCreate, UserDB
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def create_user(
|
||||
mock_user_db,
|
||||
) -> CreateUserProtocol:
|
||||
return get_create_user(mock_user_db, UserDB)
|
||||
|
||||
|
||||
@pytest.mark.router
|
||||
@pytest.mark.asyncio
|
||||
class TestCreateUser:
|
||||
@pytest.mark.parametrize(
|
||||
"email", ["king.arthur@camelot.bt", "King.Arthur@camelot.bt"]
|
||||
)
|
||||
async def test_existing_user(self, email, create_user):
|
||||
user = UserCreate(email=email, password="guinevere")
|
||||
with pytest.raises(UserAlreadyExists):
|
||||
await create_user(user)
|
||||
|
||||
@pytest.mark.parametrize("email", ["lancelot@camelot.bt", "Lancelot@camelot.bt"])
|
||||
async def test_regular_user(self, email, create_user):
|
||||
user = UserCreate(email=email, password="guinevere")
|
||||
created_user = await create_user(user)
|
||||
assert type(created_user) == UserDB
|
||||
|
||||
@pytest.mark.parametrize("safe,result", [(True, False), (False, True)])
|
||||
async def test_superuser(self, create_user, safe, result):
|
||||
user = UserCreate(
|
||||
email="lancelot@camelot.b", password="guinevere", is_superuser=True
|
||||
)
|
||||
created_user = await create_user(user, safe)
|
||||
assert type(created_user) == UserDB
|
||||
assert created_user.is_superuser is result
|
||||
|
||||
@pytest.mark.parametrize("safe,result", [(True, True), (False, False)])
|
||||
async def test_is_active(self, create_user, safe, result):
|
||||
user = UserCreate(
|
||||
email="lancelot@camelot.b", password="guinevere", is_active=False
|
||||
)
|
||||
created_user = await create_user(user, safe)
|
||||
assert type(created_user) == UserDB
|
||||
assert created_user.is_active is result
|
||||
Reference in New Issue
Block a user