chore: initialize project repository

This commit is contained in:
Codex
2026-09-09 21:55:37 +08:00
commit b1497fd2c6
167 changed files with 17690 additions and 0 deletions
+35
View File
@@ -0,0 +1,35 @@
from unittest.mock import AsyncMock, Mock
from fastapi.testclient import TestClient
from app.api.controllers.agent_runs import get_session
from app.api.dependencies.auth import build_request_context
from app.core.contracts import RequestContext
from app.main import create_app
def test_unknown_agent_returns_contract_404_without_database_write():
app = create_app()
session = AsyncMock()
session.add = Mock()
async def context():
return RequestContext(user_id="1", trace_id="test-trace", roles=("customer",),
permissions=("agent:run",))
async def database():
yield session
app.dependency_overrides[build_request_context] = context
app.dependency_overrides[get_session] = database
with TestClient(app) as client:
response = client.post("/api/v1/agent-runs", json={
"agent_type": "no_such_agent", "message": "test", "session_id": "test",
"idempotency_key": "1234567890123456",
})
assert response.status_code == 404
assert response.json()["error"]["code"] == "AGENT_TYPE_NOT_FOUND"
assert response.json()["meta"]["trace_id"] == "test-trace"
session.add.assert_not_called()
session.execute.assert_not_awaited()
session.flush.assert_not_awaited()
+88
View File
@@ -0,0 +1,88 @@
from app.api.views.agent_run_sse import encode_events, recovery_events
def test_first_connection_emits_unicode_deltas_only_after_commit():
content = "基金😀\n结果" * 100
events = recovery_events(run_id="r", trace_id="t", status="succeeded", error_code=None,
content=content, tool_calls=None, replay=False, chunk_size=11)
assert events[0][0] == "start" and events[-1][0] == "done"
assert "".join(payload["content"] for name, payload in events if name == "delta") == content
assert all(len(payload["content"]) <= 11 for name, payload in events if name == "delta")
ids = [line for event in encode_events("r", events)
for line in event.splitlines() if line.startswith("id:")]
assert len(set(ids)) == len(ids)
pending = recovery_events(run_id="r", trace_id="t", status="running", error_code=None,
content=content, tool_calls=None, replay=False)
assert [name for name, _ in pending] == ["start"]
def test_empty_result_still_emits_delta():
events = recovery_events(run_id="r", trace_id="t", status="succeeded", error_code=None,
content="", tool_calls=None, replay=False)
assert [name for name, _ in events] == ["start", "delta", "done"]
def test_pending_run_never_emits_result_events() -> None:
events = recovery_events(
run_id="run",
trace_id="trace",
status="running",
error_code=None,
content=None,
tool_calls=None,
)
assert [name for name, _ in events] == ["start"]
def test_succeeded_run_emits_tools_and_result_after_terminal_state() -> None:
events = recovery_events(
run_id="run",
trace_id="trace",
status="succeeded",
error_code=None,
content="answer",
tool_calls={"calls": []},
)
assert [name for name, _ in events] == ["start", "tools", "replace", "done"]
def test_failed_run_emits_error_without_result() -> None:
events = recovery_events(
run_id="run",
trace_id="trace",
status="failed",
error_code="MODEL_TIMEOUT",
content=None,
tool_calls=None,
)
assert [name for name, _ in events] == ["start", "error"]
def test_client_disconnect_does_not_change_recoverable_terminal_events() -> None:
events = recovery_events(
run_id="run",
trace_id="trace",
status="succeeded",
error_code=None,
content="answer",
tool_calls=None,
)
stream = encode_events("run", events)
first = next(stream)
assert "event: start" in first
del stream
recovered = list(
encode_events(
"run",
recovery_events(
run_id="run",
trace_id="trace",
status="succeeded",
error_code=None,
content="answer",
tool_calls=None,
),
)
)
assert any("event: replace" in item for item in recovered)
assert any("event: done" in item for item in recovered)
+16
View File
@@ -0,0 +1,16 @@
import ast
from pathlib import Path
def test_controllers_do_not_access_models_repositories_or_session_operations():
for path in Path("app/api/controllers").glob("*.py"):
tree = ast.parse(path.read_text(encoding="utf-8"))
for node in ast.walk(tree):
if isinstance(node, ast.ImportFrom):
assert not (node.module or "").startswith(("app.model", "app.repository")), path
if node.module == "sqlalchemy":
raise AssertionError(f"Controller imports SQL statements: {path}")
if isinstance(node, ast.Call) and isinstance(node.func, ast.Attribute):
target = node.func.value
if isinstance(target, ast.Name) and target.id == "session":
raise AssertionError(f"Controller operates on session: {path}:{node.lineno}")