code quality run
This commit is contained in:
@@ -1,8 +1,10 @@
|
||||
# tests/test_middleware.py
|
||||
import pytest
|
||||
from unittest.mock import Mock, patch
|
||||
from middleware.rate_limiter import RateLimiter
|
||||
|
||||
import pytest
|
||||
|
||||
from middleware.auth import AuthManager
|
||||
from middleware.rate_limiter import RateLimiter
|
||||
|
||||
|
||||
class TestRateLimiter:
|
||||
@@ -12,11 +14,17 @@ class TestRateLimiter:
|
||||
client_id = "test_client"
|
||||
|
||||
# Should allow first request
|
||||
assert limiter.allow_request(client_id, max_requests=10, window_seconds=3600) is True
|
||||
assert (
|
||||
limiter.allow_request(client_id, max_requests=10, window_seconds=3600)
|
||||
is True
|
||||
)
|
||||
|
||||
# Should allow subsequent requests within limit
|
||||
for _ in range(5):
|
||||
assert limiter.allow_request(client_id, max_requests=10, window_seconds=3600) is True
|
||||
assert (
|
||||
limiter.allow_request(client_id, max_requests=10, window_seconds=3600)
|
||||
is True
|
||||
)
|
||||
|
||||
def test_rate_limiter_blocks_excess_requests(self):
|
||||
"""Test rate limiter blocks requests exceeding limit"""
|
||||
|
||||
Reference in New Issue
Block a user