feat(v2): T7 网关扩展 + T9 token 计量与账单(/chat v2,/chat/legacy,/runs,/review,/metrics)
This commit is contained in:
+41
-3
@@ -1,4 +1,4 @@
|
||||
"""???????? fastapi + httpx??"""
|
||||
"""FastAPI 网关测试:v1 legacy 端点保持 + v2 端点(封闭,注入 mock 管线)。"""
|
||||
import pytest
|
||||
|
||||
pytest.importorskip("fastapi")
|
||||
@@ -6,6 +6,7 @@ pytest.importorskip("httpx")
|
||||
|
||||
from fastapi.testclient import TestClient
|
||||
|
||||
import gateway.api as ga
|
||||
from gateway.api import app
|
||||
|
||||
|
||||
@@ -14,6 +15,14 @@ def client():
|
||||
return TestClient(app)
|
||||
|
||||
|
||||
@pytest.fixture()
|
||||
def v2_client(client):
|
||||
# 用 mock worker 构建真实 v2 管线并注入(无需 API key / 真实模型)
|
||||
pipe = ga.build_v2_pipeline(worker_cfg_override={"backend": "mock"})
|
||||
ga.set_pipeline(pipe)
|
||||
return client
|
||||
|
||||
|
||||
def test_health(client):
|
||||
resp = client.get("/health")
|
||||
assert resp.status_code == 200
|
||||
@@ -22,8 +31,8 @@ def test_health(client):
|
||||
assert "code" in data["domains"]
|
||||
|
||||
|
||||
def test_chat(client):
|
||||
resp = client.post("/chat", json={"query": "? Python ???????"})
|
||||
def test_chat_legacy(client):
|
||||
resp = client.post("/chat/legacy", json={"query": "用 Python 写一个快速排序函数"})
|
||||
assert resp.status_code == 200
|
||||
data = resp.json()
|
||||
assert data["response"]
|
||||
@@ -31,6 +40,17 @@ def test_chat(client):
|
||||
assert "route" in data
|
||||
|
||||
|
||||
def test_chat_v2(v2_client):
|
||||
resp = v2_client.post("/chat", json={"query": "请介绍快速排序算法"})
|
||||
assert resp.status_code == 200
|
||||
data = resp.json()
|
||||
assert data["response"]
|
||||
assert data["request_id"]
|
||||
assert "fast_path" in data
|
||||
assert "route" in data
|
||||
assert data["status"] in ("fast_path", "done", "escalated", "failed")
|
||||
|
||||
|
||||
def test_chat_empty_query(client):
|
||||
resp = client.post("/chat", json={"query": ""})
|
||||
assert resp.status_code == 422
|
||||
@@ -42,3 +62,21 @@ def test_metrics(client):
|
||||
data = resp.json()
|
||||
assert "router" in data
|
||||
assert "cache" in data
|
||||
assert "v2" in data
|
||||
assert "review" in data
|
||||
|
||||
|
||||
def test_workspace_not_found(client):
|
||||
resp = client.get("/runs/nonexistent/workspace")
|
||||
assert resp.status_code == 404
|
||||
|
||||
|
||||
def test_review_flow(client):
|
||||
q = ga.get_review()
|
||||
rid = q.enqueue("req-x", "q", "ans", tags=["safety"], reason="test")
|
||||
assert q.count() >= 1
|
||||
resp = client.get("/review/queue")
|
||||
assert resp.status_code == 200
|
||||
resp2 = client.post(f"/review/{rid}", params={"verdict": "approve"})
|
||||
assert resp2.status_code == 200
|
||||
assert resp2.json()["ok"] is True
|
||||
|
||||
Reference in New Issue
Block a user