100 lines
3.2 KiB
Python
100 lines
3.2 KiB
Python
"""Tests to ensure profiles reference valid custom format files."""
|
|
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
import yaml
|
|
|
|
|
|
# Get the project root directory
|
|
PROJECT_ROOT = Path(__file__).parent.parent
|
|
PROFILES_DIR = PROJECT_ROOT / "profiles"
|
|
CUSTOM_FORMATS_DIR = PROJECT_ROOT / "custom_formats"
|
|
|
|
|
|
def get_all_profiles():
|
|
"""Get all profile YAML files."""
|
|
if not PROFILES_DIR.exists():
|
|
return []
|
|
return list(PROFILES_DIR.glob("*.yml"))
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def available_custom_formats_file_names():
|
|
"""Fixture to load all available custom format names once."""
|
|
if not CUSTOM_FORMATS_DIR.exists():
|
|
return set()
|
|
|
|
format_names = set()
|
|
for yml_file in CUSTOM_FORMATS_DIR.glob("*.yml"):
|
|
# Use filename without extension as format name
|
|
format_names.add(yml_file.stem)
|
|
|
|
return format_names
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def available_custom_formats_names():
|
|
"""Fixture to load all available custom format names once."""
|
|
if not CUSTOM_FORMATS_DIR.exists():
|
|
return set()
|
|
|
|
format_names = set()
|
|
for yml_file in CUSTOM_FORMATS_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:
|
|
format_names.add(data["name"])
|
|
|
|
return format_names
|
|
|
|
|
|
@pytest.mark.parametrize("profile_file", get_all_profiles())
|
|
def test_profile_custom_formats_exist(
|
|
profile_file, available_custom_formats_file_names, available_custom_formats_names
|
|
):
|
|
"""Test that every custom format referenced in a profile exists in custom_formats."""
|
|
with open(profile_file, encoding="utf-8") as f:
|
|
data = yaml.safe_load(f)
|
|
|
|
if not data or "custom_formats" not in data:
|
|
pytest.skip(f"No custom_formats found in {profile_file.name}")
|
|
|
|
missing_formats_files = []
|
|
missing_formats_names = []
|
|
|
|
for custom_format_ref in data["custom_formats"]:
|
|
format_name = custom_format_ref.get("name")
|
|
if not format_name:
|
|
continue
|
|
|
|
# Check if the referenced custom format exists in custom_formats
|
|
if format_name not in available_custom_formats_file_names:
|
|
missing_formats_files.append(format_name)
|
|
|
|
# Check if the referenced custom format exists in custom_formats
|
|
if format_name not in available_custom_formats_names:
|
|
missing_formats_names.append(format_name)
|
|
|
|
if missing_formats_files:
|
|
pytest.fail(
|
|
f"Profile '{profile_file.name}' references missing custom formats:\n"
|
|
+ "\n".join(f" - {format}" for format in missing_formats_files)
|
|
)
|
|
|
|
if missing_formats_names:
|
|
pytest.fail(
|
|
f"Profile '{profile_file.name}' references missing custom formats:\n"
|
|
+ "\n".join(f" - {format}" for format in missing_formats_names)
|
|
)
|
|
|
|
|
|
def test_profiles_directory_exists():
|
|
"""Test that the profiles directory exists."""
|
|
assert PROFILES_DIR.exists(), f"Profiles directory not found: {PROFILES_DIR}"
|
|
|
|
|
|
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}"
|