单元测试:新建用户与用户组
This commit is contained in:
29
tests/test_db_user.py
Normal file
29
tests/test_db_user.py
Normal file
@@ -0,0 +1,29 @@
|
||||
import pytest
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_user_curd():
|
||||
"""测试数据库的增删改查"""
|
||||
from models import database
|
||||
from models.group import Group
|
||||
from models.user import User
|
||||
|
||||
await database.init_db(url='sqlite:///:memory:')
|
||||
|
||||
# 新建一个测试用户组
|
||||
test_group = Group(name='test_group')
|
||||
created_group = await Group.create(test_group)
|
||||
|
||||
test_user = User(
|
||||
email='test_user',
|
||||
password='test_password',
|
||||
group_id=created_group.id
|
||||
)
|
||||
|
||||
# 测试增 Create
|
||||
created_user = await User.create(test_user)
|
||||
|
||||
# 验证用户是否存在
|
||||
assert created_user.id is not None
|
||||
assert created_user.email == 'test_user'
|
||||
assert created_user.password == 'test_password'
|
||||
assert created_user.group_id == created_group.id
|
||||
Reference in New Issue
Block a user