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/managers/network/tool_run.py | 48 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 48 insertions(+) create mode 100644 tests/managers/network/tool_run.py (limited to 'tests/managers/network/tool_run.py') diff --git a/tests/managers/network/tool_run.py b/tests/managers/network/tool_run.py new file mode 100644 index 0000000..a598a71 --- /dev/null +++ b/tests/managers/network/tool_run.py @@ -0,0 +1,48 @@ +from uuid import uuid4 + +import faker + +from generalresearch.models.network.tool_run import ( + new_tool_run_from_nmap, + run_dig, +) +fake = faker.Faker() + + +def test_create_tool_run_from_nmap(nmap_run, toolrun_manager): + scan_group_id = uuid4().hex + run = new_tool_run_from_nmap(nmap_run, scan_group_id=scan_group_id) + + toolrun_manager.create_portscan_run(run) + + run_out = toolrun_manager.get_portscan_run(run.id) + + assert run == run_out + + +def test_create_tool_run_from_dig_fixture(reverse_dns_run, toolrun_manager): + + toolrun_manager.create_rdns_run(reverse_dns_run) + + run_out = toolrun_manager.get_rdns_run(reverse_dns_run.id) + + assert reverse_dns_run == run_out + + +def test_run_dig(toolrun_manager): + reverse_dns_run = run_dig(ip="65.19.129.53") + + toolrun_manager.create_rdns_run(reverse_dns_run) + + run_out = toolrun_manager.get_rdns_run(reverse_dns_run.id) + + assert reverse_dns_run == run_out + +def test_run_dig_empty(toolrun_manager): + reverse_dns_run = run_dig(ip=fake.ipv6()) + + toolrun_manager.create_rdns_run(reverse_dns_run) + + run_out = toolrun_manager.get_rdns_run(reverse_dns_run.id) + + assert reverse_dns_run == run_out \ No newline at end of file -- cgit v1.2.3