# SPDX-License-Identifier: Apache-2.0
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project

from pathlib import Path


def pytest_addoption(parser):
    """Add custom command line options."""
    parser.addoption(
        "--config-list-file",
        default="configs/models-small.txt",
        help="File containing list of config files to test",
    )


def pytest_generate_tests(metafunc):
    """Generate test parameters from config files."""
    if "config_filename" in metafunc.fixturenames:
        config_list_file = metafunc.config.getoption("--config-list-file")

        config_list_path = Path(config_list_file)
        if not config_list_path.is_absolute():
            test_dir_path = Path(__file__).parent / config_list_file
            if test_dir_path.exists():
                config_list_path = test_dir_path
            else:
                config_list_path = Path.cwd() / config_list_file

        print(f"Looking for config list at: {config_list_path}")

        config_files = []
        if config_list_path.exists():
            config_dir = config_list_path.parent
            with open(config_list_path) as f:
                for line in f:
                    line = line.strip()
                    if line and not line.startswith("#"):
                        config_path = config_dir / line
                        if config_path.exists():
                            config_files.append(config_path)
                            print(f"  ✓ Found: {config_path}")
                        else:
                            print(f"  ✗ Missing: {config_path}")
        else:
            print(f"Config list file not found: {config_list_path}")

        if config_files:
            metafunc.parametrize(
                "config_filename",
                config_files,
                ids=[config_file.stem for config_file in config_files],
            )
        else:
            print("No config files found, test will be skipped")