python/slowapi
1
0
Fork 0
mirror of https://github.com/laurentS/slowapi.git synced 2026-03-13 09:10:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

Resolving bug of unregistered endpoints in the disabled state.

Browse Source
This commit is contained in:
Trevor Currie
2021-04-05 15:38:18 -07:00
parent 4561d83f98
commit 21652645ef
2 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
slowapi/middleware.py
View File

@@ -51,6 +51,7 @@ class SlowAPIMiddleware(BaseHTTPMiddleware):
return exception_handler(request, e)
# request.state._rate_limiting_complete = True
response = await call_next(request)
response = limiter._inject_headers(response, request.state.view_rate_limit)
if limiter.enabled:
response = limiter._inject_headers(response, request.state.view_rate_limit)
return response
return await call_next(request)

8
tests/test_fastapi_extension.py
View File

@@ -223,6 +223,10 @@ class TestDecorators(TestSlowapi):
def t2(request: Request):
return PlainTextResponse("test")
@app.get("/t3")
def t3(request: Request):
return PlainTextResponse("also a test")
client = TestClient(app)
for i in range(0, 10):
response = client.get("/t1")
@@ -231,3 +235,7 @@ class TestDecorators(TestSlowapi):
for i in range(0, 10):
response = client.get("/t2")
assert response.status_code == 200
for i in range(0, 10):
response = client.get("/t3")
assert response.status_code == 200