import pytest

@pytest.mark.asyncio
async def test_user_curd():
    """测试数据库的增删改查"""
    from sqlmodels import database, migration
    from sqlmodels.group import Group
    from sqlmodels.user import User

    await database.init_db(url='sqlite+aiosqlite:///:memory:')

    await migration.migration()

    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(
            email='test_user@test.local',
            group_id=created_group.id
        )

        # 测试增 Create
        created_user = await test_user.save(session)

        # 验证用户是否存在
        assert created_user.id is not None
        assert created_user.email == 'test_user@test.local'
        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.email == 'test_user@test.local'
        assert fetched_user.group_id == created_group.id

        # 测试改 Update
        updated_user = await fetched_user.update(
            session,
            {"email": "updated_user@test.local"}
        )

        assert updated_user is not None
        assert updated_user.email == 'updated_user@test.local'

        # 测试删除 Delete
        await updated_user.delete(session)
        deleted_user = await User.get(session, User.id == updated_user.id)

        assert deleted_user is None
        break