opensquilla--opensquilla
wehub-resource-sync
33 行
1.1 KiB
Python
33 行
1.1 KiB
Python
from __future__ import annotations
|
|
|
|
from starlette.applications import Starlette
|
|
from starlette.responses import JSONResponse
|
|
from starlette.testclient import TestClient
|
|
|
|
from opensquilla.gateway.config import GatewayConfig
|
|
from opensquilla.gateway.middleware import RateLimitMiddleware
|
|
|
|
|
|
def test_approval_polling_does_not_consume_generic_api_rate_limit() -> None:
|
|
app = Starlette()
|
|
|
|
async def approvals(_request):
|
|
return JSONResponse({"approvals": []})
|
|
|
|
async def sessions(_request):
|
|
return JSONResponse({"ok": True})
|
|
|
|
app.add_route("/api/approvals", approvals, methods=["GET"])
|
|
app.add_route("/api/sessions", sessions, methods=["GET"])
|
|
config = GatewayConfig()
|
|
config.rate_limit.enabled = True
|
|
config.rate_limit.max_requests = 1
|
|
config.rate_limit.window_seconds = 60
|
|
app.add_middleware(RateLimitMiddleware, config=config)
|
|
|
|
with TestClient(app) as client:
|
|
assert client.get("/api/approvals").status_code == 200
|
|
assert client.get("/api/approvals").status_code == 200
|
|
assert client.get("/api/sessions").status_code == 200
|
|
assert client.get("/api/sessions").status_code == 429
|