refactor: remove occurences of scanner

flowsint-transforms/tests/transforms/domain/whois.py

@@ -0,0 +1,103 @@
+from flowsint_transforms.domains.whois import WhoisTransform
+from flowsint_types.domain import Domain
+
+transform = WhoisTransform("sketch_123", "scan_123")
+
+
+def test_preprocess_valid_domains():
+    domains = [
+        Domain(domain="example.com"),
+        Domain(domain="example2.com"),
+    ]
+    result = transform.preprocess(domains)
+
+    result_domains = [d.domain for d in result]
+    expected_domains = [d.domain for d in domains]
+
+    assert result_domains == expected_domains
+
+
+def test_unprocessed_valid_domains():
+    domains = [
+        "example.com",
+        "example2.com",
+    ]
+    result = transform.preprocess(domains)
+    result_domains = [d for d in result]
+    expected_domains = [Domain(domain=d) for d in domains]
+    assert result_domains == expected_domains
+
+
+def test_preprocess_invalid_domains():
+    domains = [
+        Domain(domain="example.com"),
+        Domain(domain="invalid_domain"),
+        Domain(domain="example.org"),
+    ]
+    result = transform.preprocess(domains)
+
+    result_domains = [d.domain for d in result]
+    assert "example.com" in result_domains
+    assert "example.org" in result_domains
+    assert "invalid_domain" not in result_domains
+
+
+def test_preprocess_multiple_formats():
+    domains = [
+        {"domain": "example.com"},
+        {"invalid_key": "example.io"},
+        Domain(domain="example.org"),
+        "example.org",
+    ]
+    result = transform.preprocess(domains)
+
+    result_domains = [d.domain for d in result]
+    assert "example.com" in result_domains
+    assert "example.org" in result_domains
+    assert "invalid_domain" not in result_domains
+    assert "example.io" not in result_domains
+
+
+def test_scan_returns_whois_objects(monkeypatch):
+    # Patch `whois.whois` to avoid real network call
+    mock_whois = lambda domain: {
+        "registrar": "MockRegistrar",
+        "org": "MockOrg",
+        "city": "MockCity",
+        "country": "MockCountry",
+        "emails": ["admin@example.com"],
+        "creation_date": "2020-01-01",
+        "expiration_date": "2030-01-01",
+    }
+
+    monkeypatch.setattr("whois.whois", mock_whois)
+
+    input_data = [Domain(domain="example.com")]
+    output = transform.execute(input_data)
+    assert isinstance(output, list)
+    assert isinstance(output[0], Domain)
+    assert output[0].whois.org == "MockOrg"
+    assert output[0].whois.email.email == "admin@example.com"
+
+
+def test_schemas():
+    input_schema = transform.input_schema()
+    output_schema = transform.output_schema()
+    assert input_schema == {
+        "type": "Domain",
+        "properties": [
+            {"name": "domain", "type": "string"},
+            {"name": "subdomains", "type": "array | null"},
+            {"name": "ips", "type": "array | null"},
+            {"name": "whois", "type": "Whois | null"},
+        ],
+    }
+    assert output_schema == {
+        "type": "Domain",
+        "properties": [
+            {"name": "domain", "type": "string"},
+            {"name": "subdomains", "type": "array | null"},
+            {"name": "ips", "type": "array | null"},
+            {"name": "whois", "type": "Whois | null"},
+        ],
+    }