From 0742210fc7eaa54d8068bfd18414158dc9b33672 Mon Sep 17 00:00:00 2001
From: Max Nanis
Date: Tue, 18 Aug 2026 16:35:11 -0700
Subject: Ruff cleanup

---
 tests/models/network/test_nmap.py | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

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

diff --git a/tests/models/network/test_nmap.py b/tests/models/network/test_nmap.py
index 0be98d4..a135a13 100644
--- a/tests/models/network/test_nmap.py
+++ b/tests/models/network/test_nmap.py
@@ -1,20 +1,21 @@
 import subprocess
 
-from generalresearch.models.network.definitions import IPProtocol
-from generalresearch.models.network.nmap.execute import execute_nmap
 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 ToolName, ToolClass
+from generalresearch.models.network.tool_run import ToolClass, ToolName
 
 fake = faker.Faker()
 
 
-def resolve(host):
+def resolve(host: str):
     return subprocess.check_output(["dig", host, "+short"]).decode().strip()
 
 
-def test_execute_nmap_scanme(toolrun_manager):
+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)
-- 
cgit v1.2.3