headroom/scripts/tests/test_version_sync.py

"""Tests for version-sync.py."""

import json
import subprocess
import sys
from pathlib import Path

import pytest


@pytest.fixture
def temp_project(tmp_path: Path) -> dict[str, Path]:
    """Create a temporary project with all versioned files."""
    # Create directory structure
    root = tmp_path / "project"
    headroom = root / "headroom"
    headroom.mkdir(parents=True)
    repo_claude_plugin = root / ".claude-plugin"
    repo_claude_plugin.mkdir(parents=True)
    repo_github_plugin = root / ".github" / "plugin"
    repo_github_plugin.mkdir(parents=True)
    plugins = root / "plugins"
    openclaw = plugins / "openclaw"
    openclaw.mkdir(parents=True)
    opencode = plugins / "opencode"
    opencode.mkdir(parents=True)
    agent_hooks_claude = plugins / "headroom-agent-hooks" / ".claude-plugin"
    agent_hooks_claude.mkdir(parents=True)
    agent_hooks_github = plugins / "headroom-agent-hooks" / ".github" / "plugin"
    agent_hooks_github.mkdir(parents=True)
    sdk = root / "sdk"
    typescript = sdk / "typescript"
    typescript.mkdir(parents=True)

    # pyproject.toml
    pyproject = root / "pyproject.toml"
    pyproject.write_text('[project]\nversion = "0.5.25"\n')

    # headroom/_version.py is runtime-derived and must not be rewritten by version-sync.
    version_py = headroom / "_version.py"
    version_py.write_text('"""Package version metadata."""\n\n__version__ = "0.5.25"\n')

    # plugins/openclaw/package.json
    openclaw_pkg = openclaw / "package.json"
    openclaw_pkg.write_text(
        json.dumps(
            {
                "name": "test",
                "version": "0.5.25",
                "dependencies": {"headroom-ai": "^0.22.3"},
            }
        )
    )

    opencode_pkg = opencode / "package.json"
    opencode_pkg.write_text(
        json.dumps(
            {
                "name": "test-opencode",
                "version": "0.5.25",
                "dependencies": {"headroom-ai": "^0.22.3"},
            }
        )
    )

    repo_claude_marketplace = repo_claude_plugin / "marketplace.json"
    repo_claude_marketplace.write_text(
        json.dumps(
            {
                "metadata": {"name": "claude-marketplace", "version": "0.1.0"},
                "plugins": [{"name": "headroom-agent-hooks", "version": "0.1.0"}],
            }
        )
    )

    repo_github_marketplace = repo_github_plugin / "marketplace.json"
    repo_github_marketplace.write_text(
        json.dumps(
            {
                "metadata": {"name": "copilot-marketplace", "version": "0.1.0"},
                "plugins": [{"name": "headroom-agent-hooks", "version": "0.1.0"}],
            }
        )
    )

    claude_plugin = agent_hooks_claude / "plugin.json"
    claude_plugin.write_text(json.dumps({"name": "headroom-agent-hooks", "version": "0.1.0"}))

    github_plugin = agent_hooks_github / "plugin.json"
    github_plugin.write_text(json.dumps({"name": "headroom-agent-hooks", "version": "0.1.0"}))

    # sdk/typescript/package.json
    typescript_pkg = typescript / "package.json"
    typescript_pkg.write_text(json.dumps({"name": "test", "version": "0.5.25"}))

    # server.json — the MCP registry descriptor. Asserted byte-for-byte against
    # render_server_json(), which reads the version from pyproject.toml, so it has
    # to move with every bump or the release PR's test job fails.
    server_json = root / "server.json"
    server_json.write_text(
        json.dumps(
            {
                "name": "io.github.headroomlabs-ai/headroom",
                "version": "0.5.25",
                "packages": [{"registryType": "pypi", "version": "0.5.25"}],
            }
        )
    )

    return {
        "root": root,
        "pyproject": pyproject,
        "version_py": version_py,
        "openclaw_pkg": openclaw_pkg,
        "opencode_pkg": opencode_pkg,
        "repo_claude_marketplace": repo_claude_marketplace,
        "repo_github_marketplace": repo_github_marketplace,
        "claude_plugin": claude_plugin,
        "github_plugin": github_plugin,
        "typescript_pkg": typescript_pkg,
        "server_json": server_json,
    }


