4 files changed, 65 insertions, 61 deletions
diff --git a/tests/models/network/nmap.py b/tests/models/network/nmap.py
deleted file mode 100644
index 4fc7014..0000000
--- a/tests/models/network/nmap.py
+++ /dev/null
@@ -1,32 +0,0 @@
-import os
-
-import pytest
-
-from generalresearch.models.network.xml_parser import NmapXmlParser
-
-
-@pytest.fixture
-def nmap_xml_str(request) -> str:
-    fp = os.path.join(request.config.rootpath, "data/nmaprun1.xml")
-    with open(fp, "r") as f:
-        data = f.read()
-    return data
-
-
-@pytest.fixture
-def nmap_xml_str2(request) -> str:
-    fp = os.path.join(request.config.rootpath, "data/nmaprun2.xml")
-    with open(fp, "r") as f:
-        data = f.read()
-    return data
-
-
-def test_nmap_xml_parser(nmap_xml_str, nmap_xml_str2):
-    p = NmapXmlParser()
-    n = p.parse_xml(nmap_xml_str)
-    assert n.tcp_open_ports == [61232]
-    assert len(n.trace.hops) == 18
-
-    n = p.parse_xml(nmap_xml_str2)
-    assert n.tcp_open_ports == [22, 80, 9929, 31337]
-    assert n.trace is None
diff --git a/tests/models/network/nmap_parser.py b/tests/models/network/nmap_parser.py
new file mode 100644
index 0000000..96d7b37
--- /dev/null
+++ b/tests/models/network/nmap_parser.py
@@ -0,0 +1,22 @@
+import os
+
+import pytest
+
+from generalresearch.models.network.nmap.parser import parse_nmap_xml
+
+@pytest.fixture
+def nmap_raw_output_2(request) -> str:
+    fp = os.path.join(request.config.rootpath, "data/nmaprun2.xml")
+    with open(fp, "r") as f:
+        data = f.read()
+    return data
+
+
+def test_nmap_xml_parser(nmap_raw_output, nmap_raw_output_2):
+    n = parse_nmap_xml(nmap_raw_output)
+    assert n.tcp_open_ports == [61232]
+    assert len(n.trace.hops) == 18
+
+    n = parse_nmap_xml(nmap_raw_output_2)
+    assert n.tcp_open_ports == [22, 80, 9929, 31337]
+    assert n.trace is None
diff --git a/tests/models/network/rdns.py b/tests/models/network/rdns.py
index 9167749..64e8351 100644
--- a/tests/models/network/rdns.py
+++ b/tests/models/network/rdns.py
@@ -1,23 +1,45 @@
-from generalresearch.models.network.rdns import dig_rdns
-import faker
+# from generalresearch.models.network.rdns import run_rdns
+# import faker
+#
+# fake = faker.Faker()
+#
+#
+# def test_dig_rdns():
+#     # Actually runs dig -x. Idk how stable this is
+#     ip = "45.33.32.156"
+#     rdns_result = run_rdns(ip)
+#     assert rdns_result.primary_hostname == "scanme.nmap.org"
+#     assert rdns_result.primary_org == "nmap"
+#
+#     ip = "65.19.129.53"
+#     rdns_result = run_rdns(ip)
+#     assert rdns_result.primary_hostname == "in1-smtp.grlengine.com"
+#     assert rdns_result.primary_org == "grlengine"
+#
+#     ip = fake.ipv6()
+#     rdns_result = run_rdns(ip)
+#     assert rdns_result.primary_hostname is None
+#     assert rdns_result.primary_org is None
+#     print(rdns_result.model_dump_postgres())
 
-fake = faker.Faker()
 
-
-def test_dig_rdns():
-    # Actually runs dig -x. Idk how stable this is
-    ip = "45.33.32.156"
-    rdns_result = dig_rdns(ip)
-    assert rdns_result.primary_hostname == "scanme.nmap.org"
-    assert rdns_result.primary_org == "nmap"
-
-    ip = "65.19.129.53"
-    rdns_result = dig_rdns(ip)
-    assert rdns_result.primary_hostname == "in1-smtp.grlengine.com"
-    assert rdns_result.primary_org == "grlengine"
-
-    ip = fake.ipv6()
-    rdns_result = dig_rdns(ip)
-    assert rdns_result.primary_hostname is None
-    assert rdns_result.primary_org is None
-    print(rdns_result.model_dump_postgres())
+#
+#
+# def test_run_dig(toolrun_manager):
+#     reverse_dns_run = run_dig(ip="65.19.129.53")
+#
+#     toolrun_manager.create_rdns_run(reverse_dns_run)
+#
+#     run_out = toolrun_manager.get_rdns_run(reverse_dns_run.id)
+#
+#     assert reverse_dns_run == run_out
+#
+#
+# def test_run_dig_empty(toolrun_manager):
+#     reverse_dns_run = run_dig(ip=fake.ipv6())
+#
+#     toolrun_manager.create_rdns_run(reverse_dns_run)
+#
+#     run_out = toolrun_manager.get_rdns_run(reverse_dns_run.id)
+#
+#     assert reverse_dns_run == run_out
+\ No newline at end of file
diff --git a/tests/models/network/tool_run.py b/tests/models/network/tool_run.py
deleted file mode 100644
index c643503..0000000
--- a/tests/models/network/tool_run.py
+++ /dev/null
@@ -1,8 +0,0 @@
-from uuid import uuid4
-
-from generalresearch.models.network.tool_run import new_tool_run_from_nmap
-
-
-def test_new_tool_run_from_nmap(nmap_run):
-    scan_group_id = uuid4().hex
-    run, scan = new_tool_run_from_nmap(nmap_run, scan_group_id=scan_group_id)