ToolRunCommand + options models to handle cmd line args for all network tools

1 files changed, 33 insertions, 0 deletions

diff --git a/tests/models/network/test_rdns.py b/tests/models/network/test_rdns.py
new file mode 100644
index 0000000..e56c494
--- /dev/null
+++ b/tests/models/network/test_rdns.py
@@ -0,0 +1,33 @@
+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)