def test_version_sync_explicit_version(temp_project: dict[str, Path]) -> None:
    """Test --version flag updates all files."""
    root = temp_project["root"]
    script = Path(__file__).parent.parent / "version-sync.py"

    result = subprocess.run(
        [sys.executable, str(script), "--root", str(root), "--version", "0.7.0"],
        capture_output=True,
        text=True,
    )

    assert result.returncode == 0, f"Script failed: {result.stderr}"

    # Verify pyproject.toml
    pyproject_content = temp_project["pyproject"].read_text()
    assert 'version = "0.7.0"' in pyproject_content

    # Verify headroom/_version.py is not a synced manifest.
    version_py_content = temp_project["version_py"].read_text()
    assert '__version__ = "0.5.25"' in version_py_content

    # Verify plugins/openclaw/package.json
    openclaw_pkg = json.loads(temp_project["openclaw_pkg"].read_text())
    assert openclaw_pkg["version"] == "0.7.0"
    assert openclaw_pkg["dependencies"]["headroom-ai"] == "^0.22.3"

    opencode_pkg = json.loads(temp_project["opencode_pkg"].read_text())
    assert opencode_pkg["version"] == "0.7.0"
    assert opencode_pkg["dependencies"]["headroom-ai"] == "^0.22.3"

    # Verify sdk/typescript/package.json
    typescript_pkg = json.loads(temp_project["typescript_pkg"].read_text())
    assert typescript_pkg["version"] == "0.7.0"

    repo_claude_marketplace = json.loads(temp_project["repo_claude_marketplace"].read_text())
    assert repo_claude_marketplace["metadata"]["version"] == "0.7.0"
    assert repo_claude_marketplace["plugins"][0]["version"] == "0.7.0"

    repo_github_marketplace = json.loads(temp_project["repo_github_marketplace"].read_text())
    assert repo_github_marketplace["metadata"]["version"] == "0.7.0"
    assert repo_github_marketplace["plugins"][0]["version"] == "0.7.0"

    claude_plugin = json.loads(temp_project["claude_plugin"].read_text())
    assert claude_plugin["version"] == "0.7.0"

    github_plugin = json.loads(temp_project["github_plugin"].read_text())
    assert github_plugin["version"] == "0.7.0"

    # Verify .releasemetadata was created
    release_metadata = root / ".releasemetadata"
    assert release_metadata.exists()
    metadata = json.loads(release_metadata.read_text())
    assert metadata["version"] == "0.7.0"
    assert metadata["packages"]["pypi"] == "0.7.0"
    assert metadata["packages"]["npm-sdk"] == "0.7.0"
    assert metadata["packages"]["npm-openclaw"] == "0.7.0"
    assert metadata["packages"]["npm-opencode"] == "0.7.0"
    assert metadata["packages"]["agent-hooks-plugin"] == "0.7.0"


def test_bump_patch(temp_project: dict[str, Path]) -> None:
    """Test --bump patch bumps 0.5.25 to 0.5.26."""
    root = temp_project["root"]
    script = Path(__file__).parent.parent / "version-sync.py"

    result = subprocess.run(
        [sys.executable, str(script), "--root", str(root), "--bump", "patch"],
        capture_output=True,
        text=True,
    )

    assert result.returncode == 0, f"Script failed: {result.stderr}"

    # Verify all files updated to 0.5.26
    pyproject_content = temp_project["pyproject"].read_text()
    assert 'version = "0.5.26"' in pyproject_content

    version_py_content = temp_project["version_py"].read_text()
    assert '__version__ = "0.5.25"' in version_py_content

    openclaw_pkg = json.loads(temp_project["openclaw_pkg"].read_text())
    assert openclaw_pkg["version"] == "0.5.26"
    assert openclaw_pkg["dependencies"]["headroom-ai"] == "^0.22.3"

    opencode_pkg = json.loads(temp_project["opencode_pkg"].read_text())
    assert opencode_pkg["version"] == "0.5.26"
    assert opencode_pkg["dependencies"]["headroom-ai"] == "^0.22.3"

    typescript_pkg = json.loads(temp_project["typescript_pkg"].read_text())
    assert typescript_pkg["version"] == "0.5.26"

    claude_plugin = json.loads(temp_project["claude_plugin"].read_text())
    assert claude_plugin["version"] == "0.5.26"


