From a68a9eb9873c7502c2b7bddb55c4eb61689a48a2 Mon Sep 17 00:00:00 2001
From: stuppie
Date: Mon, 9 Mar 2026 18:42:22 -0600
Subject: add IPLabel, NmapRun, RDNSResult, ToolRun, model/managers/tests. nmap
 xml parser. + test. work in progress

---
 tests/models/network/nmap.py | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 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..4fc7014
--- /dev/null
+++ b/tests/models/network/nmap.py
@@ -0,0 +1,32 @@
+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
-- 
cgit v1.2.3