Ruff afternoon!

1 files changed, 8 insertions, 3 deletions

diff --git a/tests/models/network/test_nmap.py b/tests/models/network/test_nmap.py
index a135a13..5e9f4d0 100644
--- a/tests/models/network/test_nmap.py
+++ b/tests/models/network/test_nmap.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import subprocess
 
 import faker
@@ -5,8 +7,8 @@ import faker
 from generalresearch.managers.network.tool_run import ToolRunManager
 from generalresearch.models.network.definitions import IPProtocol
 from generalresearch.models.network.nmap.execute import execute_nmap
-from generalresearch.models.network.nmap.result import PortState
-from generalresearch.models.network.tool_run import ToolClass, ToolName
+from generalresearch.models.network.nmap.result import NmapResult, PortState
+from generalresearch.models.network.tool_run import NmapRun, Status, ToolClass, ToolName
 
 fake = faker.Faker()
 
@@ -18,10 +20,13 @@ def resolve(host: str):
 def test_execute_nmap_scanme(toolrun_manager: ToolRunManager):
     ip = resolve("scanme.nmap.org")
 
-    run = execute_nmap(ip=ip, top_ports=None, ports="20-30", enable_advanced=False)
+    run: NmapRun = 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
+    assert isinstance(run.parsed, NmapResult)
     result = run.parsed
 
     port22 = result._port_index[(IPProtocol.TCP, 22)]