Files
group_fqcd_jr/tests/integration/test_rate_limit_redis.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

78 lines
3.0 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters
This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.
"""真实 Redis 上的限流验证(文档 §3.5 的 429 / §3.6 的 `RATE_LIMITED`)。
替身测试能证明闸门逻辑,但证明不了"Redis 计数路径真的在跑":`INCR`/`TTL` 的写法、
pipeline 用法、键的过期都可能只在真实 Redis 上暴露。本用例用**真实后端**(不注入
替身)走一遍 HTTP:
- Redis 不可用时 `skip`(此时按设计降级放行,429 不会发生,测下去只会假失败);
- 计数键前缀带随机串,避免与并行进程的计数互相污染;
- 阈值临时设为 1,第二个请求必须 429 且带 `Retry-After`。
不连数据库:会话消息查询对不存在的会话返回空列表。
"""
from uuid import uuid4
import httpx
import pytest
from app.api.dependencies.auth import build_request_context
from app.core.config import get_settings
from app.core.contracts import RequestContext
from app.main import create_app
pytestmark = pytest.mark.integration
PATH = f"/api/v1/conversations/rate-limit-{uuid4()}/messages"
async def redis_is_available() -> bool:
try:
from redis.asyncio import Redis
settings = get_settings()
client = Redis.from_url(settings.redis_url,
socket_connect_timeout=settings.redis_connect_timeout_seconds)
try:
return bool(await client.ping())
finally:
await client.aclose()
except Exception:
return False
async def test_real_redis_counter_returns_429_with_retry_after(
monkeypatch: pytest.MonkeyPatch,
) -> None:
if not await redis_is_available():
pytest.skip("Redis 不可用:按设计限流降级放行,无法在此环境验证 429")
settings = get_settings()
# 前缀必须在 lambda 外算好:`get_settings()` 每次请求都会被调用,写在 lambda 里
# 会让每个请求落到不同的计数键上,限流永远不触发(这正是本用例要防住的错误)。
prefix = f"test:rate_limit:{uuid4().hex}"
monkeypatch.setattr(
"app.api.dependencies.rate_limit.get_settings",
lambda: settings.model_copy(update={
"rate_limit_enabled": True,
"rate_limit_window_seconds": 60,
"rate_limit_max_requests": 1,
"rate_limit_key_prefix": prefix,
}),
)
context = RequestContext(user_id=str(uuid4().int % 10**6), trace_id="trace-redis-limit",
roles=("customer",), permissions=())
application = create_app()
application.dependency_overrides[build_request_context] = lambda: context
async with httpx.AsyncClient(transport=httpx.ASGITransport(app=application),
base_url="http://test") as client:
first = await client.get(PATH)
second = await client.get(PATH)
assert first.status_code == 200, first.text
assert second.status_code == 429, second.text
assert second.json()["error"]["code"] == "RATE_LIMITED"
assert second.json()["error"]["retryable"] is True
assert int(second.headers["Retry-After"]) >= 1