98 lines
2.9 KiB
Python
98 lines
2.9 KiB
Python
"""内驱力审核服务测试（次生目标 approve/defer/reject）。"""
|
|
|
|
import pytest
|
|
|
|
from app.core.errors import AppError
|
|
from app.db.session import AsyncSessionLocal
|
|
from app.models.big_cycle import BigCycle
|
|
from app.models.drive import CuriosityQuestion, ExploratorySubgoal
|
|
from app.services import drive as drive_service
|
|
from app.services.characters import get_active_character_model
|
|
|
|
|
|
async def _seed_subgoal(session, character_id: str, status: str = "candidate") -> str:
|
|
cycle = BigCycle(
|
|
character_id=character_id,
|
|
trigger="test",
|
|
status="completed",
|
|
memory_count_at_trigger=20,
|
|
)
|
|
session.add(cycle)
|
|
await session.flush()
|
|
curios = CuriosityQuestion(
|
|
character_id=character_id,
|
|
cycle_id=cycle.id,
|
|
question="q",
|
|
source="memory_gap",
|
|
expected_value=0.5,
|
|
urgency=0.5,
|
|
risk_level="L2",
|
|
)
|
|
session.add(curios)
|
|
await session.flush()
|
|
sub = ExploratorySubgoal(
|
|
character_id=character_id,
|
|
cycle_id=cycle.id,
|
|
curiosity_question_id=curios.id,
|
|
title="询问项目目标",
|
|
description="主动询问",
|
|
expected_information_gain=0.6,
|
|
interruption_cost=0.3,
|
|
risk_level="L2",
|
|
status=status,
|
|
)
|
|
session.add(sub)
|
|
await session.flush()
|
|
return sub.id
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_approve_candidate_subgoal() -> None:
|
|
async with AsyncSessionLocal() as session:
|
|
character = await get_active_character_model(session)
|
|
sid = await _seed_subgoal(session, character.id)
|
|
await session.commit()
|
|
|
|
resp = await drive_service.approve_subgoal(sid)
|
|
assert resp.status == "approved"
|
|
assert resp.reviewed_by == "user"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_defer_candidate_subgoal() -> None:
|
|
async with AsyncSessionLocal() as session:
|
|
character = await get_active_character_model(session)
|
|
sid = await _seed_subgoal(session, character.id)
|
|
await session.commit()
|
|
|
|
resp = await drive_service.defer_subgoal(sid)
|
|
assert resp.status == "deferred"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_reject_candidate_subgoal() -> None:
|
|
async with AsyncSessionLocal() as session:
|
|
character = await get_active_character_model(session)
|
|
sid = await _seed_subgoal(session, character.id)
|
|
await session.commit()
|
|
|
|
resp = await drive_service.reject_subgoal(sid)
|
|
assert resp.status == "rejected"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_review_non_candidate_conflicts() -> None:
|
|
async with AsyncSessionLocal() as session:
|
|
character = await get_active_character_model(session)
|
|
sid = await _seed_subgoal(session, character.id, status="approved")
|
|
await session.commit()
|
|
|
|
with pytest.raises(AppError):
|
|
await drive_service.approve_subgoal(sid)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_approve_not_found() -> None:
|
|
with pytest.raises(AppError):
|
|
await drive_service.approve_subgoal("subgoal_missing")
|