lzf_0626
6516ccb385
相对第一版 46fc976 的完整变更。组员迁移对照表见 docs/20。
一、对外契约对齐 docs/05（破坏性，共 4 处，组员需按 docs/20 调整）
1) 配置发布端点改为文档规定的复数资源名：submit→validations、
approve→reviews（需 body decision）、activate→activations、
rollback→rollbacks；第一版这 4 个动词式路径 docs/05 从未定义过。
2) 错误码由 8 个笼统码改为 15 个具体语义码（FORBIDDEN→AGENT_PERMISSION_DENIED、
UNAUTHORIZED→AUTHENTICATION_REQUIRED、CONFLICT→RESOURCE_VERSION_CONFLICT、
RESOURCE_NOT_FOUND→RUN_NOT_FOUND/SESSION_NOT_FOUND 等），
输入类错误状态码 400→422。
3) POST /api/v1/agent-runs 与 GET /api/v1/agent-runs/{run_id} 统一为
{data, meta} 信封（data 内字段名与语义未变）。
4) 错误响应体统一为 {error:{code,message,retryable,field_errors}, meta:{trace_id}}，
不再返回 FastAPI 默认的 {"detail": ...}。
二、数据库基线与约束
新增 39 张表的基线迁移（链根）与联合唯一键纠偏（4 张表、删 8 增 4，幂等收敛）；
撤下 config_release 的双人复核 CHECK（应用层已允许自审，审核节点保留，
自审如实写入 reviewer_id）；记忆 active key 生成列与唯一键；
activate 开始记录 supersedes_release_id 使版本链可追溯。
docs/00 基线未修改，未重命名或删除任何表与字段。
三、修复会静默出错或无报错的缺陷
- 跑完集成测试后平台会静默失去生效配置：清理只删自己创建的版本，却没有恢复被它
顶成 superseded 的原生效版本，且审计一并删除因而完全无痕，表现为所有工具被拒
但没有任何报错。已修清理逻辑并加恢复。
- Worker 单轮异常导致进程退出；记忆抽取调用方的“事务已开始”异常；
召回缓存丢失 degraded 标记；连接时区未生效导致 created_at/updated_at 差 8 小时；
.env 与 os.getenv 密钥来源分裂导致“没有可用的已批准模型端点”。
- 记忆信号识别漏判与跨键误命中；SSE 未带 Accept 的协商行为。
四、功能补齐
记忆链路 P1/P2/P3（抽取、受控词表、召回与缓存、生命周期级联及投影事件）、
fin_* 场内交易只读 ORM 层、agent_intent_config 状态流转并在运行期真正生效、
限流（Redis 固定窗口、故障一律放行）、游标校验、trace_id 中间件、
示例业务 Agent fund_query_demo 与一键端到端验证脚本，以及审计/指纹/迁移状态工具。
五、文档与验证
新增 docs/19（业务 Agent 接入实操）、docs/20（第一版迁移指南）与 docs/evidence 证据；
docs/01/02/06/08/09/17 同步实现现状。
验证结果：ruff 通过、mypy 103 文件无错、unit+contract 447 passed、
integration 29 passed、acceptance_check --production 7 PASS、
demo_agent_e2e 9/9 PASS（含失败关闭反证）。
107 lines
3.7 KiB
Python
107 lines
3.7 KiB
Python
"""就绪探针（B3）单元测试：Milvus 可用/不可用/客户端缺失/探测超时。"""
|
|
|
|
import asyncio
|
|
from typing import Any
|
|
|
|
import pytest
|
|
|
|
from app.service import health_service
|
|
from app.service.health_service import HealthService
|
|
|
|
|
|
class FakeMilvusClient:
|
|
def __init__(self, *, version: Any = "pkg/v2.5.4", error: Exception | None = None,
|
|
delay: float = 0.0) -> None:
|
|
self._version = version
|
|
self._error = error
|
|
self._delay = delay
|
|
self.closed = False
|
|
|
|
async def get_server_version(self) -> Any:
|
|
if self._delay:
|
|
await asyncio.sleep(self._delay)
|
|
if self._error is not None:
|
|
raise self._error
|
|
return self._version
|
|
|
|
async def close(self) -> None:
|
|
self.closed = True
|
|
|
|
|
|
def patch_client(monkeypatch: pytest.MonkeyPatch, client: Any) -> None:
|
|
monkeypatch.setattr(health_service, "_create_milvus_client", lambda uri, token: client)
|
|
|
|
|
|
async def test_available_milvus_reports_ok(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
client = FakeMilvusClient()
|
|
patch_client(monkeypatch, client)
|
|
ready, state = await HealthService()._milvus_ready()
|
|
assert (ready, state) == (True, "ok")
|
|
assert client.closed is True
|
|
|
|
|
|
@pytest.mark.parametrize("error", [ConnectionError("refused"), RuntimeError("grpc down")])
|
|
async def test_unavailable_milvus_reports_unavailable(
|
|
monkeypatch: pytest.MonkeyPatch, error: Exception
|
|
) -> None:
|
|
client = FakeMilvusClient(error=error)
|
|
patch_client(monkeypatch, client)
|
|
ready, state = await HealthService()._milvus_ready()
|
|
assert (ready, state) == (False, "unavailable")
|
|
assert client.closed is True
|
|
|
|
|
|
async def test_missing_pymilvus_reports_client_missing(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
def missing(uri: str, token: str) -> Any:
|
|
raise ImportError("No module named 'pymilvus'")
|
|
|
|
monkeypatch.setattr(health_service, "_create_milvus_client", missing)
|
|
ready, state = await HealthService()._milvus_ready()
|
|
assert (ready, state) == (False, "client_missing")
|
|
|
|
|
|
async def test_probe_timeout_is_bounded(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
client = FakeMilvusClient(delay=5.0)
|
|
patch_client(monkeypatch, client)
|
|
ready, state = await HealthService(milvus_timeout_seconds=0.05)._milvus_ready()
|
|
assert (ready, state) == (False, "timeout")
|
|
|
|
|
|
async def test_ready_reflects_real_milvus_probe(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
client = FakeMilvusClient(error=ConnectionError("milvus stopped"))
|
|
patch_client(monkeypatch, client)
|
|
|
|
async def redis_ok(_self: HealthService) -> bool:
|
|
return True
|
|
|
|
monkeypatch.setattr(HealthService, "_redis_ready", redis_ok)
|
|
result = await HealthService().ready()
|
|
assert result["checks"]["milvus"] is False
|
|
assert result["probes"]["milvus"] == "unavailable"
|
|
assert result["status"] == "degraded"
|
|
|
|
|
|
async def test_ready_is_ready_when_all_probes_pass(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
patch_client(monkeypatch, FakeMilvusClient())
|
|
|
|
async def redis_ok(_self: HealthService) -> bool:
|
|
return True
|
|
|
|
monkeypatch.setattr(HealthService, "_redis_ready", redis_ok)
|
|
result = await HealthService().ready()
|
|
assert result["checks"]["milvus"] is True
|
|
assert result["probes"]["milvus"] == "ok"
|
|
assert result["checks"]["mysql"] is True
|
|
|
|
|
|
async def test_redis_failure_marks_readiness_degraded(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
async def unavailable(self: HealthService) -> bool:
|
|
return False
|
|
|
|
monkeypatch.setattr(HealthService, "_redis_ready", unavailable)
|
|
patch_client(monkeypatch, FakeMilvusClient())
|
|
service = HealthService()
|
|
result = await service.ready()
|
|
assert result["checks"]["redis"] is False
|
|
assert result["status"] == "degraded"
|