Some bugfix in CI

This commit is contained in:
IgorVolochay
2026-08-26 10:05:02 +03:00
parent f8a2744492
commit 7566c4dcc8
+8 -4
View File
@@ -20,11 +20,15 @@ def _reset_all():
ip_ban_manager.banned_networks.clear() ip_ban_manager.banned_networks.clear()
# Suspicious request counts via direct reference stored in app.state # 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: try:
from main import app import gc
sm = getattr(app.state, '_security_middleware', None) from guard.middleware import SecurityMiddleware
if sm is not None: for obj in gc.get_objects():
sm.suspicious_request_counts.clear() if isinstance(obj, SecurityMiddleware):
obj.suspicious_request_counts.clear()
except Exception: except Exception:
pass pass