fix PTR tests

blacklanternsecurity · Feb 1, 2024 · f7725d4 · f7725d4
1 parent 05b181d
commit f7725d4
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/bbot/test/test_step_1/test_scan.py b/bbot/test/test_step_1/test_scan.py
@@ -56,7 +56,7 @@ async def test_scan(
     assert not scan2.in_scope("1.0.0.1")
 
     dns_table = {
-        "1.1.1.1": {"PTR": ["one.one.one.one"]},
+        "1.1.1.1.in-addr.arpa": {"PTR": ["one.one.one.one"]},
         "one.one.one.one": {"A": ["1.1.1.1"]},
     }
 

diff --git a/bbot/test/test_step_2/module_tests/test_module_affiliates.py b/bbot/test/test_step_2/module_tests/test_module_affiliates.py
@@ -8,7 +8,7 @@ class TestAffiliates(ModuleTestBase):
     async def setup_before_prep(self, module_test):
         module_test.mock_dns(
             {
-                "8.8.8.8": {"PTR": ["dns.google"]},
+                "8.8.8.8.in-addr.arpa": {"PTR": ["dns.google"]},
                 "dns.google": {"A": ["8.8.8.8"], "NS": ["ns1.zdns.google"]},
                 "ns1.zdns.google": {"A": ["1.2.3.4"]},
             }

diff --git a/bbot/test/test_step_2/module_tests/test_module_asset_inventory.py b/bbot/test/test_step_2/module_tests/test_module_asset_inventory.py
@@ -10,7 +10,7 @@ class TestAsset_Inventory(ModuleTestBase):
     async def setup_before_prep(self, module_test):
         module_test.mock_dns(
             {
-                "127.0.0.1": {"PTR": ["www.bbottest.notreal"]},
+                "1.0.0.127.in-addr.arpa": {"PTR": ["www.bbottest.notreal"]},
                 "www.bbottest.notreal": {"A": ["127.0.0.1"]},
             }
         )