Guest User

Untitled

a guest
Mar 25th, 2026
15
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.88 KB | None | 0 0
  1. MY LIMITER METHODS ------------------------------------
  2. async def ip_identifier(request: Request):
  3. forwarded = request.headers.get("X-Forwarded-For")
  4. if forwarded:
  5. return forwarded.split(",")[0]
  6. return f"{request.client.host}:shared-limit"
  7.  
  8. def get_shared_limiter():
  9. return shared_limiter
  10.  
  11. def get_rate_limiter() -> RateLimiter:
  12. return RateLimiter(
  13. limiter=get_shared_limiter(),
  14. identifier=ip_identifier
  15. )
  16.  
  17. MY APP LIMITER SETUP ----------------------------------
  18. app = FastAPI(
  19. dependencies=[
  20. Depends(
  21. get_rate_limiter()
  22. )
  23. ],
  24. )
  25.  
  26. MY CURRENT TEST SETUP (DOES NOT WORK) -----------------
  27. @pytest.fixture(autouse=True, scope="session")
  28. def mock_limiter():
  29. app.dependency_overrides[get_rate_limiter] = lambda: RateLimiter(limiter=Limiter(Rate(100000, Duration.SECOND)), identifier=ip_identifier)
Advertisement
Add Comment
Please, Sign In to add comment