145 lines
4.7 KiB
Python
145 lines
4.7 KiB
Python
"""内驱力 /drive API 测试。"""
|
|
|
|
from collections.abc import AsyncGenerator
|
|
|
|
import pytest
|
|
from httpx import ASGITransport, AsyncClient
|
|
|
|
from app.db.session import AsyncSessionLocal
|
|
from app.main import app
|
|
from app.models.big_cycle import BigCycle
|
|
from app.models.drive import CuriosityQuestion, ExploratorySubgoal
|
|
from app.services.characters import get_active_character_model
|
|
|
|
|
|
@pytest.fixture
|
|
async def client() -> AsyncGenerator[AsyncClient, None]:
|
|
transport = ASGITransport(app=app)
|
|
async with AsyncClient(transport=transport, base_url="http://test") as http_client:
|
|
yield http_client
|
|
|
|
|
|
async def _seed(character_id: str, subgoal_status: str = "candidate") -> str:
|
|
async with AsyncSessionLocal() as session:
|
|
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="用户为何频繁提到项目X",
|
|
source="memory_gap",
|
|
expected_value=0.7,
|
|
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=subgoal_status,
|
|
)
|
|
session.add(sub)
|
|
await session.commit()
|
|
return sub.id
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_list_curiosities(client: AsyncClient) -> None:
|
|
async with AsyncSessionLocal() as session:
|
|
character = await get_active_character_model(session)
|
|
await _seed(character.id)
|
|
|
|
resp = await client.get(f"/api/v1/characters/{character.id}/curiosities")
|
|
assert resp.status_code == 200
|
|
data = resp.json()
|
|
assert data["total"] == 1
|
|
assert data["items"][0]["question"] == "用户为何频繁提到项目X"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_list_subgoals_with_status_filter(client: AsyncClient) -> None:
|
|
async with AsyncSessionLocal() as session:
|
|
character = await get_active_character_model(session)
|
|
await _seed(character.id, "candidate")
|
|
await _seed(character.id, "approved")
|
|
|
|
resp = await client.get(
|
|
f"/api/v1/characters/{character.id}/subgoals", params={"status": "candidate"}
|
|
)
|
|
assert resp.status_code == 200
|
|
assert resp.json()["total"] == 1
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_approve_subgoal(client: AsyncClient) -> None:
|
|
async with AsyncSessionLocal() as session:
|
|
character = await get_active_character_model(session)
|
|
sid = await _seed(character.id)
|
|
|
|
resp = await client.post(f"/api/v1/subgoals/{sid}/approve")
|
|
assert resp.status_code == 200
|
|
assert resp.json()["status"] == "approved"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_defer_subgoal(client: AsyncClient) -> None:
|
|
async with AsyncSessionLocal() as session:
|
|
character = await get_active_character_model(session)
|
|
sid = await _seed(character.id)
|
|
|
|
resp = await client.post(f"/api/v1/subgoals/{sid}/defer")
|
|
assert resp.status_code == 200
|
|
assert resp.json()["status"] == "deferred"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_reject_subgoal(client: AsyncClient) -> None:
|
|
async with AsyncSessionLocal() as session:
|
|
character = await get_active_character_model(session)
|
|
sid = await _seed(character.id)
|
|
|
|
resp = await client.post(f"/api/v1/subgoals/{sid}/reject")
|
|
assert resp.status_code == 200
|
|
assert resp.json()["status"] == "rejected"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_approve_subgoal_creates_goal(client: AsyncClient) -> None:
|
|
"""审批通过次生目标后应自动创建太一目标。"""
|
|
from sqlalchemy import select
|
|
|
|
from app.models.governance import Goal
|
|
|
|
async with AsyncSessionLocal() as session:
|
|
character = await get_active_character_model(session)
|
|
sid = await _seed(character.id)
|
|
|
|
resp = await client.post(f"/api/v1/subgoals/{sid}/approve")
|
|
assert resp.status_code == 200
|
|
|
|
# 验证生成了 Goal
|
|
async with AsyncSessionLocal() as session:
|
|
result = await session.execute(
|
|
select(Goal).where(Goal.character_id == character.id)
|
|
)
|
|
goals = result.scalars().all()
|
|
matching = [g for g in goals if g.title == "询问项目目标"]
|
|
assert len(matching) == 1
|
|
goal = matching[0]
|
|
assert goal.level == "task"
|
|
assert goal.type == "exploration"
|
|
assert goal.status == "active"
|