test_version.py

"""Tests for the protocol-version registry and comparison helpers."""

import pytest
from mcp_types.version import (
    HANDSHAKE_PROTOCOL_VERSIONS,
    KNOWN_PROTOCOL_VERSIONS,
    MODERN_PROTOCOL_VERSIONS,
    is_version_at_least,
)


@pytest.mark.parametrize(
    ("version", "minimum", "expected"),
    [
        # equal
        ("2025-11-25", "2025-11-25", True),
        ("2024-11-05", "2024-11-05", True),
        # above
        ("2025-11-25", "2025-06-18", True),
        ("2025-06-18", "2024-11-05", True),
        # below
        ("2025-06-18", "2025-11-25", False),
        ("2024-11-05", "2025-03-26", False),
    ],
)
def test_is_version_at_least_ordering(version: str, minimum: str, expected: bool) -> None:
    """Known revisions order by registry position: equal, newer, and older pairs."""
    assert is_version_at_least(version, minimum) is expected


@pytest.mark.parametrize("version", ["zzz", "", "2025-11-26", "draft", "9999-99-99"])
def test_is_version_at_least_unknown_version_is_false(version: str) -> None:
    """Unrecognized peer strings compare conservatively, never accidentally."""
    assert is_version_at_least(version, "2024-11-05") is False


def test_is_version_at_least_unknown_minimum_raises() -> None:
    """An unknown minimum is programmer error, not peer input."""
    with pytest.raises(ValueError, match="zzz"):
        is_version_at_least("2025-11-25", "zzz")


@pytest.mark.parametrize(
    ("version", "minimum"), [(v, m) for v in KNOWN_PROTOCOL_VERSIONS for m in KNOWN_PROTOCOL_VERSIONS]
)
def test_is_version_at_least_matches_lexicographic_for_known_versions(version: str, minimum: str) -> None:
    """Drop-in equivalence: for every known (date-shaped) revision pair the helper
    agrees with the string comparison it replaced."""
    assert is_version_at_least(version, minimum) is (version >= minimum)


def test_supported_versions_are_known() -> None:
    """Every negotiable revision must be in the ordering registry."""
    assert set(HANDSHAKE_PROTOCOL_VERSIONS) <= set(KNOWN_PROTOCOL_VERSIONS)
    assert set(MODERN_PROTOCOL_VERSIONS) <= set(KNOWN_PROTOCOL_VERSIONS)


def test_known_versions_are_strictly_ordered() -> None:
    """The registry tuple is the ordering source of truth: ascending, no duplicates."""
    assert list(KNOWN_PROTOCOL_VERSIONS) == sorted(set(KNOWN_PROTOCOL_VERSIONS))