Files
group_fqcd_jr/tests/unit/infrastructure/test_rate_limiter.py
T
lzf_0626 6516ccb385 feat: 第二版——接口契约对齐 docs/05,修复静默故障与数据库基线
相对第一版 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(含失败关闭反证)。
2026-09-10 15:55:54 +08:00

100 lines
3.1 KiB
Python

"""限流计数后端契约测试(不连 Redis)。
锁定两条必须成立的语义:
1. 正常路径:窗口内计数递增,并把"剩余窗口秒数"作为 `Retry-After` 的来源返回;
键首次创建(`TTL` 为 -1)时补一次过期,避免出现永不过期的脏计数键。
2. **降级路径**:Redis 构造失败或命令失败一律返回 `None`(=无法判定),由调用方放行。
限流是保护措施,不能因为 Redis 故障把正常请求全部拒掉——这条如果回归,故障时
整个平台会 429 全灭。
"""
from typing import Any
from app.infrastructure.rate_limiter import RedisCounterBackend
class FakePipeline:
def __init__(self, client: "FakeRedis") -> None:
self._client = client
async def __aenter__(self) -> "FakePipeline":
return self
async def __aexit__(self, *exc: object) -> bool:
return False
def incr(self, key: str) -> "FakePipeline":
self._client.incr_keys.append(key)
return self
def ttl(self, key: str) -> "FakePipeline":
self._client.ttl_keys.append(key)
return self
async def execute(self) -> list[Any]:
if self._client.fail:
raise ConnectionError("redis unavailable")
self._client.count += 1
return [self._client.count, self._client.ttl_value]
class FakeRedis:
def __init__(self, *, fail: bool = False, ttl_value: int = 42) -> None:
self.fail = fail
self.ttl_value = ttl_value
self.count = 0
self.incr_keys: list[str] = []
self.ttl_keys: list[str] = []
self.expired: list[tuple[str, int]] = []
def pipeline(self, transaction: bool = False) -> FakePipeline:
del transaction
return FakePipeline(self)
async def expire(self, key: str, seconds: int) -> bool:
self.expired.append((key, seconds))
return True
def backend(client: Any) -> RedisCounterBackend:
return RedisCounterBackend(
"redis://unused", connect_timeout_seconds=0.1, client_factory=lambda: client
)
async def test_first_request_counts_one_and_reports_remaining_window() -> None:
client = FakeRedis(ttl_value=42)
result = await backend(client).increment("k", 60)
assert result == (1, 42)
assert client.incr_keys == ["k"]
async def test_missing_ttl_is_repaired_with_window_expiry() -> None:
"""`TTL` 为 -1 说明键没有过期时间:必须补一次,否则计数永远不归零。"""
client = FakeRedis(ttl_value=-1)
result = await backend(client).increment("k", 60)
assert result == (1, 60)
assert client.expired == [("k", 60)]
async def test_redis_command_failure_degrades_to_unknown() -> None:
client = FakeRedis(fail=True)
assert await backend(client).increment("k", 60) is None
async def test_client_construction_failure_degrades_to_unknown() -> None:
def exploding_factory() -> Any:
raise RuntimeError("redis not installed")
instance = RedisCounterBackend(
"redis://unused", connect_timeout_seconds=0.1, client_factory=exploding_factory
)
assert await instance.increment("k", 60) is None