Test now compares to 'host' or 'url'

blacklanternsecurity · Feb 7, 2024 · eccb7a9 · eccb7a9
1 parent a8f53a0
commit eccb7a9
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/bbot/test/test_step_2/module_tests/test_module_newsletters.py b/bbot/test/test_step_2/module_tests/test_module_newsletters.py
@@ -44,11 +44,15 @@ def check(self, module_test, events):
         for event in events:
             self.log.info(f"event type: {event.type}")
             if event.type == "FINDING":
-                self.log.info(f"event data: {event.data['host']}")
+                self.log.info(f"event data: {event.data}")
                 # Verify Positive Result
                 if event.data["host"] == self.found_tgt:
                     status = 1
+                elif event.data["url"] == self.found_tgt:
+                    status = 1
                 # Verify Negative Result (should skip this statement if correct)
                 elif event.data["host"] == self.missing_tgt:
                     status = -2
+                elif event.data["url"] == self.missing_tgt:
+                    status = -2
         assert status == 1, f"NEWSLETTER Error - Expect status of 1 but got {status}"