Merge pull request #4 from IgorVolochay/frontend

app/tests/conftest.py

@@ -0,0 +1,44 @@
+"""
+conftest.py — fixtures for resetting rate-limiter state and IP-ban
+between tests so functional tests do not hit 429 errors.
+"""
+
+import pytest
+from guard import ip_ban_manager
+from guard_core.handlers.ratelimit_handler import RateLimitManager
+
+
+def _reset_all():
+    """Full reset of rate-limiter, IP-ban, and suspicious counts."""
+    # Rate limit timestamps
+    rl: RateLimitManager | None = RateLimitManager._instance
+    if rl is not None:
+        rl.request_timestamps.clear()
+
+    # IP bans
+    ip_ban_manager.banned_ips.clear()
+    ip_ban_manager.banned_networks.clear()
+
+    # Suspicious request counts via direct reference stored in app.state
+    # Because FastAPI's add_middleware creates a new instance internally,
+    # navigating app.state or app.middleware_stack is unreliable.
+    # We use gc to robustly find the active SecurityMiddleware instance(s) and clear them.
+    try:
+        import gc
+        from guard.middleware import SecurityMiddleware
+        for obj in gc.get_objects():
+            if isinstance(obj, SecurityMiddleware):
+                obj.suspicious_request_counts.clear()
+    except Exception:
+        pass
+
+
+@pytest.fixture(autouse=True)
+def reset_guard_state():
+    """
+    Synchronous fixture (autouse) that resets guard middleware
+    state before and after each test.
+    """
+    _reset_all()
+    yield
+    _reset_all()