feat: 新增风险图谱算法与系统仪表盘及操作反馈体系
后端新增风险图谱算法模块、风险观察与反馈服务、规则 DSL 校验器和可解释性引擎,完善系统仪表盘和财务仪表盘统计, 优化 agent 运行和编排执行链路,清理旧开发文档,前端新增 系统趋势、负载热力图等多种仪表盘图表组件,完善操作反馈 对话框和工作台日期选择器,优化报销创建和审批详情交互, 补充单元测试覆盖。
This commit is contained in:
149
server/tests/test_risk_rule_revision_endpoints.py
Normal file
149
server/tests/test_risk_rule_revision_endpoints.py
Normal file
@@ -0,0 +1,149 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from collections.abc import Generator
|
||||
|
||||
from fastapi.testclient import TestClient
|
||||
from sqlalchemy import create_engine
|
||||
from sqlalchemy.orm import Session, sessionmaker
|
||||
from sqlalchemy.pool import StaticPool
|
||||
|
||||
from app.api.deps import get_db
|
||||
from app.core.agent_enums import AgentAssetDomain, AgentAssetStatus
|
||||
from app.db.base import Base
|
||||
from app.main import create_app
|
||||
from app.models.agent_asset import AgentAsset
|
||||
from app.schemas.agent_asset import AgentAssetRiskRuleGenerateRequest
|
||||
from app.services.agent_asset_rule_library import AgentAssetRuleLibraryManager
|
||||
from app.services.risk_rule_generation import RiskRuleGenerationService
|
||||
|
||||
|
||||
class NullRuntimeChatService:
|
||||
def complete(self, *args, **kwargs) -> None:
|
||||
return None
|
||||
|
||||
|
||||
def build_client() -> tuple[TestClient, sessionmaker[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)
|
||||
app = create_app()
|
||||
|
||||
def override_db() -> Generator[Session, None, None]:
|
||||
db = session_factory()
|
||||
try:
|
||||
yield db
|
||||
finally:
|
||||
db.close()
|
||||
|
||||
app.dependency_overrides[get_db] = override_db
|
||||
return TestClient(app), session_factory
|
||||
|
||||
|
||||
def test_update_risk_rule_draft_endpoint_updates_unpublished_rule(tmp_path) -> None:
|
||||
client, session_factory = build_client()
|
||||
asset_id = _create_rule(session_factory, tmp_path)
|
||||
|
||||
response = client.patch(
|
||||
f"/api/v1/agent-assets/{asset_id}/risk-rules/draft",
|
||||
headers=_finance_headers(),
|
||||
json={
|
||||
"rule_title": "差旅绕行说明校验",
|
||||
"expense_category": "travel",
|
||||
"natural_language": "差旅报销存在绕行但未说明原因时,进入风险复核。",
|
||||
"requires_attachment": True,
|
||||
},
|
||||
)
|
||||
|
||||
assert response.status_code == 200
|
||||
payload = response.json()
|
||||
assert payload["name"] == "差旅绕行说明校验"
|
||||
assert payload["description"] == "差旅报销存在绕行但未说明原因时,进入风险复核。"
|
||||
assert payload["scenario_json"] == ["差旅费"]
|
||||
assert payload["config_json"]["requires_attachment"] is True
|
||||
assert payload["config_json"]["generation_status"] == "draft_updated"
|
||||
assert payload["config_json"]["last_operation"]["action"] == "update_draft"
|
||||
|
||||
|
||||
def test_update_published_risk_rule_draft_endpoint_is_blocked(tmp_path) -> None:
|
||||
client, session_factory = build_client()
|
||||
asset_id = _create_rule(session_factory, tmp_path)
|
||||
_mark_rule_published(session_factory, asset_id)
|
||||
|
||||
response = client.patch(
|
||||
f"/api/v1/agent-assets/{asset_id}/risk-rules/draft",
|
||||
headers=_finance_headers(),
|
||||
json={"natural_language": "已上线规则不能被草稿接口直接覆盖。"},
|
||||
)
|
||||
|
||||
assert response.status_code == 400
|
||||
assert "未上线" in response.json()["detail"]
|
||||
|
||||
|
||||
def test_create_risk_rule_revision_endpoint_keeps_active_version(tmp_path) -> None:
|
||||
client, session_factory = build_client()
|
||||
asset_id = _create_rule(session_factory, tmp_path)
|
||||
_mark_rule_published(session_factory, asset_id)
|
||||
|
||||
response = client.post(
|
||||
f"/api/v1/agent-assets/{asset_id}/risk-rules/revisions",
|
||||
headers=_finance_headers(),
|
||||
json={
|
||||
"rule_title": "票据城市一致性复核",
|
||||
"natural_language": "票据城市与申报目的地不一致时,要求补充说明。",
|
||||
"requires_attachment": True,
|
||||
"change_reason": "补充城市一致性判断。",
|
||||
},
|
||||
)
|
||||
|
||||
assert response.status_code == 201
|
||||
payload = response.json()
|
||||
revision = payload["config_json"]["revision_draft"]
|
||||
assert payload["status"] == AgentAssetStatus.ACTIVE.value
|
||||
assert payload["published_version"] == "v0.1.0"
|
||||
assert payload["working_version"] == "v0.1.1"
|
||||
assert revision["version"] == "v0.1.1"
|
||||
assert revision["base_version"] == "v0.1.0"
|
||||
assert revision["generation_request"]["natural_language"] == "票据城市与申报目的地不一致时,要求补充说明。"
|
||||
assert payload["config_json"]["last_operation"]["action"] == "create_revision"
|
||||
|
||||
|
||||
def _create_rule(session_factory: sessionmaker[Session], tmp_path) -> str:
|
||||
with session_factory() as db:
|
||||
return RiskRuleGenerationService(
|
||||
db,
|
||||
rule_library_manager=AgentAssetRuleLibraryManager(rule_root=tmp_path / "rules"),
|
||||
runtime_chat_service=NullRuntimeChatService(),
|
||||
).generate_rule_asset(
|
||||
AgentAssetRiskRuleGenerateRequest(
|
||||
business_domain=AgentAssetDomain.EXPENSE,
|
||||
expense_category="travel",
|
||||
rule_title="差旅规则草稿",
|
||||
natural_language="差旅报销事由缺失时,提示补充说明。",
|
||||
),
|
||||
actor="pytest",
|
||||
)
|
||||
|
||||
|
||||
def _mark_rule_published(session_factory: sessionmaker[Session], asset_id: str) -> None:
|
||||
with session_factory() as db:
|
||||
asset = db.get(AgentAsset, asset_id)
|
||||
assert asset is not None
|
||||
asset.status = AgentAssetStatus.ACTIVE.value
|
||||
asset.current_version = "v0.1.0"
|
||||
asset.published_version = "v0.1.0"
|
||||
asset.working_version = "v0.1.0"
|
||||
db.add(asset)
|
||||
db.commit()
|
||||
|
||||
|
||||
def _finance_headers() -> dict[str, str]:
|
||||
return {
|
||||
"x-auth-username": "finance",
|
||||
"x-auth-name": "finance",
|
||||
"x-auth-role-codes": "finance",
|
||||
"x-actor": "finance",
|
||||
}
|
||||
Reference in New Issue
Block a user