From 7566c4dcc82b991455543e642eeffb9eaf4ff686 Mon Sep 17 00:00:00 2001 From: IgorVolochay Date: Wed, 26 Aug 2026 10:05:02 +0300 Subject: [PATCH] Some bugfix in CI --- app/tests/conftest.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/app/tests/conftest.py b/app/tests/conftest.py index af3770a..6155a3f 100644 --- a/app/tests/conftest.py +++ b/app/tests/conftest.py @@ -20,11 +20,15 @@ def _reset_all(): 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: - from main import app - sm = getattr(app.state, '_security_middleware', None) - if sm is not None: - sm.suspicious_request_counts.clear() + 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