refactor: remove occurences of scanner

flowsint-transforms/tests/transforms/ip/cidr_to_ips.py

@@ -0,0 +1,122 @@
+from flowsint_transforms.ips.cidr_to_ips import CidrToIpsTransform
+from flowsint_types.cidr import CIDR
+from flowsint_types.ip import Ip
+from tests.logger import TestLogger
+
+logger = TestLogger()
+transform = CidrToIpsTransform("sketch_123", "scan_123", logger)
+
+
+def test_preprocess_valid_cidrs():
+    cidrs = [
+        CIDR(network="8.8.8.0/24"),
+        CIDR(network="1.1.1.0/24"),
+    ]
+    result = transform.preprocess(cidrs)
+
+    result_networks = [cidr.network for cidr in result]
+    expected_networks = [cidr.network for cidr in cidrs]
+
+    assert result_networks == expected_networks
+
+
+def test_preprocess_unprocessed_valid_cidrs():
+    cidrs = [
+        "8.8.8.0/24",
+        "1.1.1.0/24",
+    ]
+    result = transform.preprocess(cidrs)
+    result_cidrs = [c for c in result]
+    expected_cidrs = [CIDR(network=c) for c in cidrs]
+    assert result_cidrs == expected_cidrs
+
+
+def test_preprocess_invalid_cidrs():
+    cidrs = [
+        CIDR(network="8.8.8.0/24"),
+        "invalid-cidr",
+        "not-a-cidr",
+    ]
+    result = transform.preprocess(cidrs)
+    result_networks = [str(cidr.network) for cidr in result]
+    assert "8.8.8.0/24" in result_networks
+    assert "invalid-cidr" not in result_networks
+    assert "not-a-cidr" not in result_networks
+
+
+def test_preprocess_multiple_formats():
+    cidrs = [
+        {"network": "8.8.8.0/24"},
+        {"invalid_key": "1.1.1.0/24"},
+        CIDR(network="9.9.9.0/24"),
+        "InvalidCIDR",
+    ]
+    result = transform.preprocess(cidrs)
+    result_networks = [str(cidr.network) for cidr in result]
+    assert "8.8.8.0/24" in result_networks
+    assert "9.9.9.0/24" in result_networks
+    assert "1.1.1.0/24" not in result_networks
+    assert "InvalidCIDR" not in result_networks
+
+
+def test_scan_extracts_ips(monkeypatch):
+    mock_dnsx_output = """8.35.200.12
+8.35.200.112
+8.35.200.16
+8.35.200.170"""
+
+    class MockSubprocessResult:
+        def __init__(self, stdout):
+            self.stdout = stdout
+            self.returncode = 0
+
+    def mock_subprocess_run(cmd, shell, capture_output, text, timeout):
+        assert "dnsx" in cmd
+        assert "-ptr" in cmd
+        return MockSubprocessResult(mock_dnsx_output)
+
+    # Patch the subprocess call in the transform
+    monkeypatch.setattr("subprocess.run", mock_subprocess_run)
+
+    input_data = [CIDR(network="8.35.200.0/24")]
+    ips = transform.scan(input_data)
+
+    assert isinstance(ips, list)
+    assert len(ips) == 4
+
+    expected_ips = ["8.35.200.12", "8.35.200.112", "8.35.200.16", "8.35.200.170"]
+
+    for ip in ips:
+        assert isinstance(ip, Ip)
+        assert ip.address in expected_ips
+
+
+def test_scan_handles_empty_output(monkeypatch):
+    class MockSubprocessResult:
+        def __init__(self):
+            self.stdout = ""
+            self.returncode = 0
+
+    def mock_subprocess_run(cmd, shell, capture_output, text, timeout):
+        return MockSubprocessResult()
+
+    monkeypatch.setattr("subprocess.run", mock_subprocess_run)
+
+    input_data = [CIDR(network="8.8.8.0/24")]
+    ips = transform.scan(input_data)
+
+    assert isinstance(ips, list)
+    assert len(ips) == 0
+
+
+def test_scan_handles_subprocess_exception(monkeypatch):
+    def mock_subprocess_run(cmd, shell, capture_output, text, timeout):
+        raise Exception("Subprocess failed")
+
+    monkeypatch.setattr("subprocess.run", mock_subprocess_run)
+
+    input_data = [CIDR(network="8.8.8.0/24")]
+    ips = transform.scan(input_data)
+
+    assert isinstance(ips, list)
+    assert len(ips) == 0