Prevent missing regex_patterns (#15)

tests/test_custom_formats.py

@@ -0,0 +1,98 @@
+"""Tests to ensure custom format patterns reference valid regex pattern files."""
+
+from pathlib import Path
+
+import pytest
+import yaml
+
+
+# Get the project root directory
+PROJECT_ROOT = Path(__file__).parent.parent
+CUSTOM_FORMATS_DIR = PROJECT_ROOT / "custom_formats"
+REGEX_PATTERNS_DIR = PROJECT_ROOT / "regex_patterns"
+
+
+def get_all_custom_formats():
+    """Get all custom format YAML files."""
+    if not CUSTOM_FORMATS_DIR.exists():
+        return []
+    return list(CUSTOM_FORMATS_DIR.glob("*.yml"))
+
+@pytest.fixture(scope="module")
+def available_regex_patterns_file_names():
+    """Fixture to load all available regex pattern names once."""
+    if not REGEX_PATTERNS_DIR.exists():
+        return set()
+
+    pattern_names = set()
+    for yml_file in REGEX_PATTERNS_DIR.glob("*.yml"):
+        # Use filename without extension as pattern name
+        pattern_names.add(yml_file.stem)
+    return pattern_names
+
+
+@pytest.fixture(scope="module")
+def available_regex_patterns_names():
+    """Fixture to load all available regex pattern names once."""
+    if not REGEX_PATTERNS_DIR.exists():
+        return set()
+
+    pattern_names = set()
+    for yml_file in REGEX_PATTERNS_DIR.glob("*.yml"):
+        # Read name from YAML file content
+        with open(yml_file, encoding="utf-8") as f:
+            data = yaml.safe_load(f)
+            if data and "name" in data:
+                pattern_names.add(data["name"])
+    return pattern_names
+
+
+@pytest.mark.parametrize("custom_format_file", get_all_custom_formats())
+def test_custom_format_patterns_exist(
+    custom_format_file,
+    available_regex_patterns_file_names,
+    available_regex_patterns_names,
+):
+    """Test that every pattern referenced in a custom format exists in regex_patterns."""
+    with open(custom_format_file, encoding="utf-8") as f:
+        data = yaml.safe_load(f)
+
+    if not data or "conditions" not in data:
+        pytest.skip(f"No conditions found in {custom_format_file.name}")
+
+    missing_patterns_files = []
+    missing_patterns_names = []
+
+    for condition in data["conditions"]:
+        pattern_name = condition.get("pattern")
+        if not pattern_name:
+            continue
+
+        # Check if the referenced pattern exists in regex_patterns
+        if pattern_name not in available_regex_patterns_file_names:
+            missing_patterns_files.append(pattern_name)
+
+        if pattern_name not in available_regex_patterns_names:
+            missing_patterns_names.append(pattern_name)
+
+    if missing_patterns_files:
+        pytest.fail(
+            f"Custom format '{custom_format_file.name}' references missing regex patterns:\n"
+            + "\n".join(f"  - {pattern}" for pattern in missing_patterns_files)
+        )
+
+    if missing_patterns_names:
+        pytest.fail(
+            f"Custom format '{custom_format_file.name}' references missing regex patterns:\n"
+            + "\n".join(f"  - {pattern}" for pattern in missing_patterns_names)
+        )
+
+
+def test_custom_formats_directory_exists():
+    """Test that the custom_formats directory exists."""
+    assert CUSTOM_FORMATS_DIR.exists(), f"Custom formats directory not found: {CUSTOM_FORMATS_DIR}"
+
+
+def test_regex_patterns_directory_exists():
+    """Test that the regex_patterns directory exists."""
+    assert REGEX_PATTERNS_DIR.exists(), f"Regex patterns directory not found: {REGEX_PATTERNS_DIR}"