mtr manager and tool run + test

2 files changed, 267 insertions, 2 deletions

diff --git a/tests/data/mtr_fatbeam.json b/tests/data/mtr_fatbeam.json
new file mode 100644
index 0000000..6e27eb1
--- /dev/null
+++ b/tests/data/mtr_fatbeam.json
@@ -0,0 +1,206 @@
+{
+    "report": {
+        "mtr": {
+            "src": "gstupp-ThinkPad-X1-Carbon-Gen-11",
+            "dst": "167.150.6.80",
+            "tos": 0,
+            "tests": 10,
+            "psize": "64",
+            "bitpattern": "0x00"
+        },
+        "hubs": [
+            {
+                "count": 1,
+                "host": "_gateway (172.20.20.1)",
+                "ASN": "AS???",
+                "Loss%": 0.0,
+                "Snt": 10,
+                "Last": 2.408,
+                "Avg": 16.157,
+                "Best": 2.408,
+                "Wrst": 69.531,
+                "StDev": 20.69
+            },
+            {
+                "count": 2,
+                "host": "172.16.20.1",
+                "ASN": "AS???",
+                "Loss%": 0.0,
+                "Snt": 10,
+                "Last": 3.411,
+                "Avg": 16.906,
+                "Best": 2.613,
+                "Wrst": 90.7,
+                "StDev": 27.547
+            },
+            {
+                "count": 3,
+                "host": "192.168.1.254",
+                "ASN": "AS???",
+                "Loss%": 0.0,
+                "Snt": 10,
+                "Last": 17.012,
+                "Avg": 9.812,
+                "Best": 3.061,
+                "Wrst": 25.728,
+                "StDev": 8.908
+            },
+            {
+                "count": 4,
+                "host": "ipdsl-jal-ptovallarta-19-l0.uninet.net.mx (201.154.95.117)",
+                "ASN": "AS???",
+                "Loss%": 0.0,
+                "Snt": 10,
+                "Last": 6.954,
+                "Avg": 10.216,
+                "Best": 6.177,
+                "Wrst": 16.151,
+                "StDev": 3.343
+            },
+            {
+                "count": 5,
+                "host": "bb-la-onewilshire-29-ae32_0.uninet.net.mx (189.246.202.49)",
+                "ASN": "AS???",
+                "Loss%": 0.0,
+                "Snt": 10,
+                "Last": 52.557,
+                "Avg": 54.174,
+                "Best": 45.681,
+                "Wrst": 71.387,
+                "StDev": 8.011
+            },
+            {
+                "count": 6,
+                "host": "ae91.edge7.LosAngeles1.Level3.net (4.7.28.197)",
+                "ASN": "AS3356",
+                "Loss%": 0.0,
+                "Snt": 10,
+                "Last": 1079.2,
+                "Avg": 875.97,
+                "Best": 47.78,
+                "Wrst": 4150.5,
+                "StDev": 1345.7
+            },
+            {
+                "count": 7,
+                "host": "???",
+                "ASN": "AS???",
+                "Loss%": 100.0,
+                "Snt": 10,
+                "Last": 0.0,
+                "Avg": 0.0,
+                "Best": 0.0,
+                "Wrst": 0.0,
+                "StDev": 0.0
+            },
+            {
+                "count": 8,
+                "host": "ae10.cr1.lax10.us.zip.zayo.com (64.125.28.224)",
+                "ASN": "AS6461",
+                "Loss%": 70.0,
+                "Snt": 10,
+                "Last": 1186.5,
+                "Avg": 2189.8,
+                "Best": 1186.5,
+                "Wrst": 3202.8,
+                "StDev": 1008.2
+            },
+            {
+                "count": 9,
+                "host": "ae16.cr1.sjc1.us.zip.zayo.com (64.125.21.171)",
+                "ASN": "AS6461",
+                "Loss%": 0.0,
+                "Snt": 10,
+                "Last": 92.819,
+                "Avg": 414.75,
+                "Best": 90.799,
+                "Wrst": 2140.8,
+                "StDev": 690.96
+            },
+            {
+                "count": 10,
+                "host": "ae27.cs3.sjc7.us.zip.zayo.com (64.125.18.28)",
+                "ASN": "AS6461",
+                "Loss%": 90.0,
+                "Snt": 10,
+                "Last": 5234.8,
+                "Avg": 5234.8,
+                "Best": 5234.8,
+                "Wrst": 5234.8,
+                "StDev": 0.0
+            },
+            {
+                "count": 11,
+                "host": "???",
+                "ASN": "AS???",
+                "Loss%": 100.0,
+                "Snt": 10,
+                "Last": 0.0,
+                "Avg": 0.0,
+                "Best": 0.0,
+                "Wrst": 0.0,
+                "StDev": 0.0
+            },
+            {
+                "count": 12,
+                "host": "ae8.cr1.sea1.us.zip.zayo.com (64.125.28.193)",
+                "ASN": "AS6461",
+                "Loss%": 0.0,
+                "Snt": 10,
+                "Last": 93.389,
+                "Avg": 1238.6,
+                "Best": 91.537,
+                "Wrst": 5223.9,
+                "StDev": 1644.1
+            },
+            {
+                "count": 13,
+                "host": "ae7.ter2.sea1.us.zip.zayo.com (64.125.19.197)",
+                "ASN": "AS6461",
+                "Loss%": 0.0,
+                "Snt": 10,
+                "Last": 91.212,
+                "Avg": 112.17,
+                "Best": 90.979,
+                "Wrst": 178.26,
+                "StDev": 30.086
+            },
+            {
+                "count": 14,
+                "host": "208.185.33.178.IDIA-369396-ZYO.zip.zayo.com (208.185.33.178)",
+                "ASN": "AS6461",
+                "Loss%": 0.0,
+                "Snt": 10,
+                "Last": 103.95,
+                "Avg": 104.46,
+                "Best": 90.349,
+                "Wrst": 136.62,
+                "StDev": 15.726
+            },
+            {
+                "count": 15,
+                "host": "168.245.215.250",
+                "ASN": "AS55039",
+                "Loss%": 0.0,
+                "Snt": 10,
+                "Last": 85.672,
+                "Avg": 95.289,
+                "Best": 84.352,
+                "Wrst": 156.16,
+                "StDev": 21.621
+            },
+            {
+                "count": 16,
+                "host": "???",
+                "ASN": "AS???",
+                "Loss%": 100.0,
+                "Snt": 10,
+                "Last": 0.0,
+                "Avg": 0.0,
+                "Best": 0.0,
+                "Wrst": 0.0,
+                "StDev": 0.0
+            }
+        ]
+    }
+}
diff --git a/tests/managers/network/tool_run.py b/tests/managers/network/tool_run.py
index a598a71..0f9388f 100644
--- a/tests/managers/network/tool_run.py
+++ b/tests/managers/network/tool_run.py
@@ -1,11 +1,27 @@
+import os
+from datetime import datetime, timezone
 from uuid import uuid4
 
 import faker
