fix cloudcheck tests

blacklanternsecurity · Jul 28, 2024 · e9ebdf4 · e9ebdf4
1 parent b855d70
commit e9ebdf4
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/bbot/test/test_step_2/module_tests/test_module_cloudcheck.py b/bbot/test/test_step_2/module_tests/test_module_cloudcheck.py
@@ -48,11 +48,11 @@ async def setup_after_prep(self, module_test):
         other_event3._resolved_hosts = {"asdf.amazonaws.com"}
 
         for event in (ip_event, aws_event1, aws_event2, aws_event4, other_event2, other_event3):
-            await module.handle_event(event, {})
+            await module.handle_event(event)
             assert "cloud-amazon" in event.tags, f"{event} was not properly cloud-tagged"
 
         for event in (aws_event3, other_event1):
-            await module.handle_event(event, {})
+            await module.handle_event(event)
             assert "cloud-amazon" not in event.tags, f"{event} was improperly cloud-tagged"
             assert not any(
                 t for t in event.tags if t.startswith("cloud-") or t.startswith("cdn-")
@@ -64,7 +64,7 @@ async def setup_after_prep(self, module_test):
         google_event3._resolved_hosts = {"asdf.storage.googleapis.com"}
 
         for event in (google_event1, google_event2, google_event3):
-            await module.handle_event(event, {})
+            await module.handle_event(event)
             assert "cloud-google" in event.tags, f"{event} was not properly cloud-tagged"
         assert "cloud-storage-bucket" in google_event3.tags