feat(server): add workbench API, DSH planner worker and RAG pipeline

server/tests/test_readiness.py

@@ -0,0 +1,114 @@
+"""准备度检查的单元测试（不依赖网络与真实 Provider）。"""
+from types import SimpleNamespace
+from unittest.mock import patch
+
+from app.readiness import readiness_report
+
+
+def _campaign(cid=1, owner=7, org=1):
+    return SimpleNamespace(id=cid, owner_user_id=owner, organization_id=org)
+
+
+def _segment(agent_id=11):
+    return SimpleNamespace(agent_id=agent_id)
+
+
+def _agent(id=11, name="开场段 Agent", role="production", status="active", kb_ids=(1,)):
+    return SimpleNamespace(id=id, name=name, role=role, status=status, kb_ids=list(kb_ids))
+
+
+class _Query:
+    def __init__(self, rows):
+        self.rows = rows
+
+    def all(self):
+        return self.rows
+
+
+def _session_with(segments, agents_by_id, qc_rows, composer_rows):
+    """构造最小 session mock：get 按 Agent 精确命中，exec 按调用顺序返回。"""
+    plan = [_Query(segments), _Query(qc_rows), _Query(composer_rows)]
+    # kb 查询按需追加
+    state = {"plan_index": 0}
+
+    def exec(*_a, **_k):
+        query = plan[min(state["plan_index"], len(plan) - 1)]
+        state["plan_index"] += 1
+        return query
+
+    def get(model, row_id):
+        if model.__name__ == "Agent":
+            return agents_by_id.get(row_id)
+        if model.__name__ == "KnowledgeBase":
+            return SimpleNamespace(id=row_id, name="测试库", index_status="ready")
+        return None
+
+    return SimpleNamespace(exec=exec, get=get)
+
+
+POINTS_OK = {"points": 999999, "token_equivalent": 99999, "points_per_token": 10}
+
+
+def test_no_segments_blocks():
+    session = _session_with([], {}, [], [])
+    report = readiness_report(session, _campaign())
+    assert report["ready"] is False
+    assert any("拆解" in b for b in report["blockers"])
+
+
+def test_disabled_agent_blocks():
+    agent = _agent(status="disabled")
+    session = _session_with([_segment()], {11: agent}, [_agent(21, "质检", "qc")], [_agent(22, "组合", "composer")])
+    with patch("app.readiness.point_state", return_value=POINTS_OK):
+        report = readiness_report(session, _campaign())
+    assert any("停用" in b for b in report["blockers"])
+
+
+def test_missing_kb_blocks():
+    agent = _agent(kb_ids=())
+    session = _session_with([_segment()], {11: agent}, [_agent(21, "质检", "qc")], [_agent(22, "组合", "composer")])
+    with patch("app.readiness.point_state", return_value=POINTS_OK), \
+         patch("app.readiness._provider_ready", return_value=(True, "")):
+        report = readiness_report(session, _campaign())
+    assert any("未挂接知识库" in b for b in report["blockers"])
+
+
+def test_low_points_warns_not_blocks():
+    agent = _agent()
+    session = _session_with([_segment()], {11: agent}, [_agent(21, "质检", "qc")], [_agent(22, "组合", "composer")])
+    with patch("app.readiness.point_state", return_value={"points": 5000, "token_equivalent": 500, "points_per_token": 10}), \
+         patch("app.readiness._provider_ready", return_value=(True, "")):
+        report = readiness_report(session, _campaign())
+    assert report["ready"] is True
+    assert any("点数偏低" in w for w in report["warnings"])
+
+
+def test_zero_points_blocks():
+    agent = _agent()
+    session = _session_with([_segment()], {11: agent}, [_agent(21, "质检", "qc")], [_agent(22, "组合", "composer")])
+    with patch("app.readiness.point_state", return_value={"points": 0, "token_equivalent": 0, "points_per_token": 10}), \
+         patch("app.readiness._provider_ready", return_value=(True, "")):
+        report = readiness_report(session, _campaign())
+    assert report["ready"] is False
+    assert any("点数余额不足" in b for b in report["blockers"])
+
+
+def test_missing_qc_composer_blocks():
+    agent = _agent()
+    session = _session_with([_segment()], {11: agent}, [], [])
+    with patch("app.readiness.point_state", return_value=POINTS_OK), \
+         patch("app.readiness._provider_ready", return_value=(True, "")):
+        report = readiness_report(session, _campaign())
+    assert any("质检 Agent" in b for b in report["blockers"])
+    assert any("组合 Agent" in b for b in report["blockers"])
+
+
+def test_disabled_qc_blocks():
+    agent = _agent()
+    session = _session_with([_segment()], {11: agent},
+                            [_agent(21, "质检", "qc", status="disabled")],
+                            [_agent(22, "组合", "composer")])
+    with patch("app.readiness.point_state", return_value=POINTS_OK), \
+         patch("app.readiness._provider_ready", return_value=(True, "")):
+        report = readiness_report(session, _campaign())
+    assert any("质检 Agent" in b for b in report["blockers"])