feat: add database session dependency for FastAPI routes
- Introduced a new dependency in `middleware/dependencies.py` to provide an asynchronous database session using SQLModel. - This dependency can be utilized in route functions to facilitate database operations.
This commit is contained in:
@@ -5,35 +5,35 @@ async def test_group_curd():
|
||||
"""测试数据库的增删改查"""
|
||||
from models import database, migration
|
||||
from models.group import Group
|
||||
|
||||
await database.init_db(url='sqlite:///:memory:')
|
||||
|
||||
|
||||
await database.init_db(url='sqlite+aiosqlite:///:memory:')
|
||||
|
||||
await migration.migration()
|
||||
|
||||
# 测试增 Create
|
||||
test_group = Group(name='test_group')
|
||||
created_group = await Group.create(test_group)
|
||||
|
||||
assert created_group is not None
|
||||
assert created_group.id is not None
|
||||
assert created_group.name == 'test_group'
|
||||
|
||||
# 测试查 Read
|
||||
fetched_group = await Group.get(id=created_group.id)
|
||||
assert fetched_group is not None
|
||||
assert fetched_group.id == created_group.id
|
||||
assert fetched_group.name == 'test_group'
|
||||
|
||||
# 测试更新 Update
|
||||
updated_group = await Group.set(
|
||||
id=fetched_group.id,
|
||||
name='updated_group')
|
||||
|
||||
assert updated_group is not None
|
||||
assert updated_group.id == fetched_group.id
|
||||
assert updated_group.name == 'updated_group'
|
||||
|
||||
# 测试删除 Delete
|
||||
await Group.delete(id=updated_group.id)
|
||||
deleted_group = await Group.get(id=updated_group.id)
|
||||
assert deleted_group is None
|
||||
|
||||
async for session in database.get_session():
|
||||
# 测试增 Create
|
||||
test_group = Group(name='test_group')
|
||||
created_group = await test_group.save(session)
|
||||
|
||||
assert created_group is not None
|
||||
assert created_group.id is not None
|
||||
assert created_group.name == 'test_group'
|
||||
|
||||
# 测试查 Read
|
||||
fetched_group = await Group.get(session, Group.id == created_group.id)
|
||||
assert fetched_group is not None
|
||||
assert fetched_group.id == created_group.id
|
||||
assert fetched_group.name == 'test_group'
|
||||
|
||||
# 测试更新 Update
|
||||
updated_group = await fetched_group.update(session, {"name": "updated_group"})
|
||||
|
||||
assert updated_group is not None
|
||||
assert updated_group.id == fetched_group.id
|
||||
assert updated_group.name == 'updated_group'
|
||||
|
||||
# 测试删除 Delete
|
||||
await updated_group.delete(session)
|
||||
deleted_group = await Group.get(session, Group.id == updated_group.id)
|
||||
assert deleted_group is None
|
||||
break
|
||||
|
||||
@@ -6,51 +6,52 @@ async def test_user_curd():
|
||||
from models import database, migration
|
||||
from models.group import Group
|
||||
from models.user import User
|
||||
|
||||
await database.init_db(url='sqlite:///:memory:')
|
||||
|
||||
|
||||
await database.init_db(url='sqlite+aiosqlite:///:memory:')
|
||||
|
||||
await migration.migration()
|
||||
|
||||
# 新建一个测试用户组
|
||||
test_user_group = Group(name='test_user_group')
|
||||
created_group = await Group.create(test_user_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
|
||||
|
||||
# 测试查 Read
|
||||
fetched_user = await User.get(id=created_user.id)
|
||||
|
||||
assert fetched_user is not None
|
||||
assert fetched_user.email == 'test_user'
|
||||
assert fetched_user.password == 'test_password'
|
||||
assert fetched_user.group_id == created_group.id
|
||||
|
||||
# 测试改 Update
|
||||
updated_user = await User.update(
|
||||
id=fetched_user.id,
|
||||
email='updated_user',
|
||||
password='updated_password'
|
||||
)
|
||||
|
||||
assert updated_user is not None
|
||||
assert updated_user.email == 'updated_user'
|
||||
assert updated_user.password == 'updated_password'
|
||||
|
||||
# 测试删除 Delete
|
||||
await User.delete(id=updated_user.id)
|
||||
deleted_user = await User.get(id=updated_user.id)
|
||||
|
||||
assert deleted_user is None
|
||||
|
||||
async for session in database.get_session():
|
||||
# 新建一个测试用户组
|
||||
test_user_group = Group(name='test_user_group')
|
||||
created_group = await test_user_group.save(session)
|
||||
|
||||
test_user = User(
|
||||
username='test_user',
|
||||
password='test_password',
|
||||
group_id=created_group.id
|
||||
)
|
||||
|
||||
# 测试增 Create
|
||||
created_user = await test_user.save(session)
|
||||
|
||||
# 验证用户是否存在
|
||||
assert created_user.id is not None
|
||||
assert created_user.username == 'test_user'
|
||||
assert created_user.password == 'test_password'
|
||||
assert created_user.group_id == created_group.id
|
||||
|
||||
# 测试查 Read
|
||||
fetched_user = await User.get(session, User.id == created_user.id)
|
||||
|
||||
assert fetched_user is not None
|
||||
assert fetched_user.username == 'test_user'
|
||||
assert fetched_user.password == 'test_password'
|
||||
assert fetched_user.group_id == created_group.id
|
||||
|
||||
# 测试改 Update
|
||||
updated_user = await fetched_user.update(
|
||||
session,
|
||||
{"username": "updated_user", "password": "updated_password"}
|
||||
)
|
||||
|
||||
assert updated_user is not None
|
||||
assert updated_user.username == 'updated_user'
|
||||
assert updated_user.password == 'updated_password'
|
||||
|
||||
# 测试删除 Delete
|
||||
await updated_user.delete(session)
|
||||
deleted_user = await User.get(session, User.id == updated_user.id)
|
||||
|
||||
assert deleted_user is None
|
||||
break
|
||||
|
||||
Reference in New Issue
Block a user