单元测试：新建用户与用户组

tests/test_database.py

@@ -25,52 +25,4 @@ async def test_initialize_db():
     
     await database.init_db(url='sqlite:///:memory:')
     
-    await migration.init_default_settings()
-
-@pytest.mark.asyncio
-async def test_add_settings():
-    """测试数据库的增删改查"""
-    from models import database
-    from models.setting import Setting
-    
-    await database.init_db(url='sqlite:///:memory:')
-    
-    # 测试增 Create
-    await Setting.add(
-        type='example_type', 
-        name='example_name', 
-        value='example_value')
-    
-    # 测试查 Read
-    setting = await Setting.get(
-        type='example_type', 
-        name='example_name')
-    
-    assert setting is not None, "设置项应该存在"
-    assert setting.value == 'example_value', "设置值不匹配"
-    
-    # 测试改 Update
-    await Setting.set(
-        type='example_type', 
-        name='example_name', 
-        value='updated_value')
-    
-    after_update_setting = await Setting.get(
-        type='example_type', 
-        name='example_name'
-        )
-    
-    assert after_update_setting is not None, "设置项应该存在"
-    assert after_update_setting.value == 'updated_value', "更新后的设置值不匹配"
-    
-    # 测试删 Delete
-    await Setting.delete(
-        type='example_type', 
-        name='example_name')
-    
-    after_delete_setting = await Setting.get(
-        type='example_type', 
-        name='example_name'
-    )
-    
-    assert after_delete_setting is None, "设置项应该被删除"
+    await migration.init_default_settings()

tests/test_db_settings.py

@@ -0,0 +1,49 @@
+import pytest
+
+@pytest.mark.asyncio
+async def test_settings_curd():
+    """测试数据库的增删改查"""
+    from models import database
+    from models.setting import Setting
+    
+    await database.init_db(url='sqlite:///:memory:')
+    
+    # 测试增 Create
+    await Setting.add(
+        type='example_type', 
+        name='example_name', 
+        value='example_value')
+    
+    # 测试查 Read
+    setting = await Setting.get(
+        type='example_type', 
+        name='example_name')
+    
+    assert setting is not None, "设置项应该存在"
+    assert setting == 'example_value', "设置值不匹配"
+    
+    # 测试改 Update
+    await Setting.set(
+        type='example_type', 
+        name='example_name', 
+        value='updated_value')
+    
+    after_update_setting = await Setting.get(
+        type='example_type', 
+        name='example_name'
+        )
+    
+    assert after_update_setting is not None, "设置项应该存在"
+    assert after_update_setting == 'updated_value', "更新后的设置值不匹配"
+    
+    # 测试删 Delete
+    await Setting.delete(
+        type='example_type', 
+        name='example_name')
+    
+    after_delete_setting = await Setting.get(
+        type='example_type', 
+        name='example_name'
+    )
+    
+    assert after_delete_setting is None, "设置项应该被删除"

tests/test_db_user.py

@@ -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