feat: 扩展风险规则体系、审批动态路由与预算中心列表化改造
- 新增 25+ 条风险规则（预算/报销/申请/通用类），完善风险规则模拟与反馈发布机制 - 引入费用审批动态路由、平台风险分级、预审与风险阶段管理 - 预算中心列表化改造，优化票据夹仪表盘与数字员工工作看板 - 新增 Hermes 风险线索收集器、Agent 链路追踪中心 - 扩展数字员工能力库（18 个领域 Skill）与交通费用自动预估 - 完善报销申请快速预览、权限控制与前端测试覆盖
This commit is contained in:
caoxiaozhu
159
server/tests/test_digital_employee_dashboard_service.py
Normal file
159
server/tests/test_digital_employee_dashboard_service.py
Normal file
@@ -0,0 +1,159 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from datetime import UTC, datetime, timedelta
|
||||
|
||||
from sqlalchemy import create_engine
|
||||
from sqlalchemy.orm import Session, sessionmaker
|
||||
from sqlalchemy.pool import StaticPool
|
||||
|
||||
from app.db.base import Base
|
||||
from app.models.agent_run import AgentRun, AgentToolCall
|
||||
from app.services.digital_employee_dashboard import DigitalEmployeeDashboardService
|
||||
|
||||
|
||||
def build_session() -> Session:
|
||||
engine = create_engine(
|
||||
"sqlite+pysqlite:///:memory:",
|
||||
connect_args={"check_same_thread": False},
|
||||
poolclass=StaticPool,
|
||||
)
|
||||
Base.metadata.create_all(bind=engine)
|
||||
session_factory = sessionmaker(bind=engine, autoflush=False, autocommit=False)
|
||||
return session_factory()
|
||||
|
||||
|
||||
def test_digital_employee_dashboard_aggregates_daily_work_from_agent_runs() -> None:
|
||||
now = datetime.now(UTC)
|
||||
|
||||
with build_session() as db:
|
||||
db.add_all(
|
||||
[
|
||||
AgentRun(
|
||||
run_id="run-digital-risk-001",
|
||||
agent="hermes",
|
||||
source="schedule",
|
||||
user_id="system",
|
||||
status="succeeded",
|
||||
route_json={"task_code": "task.hermes.global_risk_scan"},
|
||||
result_summary="财务风险图谱巡检完成。",
|
||||
started_at=now - timedelta(hours=4),
|
||||
finished_at=now - timedelta(hours=4, minutes=-3),
|
||||
tool_calls=[
|
||||
AgentToolCall(
|
||||
run_id="run-digital-risk-001",
|
||||
tool_type="rule_engine",
|
||||
tool_name="digital_employee.financial_risk_graph.scan",
|
||||
request_json={"task_type": "global_risk_scan"},
|
||||
response_json={
|
||||
"scanned_claim_count": 18,
|
||||
"risk_observation_count": 3,
|
||||
},
|
||||
status="succeeded",
|
||||
duration_ms=1200,
|
||||
created_at=now - timedelta(hours=4),
|
||||
)
|
||||
],
|
||||
),
|
||||
AgentRun(
|
||||
run_id="run-digital-clue-001",
|
||||
agent="hermes",
|
||||
source="schedule",
|
||||
user_id="system",
|
||||
status="failed",
|
||||
route_json={"report_type": "risk_clue_collect"},
|
||||
result_summary="风险线索归集失败。",
|
||||
started_at=now - timedelta(hours=3),
|
||||
finished_at=now - timedelta(hours=3, minutes=-1),
|
||||
tool_calls=[
|
||||
AgentToolCall(
|
||||
run_id="run-digital-clue-001",
|
||||
tool_type="database",
|
||||
tool_name="digital_employee.risk_clue.collect",
|
||||
request_json={"task_type": "risk_clue_collect"},
|
||||
response_json={
|
||||
"fact_count": 12,
|
||||
"rule_hit_count": 5,
|
||||
"risk_clue_count": 2,
|
||||
},
|
||||
status="failed",
|
||||
duration_ms=800,
|
||||
error_message="collector failed",
|
||||
created_at=now - timedelta(hours=3),
|
||||
)
|
||||
],
|
||||
),
|
||||
AgentRun(
|
||||
run_id="run-digital-knowledge-001",
|
||||
agent="hermes",
|
||||
source="user_message",
|
||||
user_id="admin",
|
||||
status="running",
|
||||
route_json={
|
||||
"job_type": "knowledge_index_sync",
|
||||
"requested_document_ids": ["doc-1", "doc-2"],
|
||||
},
|
||||
result_summary="知识归纳任务已入队。",
|
||||
started_at=now - timedelta(hours=1),
|
||||
),
|
||||
AgentRun(
|
||||
run_id="run-user-001",
|
||||
agent="user_agent",
|
||||
source="user_message",
|
||||
user_id="employee",
|
||||
status="succeeded",
|
||||
result_summary="普通报销预审。",
|
||||
started_at=now - timedelta(hours=2),
|
||||
finished_at=now - timedelta(hours=2, minutes=-1),
|
||||
tool_calls=[
|
||||
AgentToolCall(
|
||||
run_id="run-user-001",
|
||||
tool_type="llm",
|
||||
tool_name="expense_claim.review",
|
||||
request_json={},
|
||||
response_json={},
|
||||
status="succeeded",
|
||||
duration_ms=500,
|
||||
created_at=now - timedelta(hours=2),
|
||||
)
|
||||
],
|
||||
),
|
||||
]
|
||||
)
|
||||
db.commit()
|
||||
|
||||
dashboard = DigitalEmployeeDashboardService(db).build_dashboard(days=7)
|
||||
|
||||
assert dashboard.has_real_data is True
|
||||
assert dashboard.totals["totalRuns"] == 3
|
||||
assert dashboard.totals["successRuns"] == 1
|
||||
assert dashboard.totals["failedRuns"] == 1
|
||||
assert dashboard.totals["runningRuns"] == 1
|
||||
assert dashboard.totals["toolCalls"] == 2
|
||||
assert dashboard.totals["riskObservations"] == 3
|
||||
assert dashboard.totals["riskClues"] == 2
|
||||
assert dashboard.totals["knowledgeDocuments"] == 2
|
||||
assert dashboard.totals["businessOutputs"] == 7
|
||||
assert dashboard.totals["successRate"] == 33.3
|
||||
|
||||
category_counts = {item["name"]: item["count"] for item in dashboard.category_distribution}
|
||||
assert category_counts["评估"] == 1
|
||||
assert category_counts["升级"] == 1
|
||||
assert category_counts["整理"] == 1
|
||||
assert category_counts["积累"] == 0
|
||||
|
||||
task_names = {item["name"] for item in dashboard.task_distribution}
|
||||
assert task_names == {"财务风险图谱巡检", "风险线索归集", "知识制度整理"}
|
||||
|
||||
assert sum(item["total"] for item in dashboard.daily_work) == 3
|
||||
assert dashboard.recent_runs[0]["runId"] == "run-digital-knowledge-001"
|
||||
assert dashboard.recent_runs[0]["statusLabel"] == "运行中"
|
||||
|
||||
|
||||
def test_digital_employee_dashboard_keeps_empty_payload_without_fake_data() -> None:
|
||||
with build_session() as db:
|
||||
dashboard = DigitalEmployeeDashboardService(db).build_dashboard(days=7)
|
||||
|
||||
assert dashboard.has_real_data is False
|
||||
assert dashboard.totals["totalRuns"] == 0
|
||||
assert dashboard.daily_work
|
||||
assert dashboard.task_distribution == []
|
||||
Reference in New Issue
Block a user