1 files changed, 86 insertions, 0 deletions
diff --git a/test_utils/managers/network/conftest.py b/test_utils/managers/network/conftest.py
new file mode 100644
index 0000000..70fda4e
--- /dev/null
+++ b/test_utils/managers/network/conftest.py
@@ -0,0 +1,86 @@
+import os
+from datetime import datetime, timezone
+from typing import Callable, TYPE_CHECKING
+from uuid import uuid4
+
+import pytest
+
+from generalresearch.managers.network.label import IPLabelManager
+from generalresearch.managers.network.nmap import NmapManager
+from generalresearch.managers.network.tool_run import ToolRunManager
+from generalresearch.models.network.rdns import (
+    RDNSResult,
+    get_dig_version,
+    get_dig_rdns_command,
+)
+from generalresearch.models.network.tool_run import (
+    RDnsRun,
+    ToolName,
+    ToolClass,
+    Status,
+)
+from generalresearch.models.network.tool_utils import ToolRunCommand
+from generalresearch.models.network.xml_parser import NmapXmlParser
+
+
+@pytest.fixture(scope="session")
+def iplabel_manager(thl_web_rw) -> IPLabelManager:
+    assert "/unittest-" in thl_web_rw.dsn.path
+
+    return IPLabelManager(pg_config=thl_web_rw)
+
+
+@pytest.fixture(scope="session")
+def nmap_manager(thl_web_rw) -> NmapManager:
+    assert "/unittest-" in thl_web_rw.dsn.path
+
+    return NmapManager(pg_config=thl_web_rw)
+
+
+@pytest.fixture(scope="session")
+def toolrun_manager(thl_web_rw) -> ToolRunManager:
+    assert "/unittest-" in thl_web_rw.dsn.path
+
+    return ToolRunManager(pg_config=thl_web_rw)
+
+
+@pytest.fixture(scope="session")
+def nmap_xml_str(request) -> str:
+    fp = os.path.join(request.config.rootpath, "data/nmaprun1.xml")
+    with open(fp, "r") as f:
+        data = f.read()
+    return data
+
+
+@pytest.fixture(scope="session")
+def nmap_run(nmap_xml_str):
+    return NmapXmlParser.parse_xml(nmap_xml_str)
+
+
+@pytest.fixture(scope="session")
+def raw_dig_output():
+    return "156.32.33.45.in-addr.arpa. 300	IN	PTR	scanme.nmap.org."
+
+
+@pytest.fixture(scope="session")
+def reverse_dns_run(raw_dig_output):
+    ip = "45.33.32.156"
+    rdns_result = RDNSResult.from_dig(ip=ip, raw_output=raw_dig_output)
+    scan_group_id = uuid4().hex
+    started_at = datetime.now(tz=timezone.utc)
+    tool_version = get_dig_version()
+    finished_at = datetime.now(tz=timezone.utc)
+    raw_command = get_dig_rdns_command(ip)
+    return RDnsRun(
+        tool_name=ToolName.DIG,
+        tool_class=ToolClass.RDNS,
+        tool_version=tool_version,
+        status=Status.SUCCESS,
+        ip=ip,
+        started_at=started_at,
+        finished_at=finished_at,
+        raw_command=raw_command,
+        scan_group_id=scan_group_id or uuid4().hex,
+        config=ToolRunCommand.from_raw_command(raw_command),
+        parsed=rdns_result,
+    )