llms-txt-mcp

"""Test format detection for different llms.txt file formats.""" from pathlib import Path import pytest from src.parser import detect_format @pytest.fixture def fixtures_dir() -> Path: """Return the fixtures directory path.""" return Path("tests/fixtures") @pytest.fixture def load_fixture(fixtures_dir: Path): """Factory fixture to load file content.""" def _load(filename: str) -> str: with open(fixtures_dir / filename, encoding="utf-8") as f: return f.read() return _load class TestStandardFullFormat: """Test detection of standard-full-llms-txt format.""" @pytest.mark.parametrize( "fixture_name", [ "zod-dev-llms-full.txt", "hono-dev-llms-full.txt", "orm-drizzle-team-llms-full.txt", ], ) def test_detect_standard_full_format(self, load_fixture, fixture_name: str): """Test that standard full format files are correctly detected.""" content = load_fixture(fixture_name) assert detect_format(content) == "standard-full-llms-txt" class TestYamlFrontmatterFormat: """Test detection of yaml-frontmatter-full-llms-txt format.""" @pytest.mark.parametrize( "fixture_name", [ "ai-sdk-dev-llms.txt", "nextjs-org-docs-llms-full.txt", "vercel-com-docs-llms-full.txt", ], ) def test_detect_yaml_frontmatter_format(self, load_fixture, fixture_name: str): """Test that YAML frontmatter format files are correctly detected.""" content = load_fixture(fixture_name) assert detect_format(content) == "yaml-frontmatter-full-llms-txt" class TestStandardFormat: """Test detection of standard-llms-txt format.""" def test_detect_standard_format_docker(self, load_fixture): """Test that Docker docs use standard format.""" content = load_fixture("docs-docker-com-llms.txt") assert detect_format(content) == "standard-llms-txt" def test_detect_standard_format_minimal(self, load_fixture): """Test that minimal.llms.txt uses standard format with links.""" content = load_fixture("minimal.llms.txt") assert detect_format(content) == "standard-llms-txt"