| 12345678910111213141516171819202122232425262728293031323334353637383940 |
- """Legacy test file - tests moved to test_template_enricher.py"""
- # This file is kept for backwards compatibility
- # All tests have been moved to test_template_enricher.py
- # with proper mocking and more comprehensive coverage
- from pathlib import Path
- import pytest
- from flowsint_core.core.template_enricher import TemplateEnricher
- from flowsint_core.templates.loader.yaml_loader import YamlLoader
- from flowsint_core.templates.types import Template
- TEST_DIR = Path(__file__).parent
- def test_enricher_init():
- """Test basic enricher initialization from YAML template."""
- template = YamlLoader.get_template_from_file(str(TEST_DIR / "example.yaml"))
- assert isinstance(template, Template)
- assert template.name == "ip-api-lookup"
- assert template.category == "Ip"
- enricher = TemplateEnricher(sketch_id="123", scan_id="123", template=template)
- assert enricher.name() == "ip-api-lookup"
- assert enricher.category() == "Ip"
- assert enricher.key() == "address"
- def test_enricher_preprocess():
- """Test enricher preprocessing converts strings to typed objects."""
- template = YamlLoader.get_template_from_file(str(TEST_DIR / "example.yaml"))
- enricher = TemplateEnricher(sketch_id="123", scan_id="123", template=template)
- pre = enricher.preprocess(["8.8.8.8"])
- assert len(pre) == 1
- assert hasattr(pre[0], "address")
- assert pre[0].address == "8.8.8.8"
|
