selfrelease
fad458b2a7
- UIUX 文档：填充 19 个缺口（多主体画像/健康度/AI+看板/增长域/洞察域/创始人端/OODA/助推/商密） - UIUX 文档：插入 6 个新章节（十四~十九），旧章节重编号为二十~三十一，更新目录和交叉引用 - 作业指导书 x5：导航改为 6 域分组，新增 Context Bar/工作模式/Insight Rail/决策线程/多工作区等 UI 概念 - 新建 docs/2-task-uiux.md：50 个代码落地开发任务，按 P0-P6 分优先级 + 8 Sprint 规划 - 后端/前端：大量新增模型、路由、组件（来自之前 Phase 开发）
59 lines
2.2 KiB
Python
59 lines
2.2 KiB
Python
"""T4.9 RAG 知识库 + 语义搜索测试。"""
|
|
|
|
import pytest
|
|
from fastapi.testclient import TestClient
|
|
|
|
|
|
class TestKnowledgeSearch:
|
|
"""语义搜索接口测试。"""
|
|
|
|
def test_search_empty_db(self, client: TestClient, auth_headers: dict):
|
|
"""知识库为空时应返回空列表。"""
|
|
resp = client.get("/api/v1/knowledge/search?q=跑道不足", headers=auth_headers)
|
|
assert resp.status_code == 200
|
|
assert isinstance(resp.json()["data"], list)
|
|
|
|
def test_search_with_top_k(self, client: TestClient, auth_headers: dict):
|
|
"""指定 top_k 参数。"""
|
|
resp = client.get("/api/v1/knowledge/search?q=融资&top_k=3", headers=auth_headers)
|
|
assert resp.status_code == 200
|
|
assert isinstance(resp.json()["data"], list)
|
|
|
|
def test_search_no_auth(self, client: TestClient):
|
|
"""未认证应返回 401。"""
|
|
resp = client.get("/api/v1/knowledge/search?q=test")
|
|
assert resp.status_code == 401
|
|
|
|
def test_search_empty_query(self, client: TestClient, auth_headers: dict):
|
|
"""空查询应返回 422。"""
|
|
resp = client.get("/api/v1/knowledge/search?q=", headers=auth_headers)
|
|
assert resp.status_code == 422
|
|
|
|
|
|
class TestKnowledgeContext:
|
|
"""RAG 上下文构建接口测试。"""
|
|
|
|
def test_build_context_success(self, client: TestClient, auth_headers: dict):
|
|
"""构建 RAG 上下文。"""
|
|
resp = client.post(
|
|
"/api/v1/knowledge/context",
|
|
json={"query": "企业跑道不足的风险"},
|
|
headers=auth_headers,
|
|
)
|
|
assert resp.status_code == 200
|
|
assert isinstance(resp.json()["data"], str)
|
|
|
|
def test_build_context_no_auth(self, client: TestClient):
|
|
"""未认证应返回 401。"""
|
|
resp = client.post("/api/v1/knowledge/context", json={})
|
|
assert resp.status_code == 401
|
|
|
|
def test_build_context_empty_query(self, client: TestClient, auth_headers: dict):
|
|
"""空查询应正常返回空字符串。"""
|
|
resp = client.post(
|
|
"/api/v1/knowledge/context",
|
|
json={"query": ""},
|
|
headers=auth_headers,
|
|
)
|
|
assert resp.status_code == 200
|