Allow setting stack to none in order to disable stack check

AikidoSec · Oct 8, 2024 · eb3fed0 · eb3fed0
1 parent cb4ea57
commit eb3fed0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/end2end/quart_postgres_uvicorn_test.py b/end2end/quart_postgres_uvicorn_test.py
@@ -11,7 +11,7 @@ def test_firewall_started_okay():
     events = fetch_events_from_mock("http://localhost:5000")
     started_events = filter_on_event_type(events, "started")
     assert len(started_events) == 1
-    validate_started_event(started_events[0], ["flask", "quart", "asyncpg"])
+    validate_started_event(started_events[0], None)
 
 def test_safe_response_with_firewall():
     dog_name = "Bobby Tables"

diff --git a/end2end/server/check_events_from_mock.py b/end2end/server/check_events_from_mock.py
@@ -17,4 +17,5 @@ def validate_started_event(event, stack, dry_mode=False, serverless=False, os_na
     assert event["agent"]["os"]["name"] == os_name
     assert event["agent"]["platform"]["name"] == platform
     assert event["agent"]["serverless"] ==  serverless
-    assert set(event["agent"]["stack"]) == set(stack)
+    if stack is not None:
+        assert set(event["agent"]["stack"]) == set(stack)