From d9ab70d98624e22b87dfe40cc5e18c8ad5eb44c4 Mon Sep 17 00:00:00 2001
From: stuppie
Date: Thu, 12 Mar 2026 11:53:18 -0600
Subject: some tests to execute the full Toolrun pipeline for each tool. fix
 small bugs

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

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

diff --git a/tests/models/network/nmap.py b/tests/models/network/nmap.py
new file mode 100644
index 0000000..f034bf0
--- /dev/null
+++ b/tests/models/network/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=20)
+    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