ceos/fastmcp
1
0
Fork 0
mirror of https://github.com/PrefectHQ/fastmcp.git synced 2026-08-22 21:44:18 +02:00
Code Issues Projects Releases Packages Wiki Activity

style: apply ruff format to rate limiting LRU tests

Browse source 
🤖 Generated with Claude Code

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
This commit is contained in:
strawgate 2026-05-17 10:28:14 -05:00
commit 3f31f157cf
1 changed files with 4 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

16
tests/server/middleware/test_rate_limiting.py
View file

@ -274,9 +274,7 @@ class TestRateLimiterLRUEviction:
"""LRU bounding of per-client limiters (issue #4053)."""
def test_token_bucket_evicts_lru_when_full(self):
mw = RateLimitingMiddleware(
get_client_id=lambda ctx: ctx, max_clients=2
)
mw = RateLimitingMiddleware(get_client_id=lambda ctx: ctx, max_clients=2)
mw._get_limiter("a")
mw._get_limiter("b")
# Cache full at 2; adding "c" evicts the LRU ("a"), not the newcomer.
@ -284,9 +282,7 @@ class TestRateLimiterLRUEviction:
assert list(mw._client_limiters) == ["b", "c"]
def test_token_bucket_move_to_end_on_access_protects_active_client(self):
mw = RateLimitingMiddleware(
get_client_id=lambda ctx: ctx, max_clients=2
)
mw = RateLimitingMiddleware(get_client_id=lambda ctx: ctx, max_clients=2)
mw._get_limiter("a")
mw._get_limiter("b")
# Touch "a" so it is most-recently-used; "b" is now LRU.
@ -300,9 +296,7 @@ class TestRateLimiterLRUEviction:
assert mw._get_limiter("a") is first
def test_token_bucket_bound_never_exceeds_max_clients(self):
mw = RateLimitingMiddleware(
get_client_id=lambda ctx: ctx, max_clients=5
)
mw = RateLimitingMiddleware(get_client_id=lambda ctx: ctx, max_clients=5)
for i in range(100):
mw._get_limiter(f"client-{i}")
assert len(mw._client_limiters) == 5
@ -337,9 +331,7 @@ class TestRateLimiterLRUEviction:
SlidingWindowRateLimitingMiddleware(max_requests=10, max_clients=bad)
def test_max_clients_one_keeps_only_newest(self):
mw = RateLimitingMiddleware(
get_client_id=lambda ctx: ctx, max_clients=1
)
mw = RateLimitingMiddleware(get_client_id=lambda ctx: ctx, max_clients=1)
mw._get_limiter("a")
mw._get_limiter("b")
assert list(mw._client_limiters) == ["b"]