-
+import pytest
+
+from generalresearch.models.network.definitions import IPProtocol
+from generalresearch.models.network.mtr import (
+    get_mtr_version,
+    parse_raw_output,
+    MTRReport,
+    get_mtr_command,
+)
 from generalresearch.models.network.tool_run import (
     new_tool_run_from_nmap,
     run_dig,
+    MtrRun,
+    ToolName,
+    ToolClass,
+    Status,
 )
+from generalresearch.models.network.tool_utils import ToolRunCommand
+
 fake = faker.Faker()
 
 
@@ -38,6 +54,7 @@ def test_run_dig(toolrun_manager):
 
     assert reverse_dns_run == run_out
 
+
 def test_run_dig_empty(toolrun_manager):
     reverse_dns_run = run_dig(ip=fake.ipv6())
 
@@ -45,4 +62,46 @@ def test_run_dig_empty(toolrun_manager):
 
     run_out = toolrun_manager.get_rdns_run(reverse_dns_run.id)
 
-    assert reverse_dns_run == run_out
\ No newline at end of file
+    assert reverse_dns_run == run_out
+
+
+@pytest.fixture(scope="session")
+def mtr_report(request) -> MTRReport:
+    fp = os.path.join(request.config.rootpath, "data/mtr_fatbeam.json")
+    with open(fp, "r") as f:
+        s = f.read()
+    data = parse_raw_output(s)
+    data["port"] = 443
+    data["protocol"] = IPProtocol.TCP
+    return MTRReport.model_validate(data)
+
+
+def test_create_tool_run_from_mtr(toolrun_manager, mtr_report):
+    started_at = datetime.now(tz=timezone.utc)
+    tool_version = get_mtr_version()
+
+    ip = mtr_report.destination
+
+    finished_at = datetime.now(tz=timezone.utc)
+    raw_command = " ".join(get_mtr_command(ip))
+
+    run = MtrRun(
+        tool_name=ToolName.MTR,
+        tool_class=ToolClass.TRACEROUTE,
+        tool_version=tool_version,
+        status=Status.SUCCESS,
+        ip=ip,
+        started_at=started_at,
+        finished_at=finished_at,
+        raw_command=raw_command,
+        scan_group_id=uuid4().hex,
+        config=ToolRunCommand.from_raw_command(raw_command),
+        parsed=mtr_report,
+        source_ip="1.1.1.1"
+    )
+
+    toolrun_manager.create_mtr_run(run)
+
+    run_out = toolrun_manager.get_mtr_run(run.id)
+
+    assert run == run_out