From efe1c368b8b49a2c6b3bf2193a5b89eb5426eba3 Mon Sep 17 00:00:00 2001
From: stuppie
Date: Fri, 13 Mar 2026 13:31:22 -0600
Subject: ToolRunCommand + options models to handle cmd line args for all
 network tools

---
 tests/models/network/test_nmap.py | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)
 create mode 100644 tests/models/network/test_nmap.py

(limited to 'tests/models/network/test_nmap.py')

diff --git a/tests/models/network/test_nmap.py b/tests/models/network/test_nmap.py
new file mode 100644
index 0000000..0be98d4
--- /dev/null
+++ b/tests/models/network/test_nmap.py
@@ -0,0 +1,29 @@
+import subprocess
+
+from generalresearch.models.network.definitions import IPProtocol
+from generalresearch.models.network.nmap.execute import execute_nmap
+import faker
+
+from generalresearch.models.network.nmap.result import PortState
+from generalresearch.models.network.tool_run import ToolName, ToolClass
+
+fake = faker.Faker()
+
+
+def resolve(host):
+    return subprocess.check_output(["dig", host, "+short"]).decode().strip()
+
+
+def test_execute_nmap_scanme(toolrun_manager):
+    ip = resolve("scanme.nmap.org")
+
+    run = execute_nmap(ip=ip, top_ports=None, ports="20-30", enable_advanced=False)
+    assert run.tool_name == ToolName.NMAP
+    assert run.tool_class == ToolClass.PORT_SCAN
+    assert run.ip == ip
+    result = run.parsed
+
+    port22 = result._port_index[(IPProtocol.TCP, 22)]
+    assert port22.state == PortState.OPEN
+
+    toolrun_manager.create_nmap_run(run)
-- 
cgit v1.2.3