def test_bump_minor(temp_project: dict[str, Path]) -> None:
    """Test --bump minor bumps 0.5.25 to 0.6.0."""
    root = temp_project["root"]
    script = Path(__file__).parent.parent / "version-sync.py"

    result = subprocess.run(
        [sys.executable, str(script), "--root", str(root), "--bump", "minor"],
        capture_output=True,
        text=True,
    )

    assert result.returncode == 0, f"Script failed: {result.stderr}"

    # Verify all files updated to 0.6.0
    pyproject_content = temp_project["pyproject"].read_text()
    assert 'version = "0.6.0"' in pyproject_content

    version_py_content = temp_project["version_py"].read_text()
    assert '__version__ = "0.5.25"' in version_py_content

    openclaw_pkg = json.loads(temp_project["openclaw_pkg"].read_text())
    assert openclaw_pkg["version"] == "0.6.0"
    assert openclaw_pkg["dependencies"]["headroom-ai"] == "^0.22.3"

    opencode_pkg = json.loads(temp_project["opencode_pkg"].read_text())
    assert opencode_pkg["version"] == "0.6.0"
    assert opencode_pkg["dependencies"]["headroom-ai"] == "^0.22.3"

    typescript_pkg = json.loads(temp_project["typescript_pkg"].read_text())
    assert typescript_pkg["version"] == "0.6.0"

    github_plugin = json.loads(temp_project["github_plugin"].read_text())
    assert github_plugin["version"] == "0.6.0"


def test_bump_major(temp_project: dict[str, Path]) -> None:
    """Test --bump major bumps 0.5.25 to 1.0.0."""
    root = temp_project["root"]
    script = Path(__file__).parent.parent / "version-sync.py"

    result = subprocess.run(
        [sys.executable, str(script), "--root", str(root), "--bump", "major"],
        capture_output=True,
        text=True,
    )

    assert result.returncode == 0, f"Script failed: {result.stderr}"

    # Verify all files updated to 1.0.0
    pyproject_content = temp_project["pyproject"].read_text()
    assert 'version = "1.0.0"' in pyproject_content

    version_py_content = temp_project["version_py"].read_text()
    assert '__version__ = "0.5.25"' in version_py_content

    openclaw_pkg = json.loads(temp_project["openclaw_pkg"].read_text())
    assert openclaw_pkg["version"] == "1.0.0"
    assert openclaw_pkg["dependencies"]["headroom-ai"] == "^0.22.3"

    opencode_pkg = json.loads(temp_project["opencode_pkg"].read_text())
    assert opencode_pkg["version"] == "1.0.0"
    assert opencode_pkg["dependencies"]["headroom-ai"] == "^0.22.3"

    typescript_pkg = json.loads(temp_project["typescript_pkg"].read_text())
    assert typescript_pkg["version"] == "1.0.0"

    repo_claude_marketplace = json.loads(temp_project["repo_claude_marketplace"].read_text())
    assert repo_claude_marketplace["metadata"]["version"] == "1.0.0"


def test_release_metadata_written(temp_project: dict[str, Path]) -> None:
    """Test .releasemetadata is written correctly."""
    root = temp_project["root"]
    script = Path(__file__).parent.parent / "version-sync.py"

    result = subprocess.run(
        [sys.executable, str(script), "--root", str(root), "--version", "0.6.0"],
        capture_output=True,
        text=True,
    )

    assert result.returncode == 0, f"Script failed: {result.stderr}"

    release_metadata = root / ".releasemetadata"
    assert release_metadata.exists()

    metadata = json.loads(release_metadata.read_text())
    assert metadata == {
        "version": "0.6.0",
        "packages": {
            "pypi": "0.6.0",
            "npm-sdk": "0.6.0",
            "npm-openclaw": "0.6.0",
            "npm-opencode": "0.6.0",
            "agent-hooks-plugin": "0.6.0",
        },
    }


