some tests to execute the full Toolrun pipeline for each tool. fix small bugs

1 files changed, 33 insertions, 45 deletions

diff --git a/tests/models/network/rdns.py b/tests/models/network/rdns.py
index 64e8351..e56c494 100644
--- a/tests/models/network/rdns.py
+++ b/tests/models/network/rdns.py
@@ -1,45 +1,33 @@
-# 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())
-
-
-#
-#
-# 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
+from generalresearch.models.network.rdns.execute import execute_rdns
+import faker
+
+from generalresearch.models.network.tool_run import ToolName, ToolClass
+
+fake = faker.Faker()
+
+
+def test_execute_rdns_grl(toolrun_manager):
+    ip = "65.19.129.53"
+    run = execute_rdns(ip=ip)
+    assert run.tool_name == ToolName.DIG
+    assert run.tool_class == ToolClass.RDNS
+    assert run.ip == ip
+    result = run.parsed
+    assert result.primary_hostname == "in1-smtp.grlengine.com"
+    assert result.primary_domain == "grlengine.com"
+    assert result.hostname_count == 1
+
+    toolrun_manager.create_rdns_run(run)
+
+
+def test_execute_rdns_none(toolrun_manager):
+    ip = fake.ipv6()
+    run = execute_rdns(ip)
+    result = run.parsed
+
+    assert result.primary_hostname is None
+    assert result.primary_domain is None
+    assert result.hostname_count == 0
+    assert result.hostnames == []
+
+    toolrun_manager.create_rdns_run(run)