From cf239865ce440e1a71ee2360514eaeb018620ac9 Mon Sep 17 00:00:00 2001
From: Max Nanis
Date: Wed, 26 Aug 2026 16:46:45 -0700
Subject: Ruff afternoon!

---
 tests/models/network/test_nmap_parser.py | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

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

diff --git a/tests/models/network/test_nmap_parser.py b/tests/models/network/test_nmap_parser.py
index 7822380..473a63f 100644
--- a/tests/models/network/test_nmap_parser.py
+++ b/tests/models/network/test_nmap_parser.py
@@ -1,8 +1,14 @@
+from __future__ import annotations
+
 import os
 
 import pytest
 
 from generalresearch.models.network.nmap.parser import parse_nmap_xml
+from generalresearch.models.network.nmap.result import (
+    NmapResult,
+    NmapTrace,
+)
 
 
 @pytest.fixture
@@ -13,9 +19,11 @@ def nmap_raw_output_2(request) -> str:
     return data
 
 
-def test_nmap_xml_parser(nmap_raw_output, nmap_raw_output_2):
-    n = parse_nmap_xml(nmap_raw_output)
+def test_nmap_xml_parser(nmap_raw_output: str, nmap_raw_output_2: str):
+    n: NmapResult = parse_nmap_xml(nmap_raw_output)
     assert n.tcp_open_ports == [61232]
+
+    assert isinstance(n.trace, NmapTrace)
     assert len(n.trace.hops) == 18
 
     n = parse_nmap_xml(nmap_raw_output_2)
-- 
cgit v1.2.3