def test_plugin_manifests_only_leaves_package_versions_unchanged(
    temp_project: dict[str, Path],
) -> None:
    """Test plugin-only sync leaves canonical package versions alone."""
    root = temp_project["root"]
    script = Path(__file__).parent.parent / "version-sync.py"

    result = subprocess.run(
        [
            sys.executable,
            str(script),
            "--root",
            str(root),
            "--version",
            "0.8.0",
            "--plugin-manifests-only",
        ],
        capture_output=True,
        text=True,
    )

    assert result.returncode == 0, f"Script failed: {result.stderr}"
    assert 'version = "0.5.25"' in temp_project["pyproject"].read_text()
    assert '__version__ = "0.5.25"' in temp_project["version_py"].read_text()
    assert json.loads(temp_project["openclaw_pkg"].read_text())["version"] == "0.5.25"
    assert json.loads(temp_project["opencode_pkg"].read_text())["version"] == "0.5.25"
    assert json.loads(temp_project["typescript_pkg"].read_text())["version"] == "0.5.25"
    assert json.loads(temp_project["claude_plugin"].read_text())["version"] == "0.8.0"
    assert (
        json.loads(temp_project["repo_github_marketplace"].read_text())["metadata"]["version"]
        == "0.8.0"
    )
    assert not (root / ".releasemetadata").exists()


def test_openclaw_headroom_dependency_is_preserved_for_registry_installability(
    temp_project: dict[str, Path],
) -> None:
    """Source package stays installable even when the next SDK is not on npm yet."""
    root = temp_project["root"]
    script = Path(__file__).parent.parent / "version-sync.py"

    result = subprocess.run(
        [sys.executable, str(script), "--root", str(root), "--version", "0.28.0"],
        capture_output=True,
        text=True,
    )

    assert result.returncode == 0, f"Script failed: {result.stderr}"
    openclaw_pkg = json.loads(temp_project["openclaw_pkg"].read_text())
    assert openclaw_pkg["version"] == "0.28.0"
    assert openclaw_pkg["dependencies"]["headroom-ai"] == "^0.22.3"


def test_server_json_version_is_synchronized(temp_project: dict[str, Path]) -> None:
    """server.json must track the bump or the release PR's test job fails.

    ``tests/test_mcp_registry/test_server_json.py::test_root_server_json_matches_builder``
    asserts the tracked file equals ``render_server_json()``, which reads the version
    from ``pyproject.toml``. Nothing regenerated server.json, so it fell behind every
    release and blocked v0.33.0 (PR #2339).
    """
    root = temp_project["root"]
    script = Path(__file__).parent.parent / "version-sync.py"

    result = subprocess.run(
        [sys.executable, str(script), "--root", str(root), "--version", "0.33.0"],
        capture_output=True,
        text=True,
    )

    assert result.returncode == 0, f"Script failed: {result.stderr}"
    server_json = json.loads(temp_project["server_json"].read_text())
    assert server_json["version"] == "0.33.0"
    # The packages[] entry carries its own version and is checked by the builder too.
    assert [p["version"] for p in server_json["packages"]] == ["0.33.0"]
    # Untouched keys must survive so the file still matches the builder's output.
    assert server_json["name"] == "io.github.headroomlabs-ai/headroom"
    assert server_json["packages"][0]["registryType"] == "pypi"


def test_opencode_headroom_dependency_is_preserved_for_registry_installability(
    temp_project: dict[str, Path],
) -> None:
    """Source package stays installable even when the next SDK is not on npm yet."""
    root = temp_project["root"]
    script = Path(__file__).parent.parent / "version-sync.py"

    result = subprocess.run(
        [sys.executable, str(script), "--root", str(root), "--version", "0.28.0"],
        capture_output=True,
        text=True,
    )

    assert result.returncode == 0, f"Script failed: {result.stderr}"
    opencode_pkg = json.loads(temp_project["opencode_pkg"].read_text())
    assert opencode_pkg["version"] == "0.28.0"
    assert opencode_pkg["dependencies"]["headroom-ai"] == "^0.22.3"