ToolRunCommand + options models to handle cmd line args for all network tools

1 files changed, 25 insertions, 0 deletions

diff --git a/tests/managers/network/test_tool_run.py b/tests/managers/network/test_tool_run.py
new file mode 100644
index 0000000..a815809
--- /dev/null
+++ b/tests/managers/network/test_tool_run.py
@@ -0,0 +1,25 @@
+def test_create_tool_run_from_nmap_run(nmap_run, toolrun_manager):
+
+    toolrun_manager.create_nmap_run(nmap_run)
+
+    run_out = toolrun_manager.get_nmap_run(nmap_run.id)
+
+    assert nmap_run == run_out
+
+
+def test_create_tool_run_from_rdns_run(rdns_run, toolrun_manager):
+
+    toolrun_manager.create_rdns_run(rdns_run)
+
+    run_out = toolrun_manager.get_rdns_run(rdns_run.id)
+
+    assert rdns_run == run_out
+
+
+def test_create_tool_run_from_mtr_run(mtr_run, toolrun_manager):
+
+    toolrun_manager.create_mtr_run(mtr_run)
+
+    run_out = toolrun_manager.get_mtr_run(mtr_run.id)
+
+    assert mtr_run == run_out