more work on tests

blacklanternsecurity · Sep 21, 2024 · c58cf71 · c58cf71
1 parent 126d317
commit c58cf71
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/bbot/test/test_step_2/module_tests/test_module_stdout.py b/bbot/test/test_step_2/module_tests/test_module_stdout.py
@@ -46,8 +46,10 @@ def check(self, module_test, events):
             event = json.loads(line)
             if i == 0:
                 assert event["type"] == "SCAN"
-            elif i == 2:
+            elif i == 1:
                 assert event["type"] == "DNS_NAME" and event["data"] == "blacklanternsecurity.com"
+            if i == 2:
+                assert event["type"] == "SCAN"
 
 
 class TestStdoutJSONFields(TestStdout):