feat(agent): complete closed-loop runtime and pricing fallback

2026-03-05 15:34:37 +08:00
parent b02a322bf3
commit b486e78ff3
67 changed files with 3832 additions and 7 deletions
@@ -0,0 +1,107 @@
+from __future__ import annotations
+
+from types import SimpleNamespace
+from uuid import uuid4
+
+from fastapi.testclient import TestClient
+
+from app import app
+from core.auth.models import CurrentUser
+from v1.agent.dependencies import get_agent_service
+from v1.users.dependencies import get_current_user
+
+
+class _FakeAgentService:
+    def __init__(self) -> None:
+        self._stream_called = False
+
+    async def enqueue_run(
+        self, *, session_id: str | None, prompt: str, current_user: CurrentUser
+    ):
+        del prompt, current_user
+        resolved_session = session_id or "auto-created-session"
+        return SimpleNamespace(
+            task_id="task-run-1",
+            session_id=resolved_session,
+            created=session_id is None,
+        )
+
+    async def enqueue_resume(
+        self,
+        *,
+        session_id: str,
+        tool_call_id: str,
+        current_user: CurrentUser,
+    ):
+        del tool_call_id, current_user
+        return SimpleNamespace(
+            task_id="task-resume-1", session_id=session_id, created=False
+        )
+
+    async def stream_events(
+        self,
+        *,
+        session_id: str,
+        last_event_id: str | None,
+        current_user: CurrentUser,
+    ) -> list[dict[str, object]]:
+        del session_id, current_user
+        if self._stream_called:
+            return []
+        self._stream_called = True
+        return [
+            {"id": "2-0", "event": {"type": "RUN_STARTED"}, "cursor": last_event_id}
+        ]
+
+
+def test_run_requires_auth_and_returns_202_task_id() -> None:
+    app.dependency_overrides[get_agent_service] = lambda: _FakeAgentService()
+    client = TestClient(app)
+
+    try:
+        unauthorized = client.post(
+            "/api/v1/agent/runs",
+            json={"session_id": "session-1", "prompt": "hello"},
+        )
+        assert unauthorized.status_code == 401
+
+        app.dependency_overrides[get_current_user] = lambda: CurrentUser(
+            id=uuid4(), email="user@example.com"
+        )
+        authorized = client.post(
+            "/api/v1/agent/runs",
+            json={"session_id": "session-1", "prompt": "hello"},
+        )
+        assert authorized.status_code == 202
+        assert authorized.json()["task_id"] == "task-run-1"
+        assert authorized.json()["created"] is False
+
+        first_chat = client.post(
+            "/api/v1/agent/runs",
+            json={"prompt": "hello"},
+        )
+        assert first_chat.status_code == 202
+        assert first_chat.json()["session_id"] == "auto-created-session"
+        assert first_chat.json()["created"] is True
+    finally:
+        app.dependency_overrides = {}
+
+
+def test_stream_reads_from_last_event_id() -> None:
+    app.dependency_overrides[get_agent_service] = lambda: _FakeAgentService()
+    app.dependency_overrides[get_current_user] = lambda: CurrentUser(
+        id=uuid4(), email="user@example.com"
+    )
+    client = TestClient(app)
+
+    try:
+        response = client.get(
+            "/api/v1/agent/runs/session-1/events?idle_limit=1",
+            headers={"Last-Event-ID": "1-0"},
+        )
+        assert response.status_code == 200
+        assert response.headers["content-type"].startswith("text/event-stream")
+        assert "id: 2-0" in response.text
+        assert "event: RUN_STARTED" in response.text
+    finally:
+        app.dependency_overrides = {}