54 lines
1.8 KiB
Python
54 lines
1.8 KiB
Python
"""司命台治理种子数据测试。"""
|
|
|
|
from collections.abc import AsyncGenerator
|
|
|
|
import pytest
|
|
from sqlalchemy import select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.db.governance_seed import _GOALS, ensure_governance_seed
|
|
from app.db.session import AsyncSessionLocal
|
|
from app.models.governance import Goal
|
|
|
|
|
|
@pytest.fixture
|
|
async def db_session() -> AsyncGenerator[AsyncSession, None]:
|
|
async with AsyncSessionLocal() as session:
|
|
yield session
|
|
|
|
|
|
class TestGoalSeedsHaveCharacterId:
|
|
"""_GOALS 每条 seed 都包含 character_id 与 type 字段。"""
|
|
|
|
def test_all_goals_have_character_id_key(self) -> None:
|
|
for item in _GOALS:
|
|
assert "character_id" in item, f"{item['id']} 缺少 character_id"
|
|
|
|
def test_all_goals_have_type_key(self) -> None:
|
|
for item in _GOALS:
|
|
assert "type" in item, f"{item['id']} 缺少 type"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_seed_goals_persist_with_new_fields(
|
|
db_session: AsyncSession,
|
|
) -> None:
|
|
"""验证种子写入数据库后 character_id / type 字段被正确持久化。"""
|
|
# 先清理已有数据，确保从头写入种子
|
|
await db_session.execute(Goal.__table__.delete())
|
|
await db_session.commit()
|
|
|
|
await ensure_governance_seed(db_session)
|
|
|
|
result = await db_session.execute(select(Goal))
|
|
goals = result.scalars().all()
|
|
|
|
assert len(goals) > 0
|
|
for g in goals:
|
|
assert hasattr(g, "character_id"), f"{g.id} 缺少 character_id 属性"
|
|
assert hasattr(g, "type"), f"{g.id} 缺少 type 属性"
|
|
# character_id 在 seed 中全为 None，确保字段可空
|
|
assert g.character_id is None
|
|
# type 应与 seed 定义一致
|
|
assert g.type == "mission", f"{g.id} type 应为 mission，实际 {g.type}"
|