fix aggregate tests

blacklanternsecurity · Feb 7, 2024 · e5d88dc · e5d88dc
1 parent 652b233
commit e5d88dc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bbot/test/test_step_2/module_tests/test_module_aggregate.py b/bbot/test/test_step_2/module_tests/test_module_aggregate.py
@@ -5,7 +5,7 @@ class TestAggregate(ModuleTestBase):
     config_overrides = {"dns_resolution": True, "scope_report_distance": 1}
 
     async def setup_before_prep(self, module_test):
-        module_test.scan.helpers.dns.mock_dns({("blacklanternsecurity.com", "A"): "1.2.3.4"})
+        module_test.mock_dns({"blacklanternsecurity.com": {"A": ["1.2.3.4"]}})
 
     def check(self, module_test, events):
         filename = next(module_test.scan.home.glob("scan-stats-table*.txt"))