78 lines
3.0 KiB
Python
78 lines
3.0 KiB
Python
"""真实 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
|