65 lines
2.2 KiB
Python
65 lines
2.2 KiB
Python
"""大周天循环骨架测试（触发 + 阶段化执行）。"""
|
|
|
|
import pytest
|
|
|
|
from app.application.big_cycle import drain_pending_big_cycles, maybe_trigger_big_cycle
|
|
from app.db.session import AsyncSessionLocal
|
|
from app.models.memory import Memory
|
|
from app.repositories import big_cycle as big_cycle_repo
|
|
from app.services.characters import get_active_character_model
|
|
|
|
|
|
async def _seed_memories(character_id: str, count: int) -> None:
|
|
async with AsyncSessionLocal() as session:
|
|
for i in range(count):
|
|
session.add(
|
|
Memory(
|
|
character_id=character_id,
|
|
type="user",
|
|
title=f"记忆{i}",
|
|
content=f"内容{i}",
|
|
)
|
|
)
|
|
await session.commit()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_big_cycle_runs_when_threshold_met() -> None:
|
|
async with AsyncSessionLocal() as session:
|
|
character = await get_active_character_model(session)
|
|
await _seed_memories(character.id, 20)
|
|
|
|
cycle_id = await maybe_trigger_big_cycle(character.id, trigger="test")
|
|
assert cycle_id is not None
|
|
await drain_pending_big_cycles()
|
|
|
|
async with AsyncSessionLocal() as session:
|
|
cycle = await big_cycle_repo.get_cycle(session, cycle_id)
|
|
assert cycle is not None
|
|
assert cycle.status == "completed"
|
|
assert cycle.memory_count_at_trigger >= 20
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_big_cycle_skips_below_threshold() -> None:
|
|
async with AsyncSessionLocal() as session:
|
|
character = await get_active_character_model(session)
|
|
# 默认人物 0 记忆，低于阈值
|
|
cycle_id = await maybe_trigger_big_cycle(character.id, trigger="test")
|
|
assert cycle_id is None
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_big_cycle_respects_min_interval() -> None:
|
|
async with AsyncSessionLocal() as session:
|
|
character = await get_active_character_model(session)
|
|
await _seed_memories(character.id, 20)
|
|
|
|
first = await maybe_trigger_big_cycle(character.id, trigger="test")
|
|
assert first is not None
|
|
await drain_pending_big_cycles()
|
|
|
|
# 间隔窗口内再次触发 → 跳过
|
|
second = await maybe_trigger_big_cycle(character.id, trigger="test")
|
|
assert second is None
|