80 lines
3.2 KiB
Python
80 lines
3.2 KiB
Python
"""时态纪念日数据访问测试（辰正 §5.3）。"""
|
|
|
|
import pytest
|
|
|
|
from app.db.session import AsyncSessionLocal
|
|
from app.models.time_anniversary import TimeAnniversary
|
|
from app.repositories import time_anniversaries as anni_repo
|
|
from app.services.characters import get_active_character_model
|
|
|
|
|
|
def _anniversary(character_id: str, **overrides) -> TimeAnniversary:
|
|
base = {
|
|
"character_id": character_id,
|
|
"name": "我们的相遇日",
|
|
"calendar": "solar",
|
|
"date": "2025-09-19",
|
|
"recurs_yearly": True,
|
|
"kind": "anniversary",
|
|
"subject_type": "relationship",
|
|
"source": "auto",
|
|
"metadata_json": "{}",
|
|
}
|
|
base.update(overrides)
|
|
return TimeAnniversary(**base)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_create_and_get_anniversary() -> None:
|
|
async with AsyncSessionLocal() as session:
|
|
character = await get_active_character_model(session)
|
|
created = await anni_repo.create_anniversary(session, _anniversary(character.id))
|
|
fetched = await anni_repo.get_anniversary(session, created.id)
|
|
assert fetched is not None
|
|
assert fetched.name == "我们的相遇日"
|
|
assert fetched.id.startswith("anniv_")
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_list_anniversaries_by_character() -> None:
|
|
async with AsyncSessionLocal() as session:
|
|
character = await get_active_character_model(session)
|
|
await anni_repo.create_anniversary(session, _anniversary(character.id, date="2025-03-01"))
|
|
await anni_repo.create_anniversary(session, _anniversary(character.id, date="2025-09-19"))
|
|
items = await anni_repo.list_anniversaries(session, character.id)
|
|
# 默认人物 seed 时已带 1 条 birthday 纪念日；此处只验证 anniversary kind 的列表行为
|
|
anniversary_items = [a for a in items if a.kind == "anniversary"]
|
|
assert len(anniversary_items) == 2
|
|
# list 默认按 date 升序
|
|
assert anniversary_items[0].date < anniversary_items[1].date
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_update_anniversary() -> None:
|
|
async with AsyncSessionLocal() as session:
|
|
character = await get_active_character_model(session)
|
|
created = await anni_repo.create_anniversary(session, _anniversary(character.id))
|
|
updated = await anni_repo.update_anniversary(
|
|
session, created.id, name="相识纪念日", source="manual"
|
|
)
|
|
assert updated is not None
|
|
assert updated.name == "相识纪念日"
|
|
assert updated.source == "manual"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_update_anniversary_missing_returns_none() -> None:
|
|
async with AsyncSessionLocal() as session:
|
|
result = await anni_repo.update_anniversary(session, "anniv_nonexistent", name="x")
|
|
assert result is None
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_delete_anniversary() -> None:
|
|
async with AsyncSessionLocal() as session:
|
|
character = await get_active_character_model(session)
|
|
created = await anni_repo.create_anniversary(session, _anniversary(character.id))
|
|
assert await anni_repo.delete_anniversary(session, created.id) is True
|
|
assert await anni_repo.get_anniversary(session, created.id) is None
|
|
assert await anni_repo.delete_anniversary(session, created.id) is False
|