张胜宇
e239eb778b
1) 客服 Agent 四份交付文档 + 构建脚手架：品牌由包装占位 XX科技 / 旧名 南方财富 统一为南方基金（热线 400-889-8899 / 官网 nffund.com），系统名改为「智能服务系统」； 同步追加 §0.4 修订记录行，工程记录行保留原占位字面以支撑硬编码扫描验收。 2) 开发文档：清理 28 份已作废/残留文档（14 份移出归档 + 14 份仓库副本）， 新增《文档规整方案与开发前待决事项-2026-09-17》。 3) 客服agent 四份交付文档首次纳入本分支。
127 lines
4.9 KiB
Python
127 lines
4.9 KiB
Python
"""SSE `Accept` 契约测试（文档 §3.2 请求头表 + §3.5 状态码 + §6.4 主要错误）。
|
||
|
||
权威规则：`Accept` **非必填**，"默认 `application/json`；SSE 为 `text/event-stream`"，
|
||
`SSE_NOT_ACCEPTABLE` 是 SSE 接口的主要错误之一（406）。
|
||
|
||
覆盖两条容易写错的分支：
|
||
1. **未携带** `Accept` 必须放行——把"未携带"当成"不接受"会把所有默认客户端挡在门外；
|
||
2. `q=0` 属于显式拒绝，不能被通配符掩盖。
|
||
|
||
校验顺序另在 `stream_agent_run_events` 中做了断言：可见性（RUN_NOT_FOUND）先行，
|
||
否则 406/404 的差异就等于一次运行存在性枚举。
|
||
"""
|
||
|
||
from typing import Any
|
||
|
||
import pytest
|
||
from fastapi.testclient import TestClient
|
||
|
||
from app.api.controllers.agent_runs import accepts_event_stream
|
||
from app.api.dependencies.auth import build_request_context
|
||
from app.core.contracts import RequestContext
|
||
from app.core.errors import RunNotFoundError
|
||
from app.main import create_app
|
||
from app.service.run_query_service import RunSnapshot
|
||
|
||
EVENTS_PATH = "/api/v1/agent-runs/run-1/events"
|
||
|
||
|
||
async def resolve_context() -> RequestContext:
|
||
return RequestContext(user_id="1", trace_id="trace-1", permissions=("agent:run",))
|
||
|
||
|
||
def app_with_query(monkeypatch: pytest.MonkeyPatch, query: Any) -> TestClient:
|
||
application = create_app()
|
||
application.dependency_overrides[build_request_context] = resolve_context
|
||
monkeypatch.setattr("app.api.controllers.agent_runs.RunQueryService", lambda: query)
|
||
return TestClient(application)
|
||
|
||
|
||
class StubQuery:
|
||
"""`get` 返回终态快照、`watch` 立即收流，避免测试连接挂住。"""
|
||
|
||
def __init__(self, status: str = "succeeded") -> None:
|
||
self.status = status
|
||
|
||
async def get(self, run_id: str, _ctx: RequestContext) -> RunSnapshot:
|
||
return RunSnapshot(run_id=run_id, trace_id="trace-1", status=self.status,
|
||
agent_type="customer_service", session_id="s", result=None,
|
||
error_code=None, created_at="2026-01-01T00:00:00Z",
|
||
completed_at="2026-01-01T00:00:01Z")
|
||
|
||
async def watch(self, initial: RunSnapshot, _ctx: RequestContext):
|
||
yield initial
|
||
|
||
|
||
class MissingQuery:
|
||
async def get(self, _run_id: str, _ctx: RequestContext) -> RunSnapshot:
|
||
raise RunNotFoundError("运行不存在或不可见")
|
||
|
||
async def watch(self, initial: RunSnapshot, _ctx: RequestContext):
|
||
yield initial
|
||
|
||
|
||
@pytest.mark.parametrize(
|
||
"accept",
|
||
[None, "", "*/*", "text/*", "text/event-stream", "text/event-stream;q=0.8",
|
||
"application/json, text/event-stream"],
|
||
)
|
||
def test_acceptable_or_absent_accept_is_allowed(accept: str | None) -> None:
|
||
assert accepts_event_stream(accept) is True
|
||
|
||
|
||
@pytest.mark.parametrize(
|
||
"accept",
|
||
["application/json", "text/html", "application/json, text/html",
|
||
"*/*;q=0", "text/event-stream;q=0", "text/*;q=0, application/json"],
|
||
)
|
||
def test_explicitly_unacceptable_accept_is_rejected(accept: str) -> None:
|
||
assert accepts_event_stream(accept) is False
|
||
|
||
|
||
@pytest.mark.parametrize("accept", [None, "*/*", "text/event-stream"])
|
||
def test_missing_wildcard_and_exact_accept_are_allowed_by_endpoint(
|
||
monkeypatch: pytest.MonkeyPatch, accept: str | None
|
||
) -> None:
|
||
headers = {} if accept is None else {"Accept": accept}
|
||
with app_with_query(monkeypatch, StubQuery()) as client:
|
||
response = client.get(EVENTS_PATH, headers=headers)
|
||
|
||
assert response.status_code == 200
|
||
assert response.headers["content-type"].startswith("text/event-stream")
|
||
|
||
|
||
def test_json_only_accept_is_rejected_with_documented_code(
|
||
monkeypatch: pytest.MonkeyPatch
|
||
) -> None:
|
||
with app_with_query(monkeypatch, StubQuery()) as client:
|
||
response = client.get(EVENTS_PATH, headers={"Accept": "application/json"})
|
||
|
||
assert response.status_code == 406
|
||
assert response.json()["error"]["code"] == "SSE_NOT_ACCEPTABLE"
|
||
assert response.json()["error"]["retryable"] is False
|
||
assert response.json()["error"]["field_errors"] == []
|
||
|
||
|
||
def test_visibility_is_checked_before_accept(
|
||
monkeypatch: pytest.MonkeyPatch
|
||
) -> None:
|
||
"""不可见运行的非法 Accept 也必须得到 `RUN_NOT_FOUND`，不泄露存在性差异。"""
|
||
cases = [{"Accept": "application/json"}, {}]
|
||
for headers in cases:
|
||
with app_with_query(monkeypatch, MissingQuery()) as client:
|
||
response = client.get(EVENTS_PATH, headers=headers)
|
||
|
||
assert response.status_code == 404
|
||
assert response.json()["error"]["code"] == "RUN_NOT_FOUND"
|
||
|
||
|
||
def test_unacceptable_accept_does_not_start_a_stream(
|
||
monkeypatch: pytest.MonkeyPatch
|
||
) -> None:
|
||
"""拒绝必须发生在响应头发送前：文档 §6.4 要求此时返回统一 JSON 错误。"""
|
||
with app_with_query(monkeypatch, StubQuery()) as client:
|
||
response = client.get(EVENTS_PATH, headers={"Accept": "application/json"})
|
||
|
||
assert response.headers["content-type"].startswith("application/json")
|