129 lines
4.3 KiB
Python
129 lines
4.3 KiB
Python
"""Tests for repository-owned Python release versions."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
import runpy
|
|
import stat
|
|
from pathlib import Path
|
|
from types import SimpleNamespace
|
|
|
|
import pytest
|
|
|
|
|
|
ROOT = Path(__file__).resolve().parents[3]
|
|
SCRIPT = ROOT / "scripts" / "build-python-release.py"
|
|
build_python_release = SimpleNamespace(**runpy.run_path(str(SCRIPT)))
|
|
|
|
|
|
def helper_header(target: str) -> bytes:
|
|
header = bytearray(20)
|
|
if target.startswith("linux-"):
|
|
header[:6] = b"\x7fELF\x02\x01"
|
|
machine = 62 if target == "linux-x64" else 183
|
|
header[18:20] = machine.to_bytes(2, "little")
|
|
else:
|
|
header[:4] = b"\xcf\xfa\xed\xfe"
|
|
header[4:8] = (0x0100000C).to_bytes(4, "little")
|
|
return bytes(header)
|
|
|
|
|
|
def test_repository_version_matches_root_package_json() -> None:
|
|
expected = json.loads((ROOT / "package.json").read_text())["version"]
|
|
|
|
assert build_python_release.repository_version() == expected
|
|
|
|
|
|
def test_release_tag_is_optional_for_non_release_builds() -> None:
|
|
build_python_release.validate_release_tag(None, "1.2.3")
|
|
|
|
|
|
def test_release_tag_must_match_repository_version() -> None:
|
|
build_python_release.validate_release_tag("python-v1.2.3", "1.2.3")
|
|
|
|
with pytest.raises(ValueError, match="expected 'python-v1.2.3'"):
|
|
build_python_release.validate_release_tag("python-v1.2.4", "1.2.3")
|
|
|
|
|
|
def test_repository_version_rejects_non_stable_versions(tmp_path: Path) -> None:
|
|
(tmp_path / "package.json").write_text('{"version":"1.2.3-dev"}\n')
|
|
|
|
with pytest.raises(ValueError, match="must be stable X.Y.Z"):
|
|
build_python_release.repository_version(tmp_path)
|
|
|
|
|
|
def test_stage_runtime_copies_executable_and_spawn_helper(tmp_path: Path) -> None:
|
|
executable = tmp_path / "dsh-jsonrpc-agent-pkg-macos-arm64"
|
|
executable.write_bytes(b"runtime")
|
|
executable.chmod(0o755)
|
|
spawn_helper = Path(f"{executable}-spawn-helper")
|
|
spawn_helper.write_bytes(helper_header("macos-arm64"))
|
|
spawn_helper.chmod(0o751)
|
|
destination = tmp_path / "staging"
|
|
|
|
build_python_release.stage_runtime(
|
|
destination,
|
|
"1.2.3",
|
|
executable,
|
|
executable.name,
|
|
)
|
|
|
|
runtime_dir = destination / "src" / "deepseek_harness_runtime" / "runtime"
|
|
assert (runtime_dir / executable.name).read_bytes() == b"runtime"
|
|
copied_helper = runtime_dir / spawn_helper.name
|
|
assert copied_helper.read_bytes() == helper_header("macos-arm64")
|
|
assert copied_helper.stat().st_mode & stat.S_IXUSR
|
|
|
|
|
|
def test_stage_runtime_rejects_missing_spawn_helper(tmp_path: Path) -> None:
|
|
executable = tmp_path / "dsh-jsonrpc-agent-pkg-linux-x64"
|
|
executable.write_bytes(b"runtime")
|
|
executable.chmod(0o755)
|
|
|
|
with pytest.raises(FileNotFoundError, match="spawn helper"):
|
|
build_python_release.stage_runtime(
|
|
tmp_path / "staging",
|
|
"1.2.3",
|
|
executable,
|
|
executable.name,
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize("target", ["linux-x64", "linux-arm64", "macos-arm64"])
|
|
def test_spawn_helper_binary_target(target: str) -> None:
|
|
assert build_python_release.spawn_helper_binary_target(helper_header(target)) == target
|
|
|
|
|
|
def test_stage_runtime_rejects_mismatched_spawn_helper(tmp_path: Path) -> None:
|
|
executable = tmp_path / "dsh-jsonrpc-agent-pkg-linux-x64"
|
|
executable.write_bytes(b"runtime")
|
|
executable.chmod(0o755)
|
|
spawn_helper = Path(f"{executable}-spawn-helper")
|
|
spawn_helper.write_bytes(helper_header("linux-arm64"))
|
|
spawn_helper.chmod(0o755)
|
|
|
|
with pytest.raises(ValueError, match="expected linux-x64, found linux-arm64"):
|
|
build_python_release.stage_runtime(
|
|
tmp_path / "staging",
|
|
"1.2.3",
|
|
executable,
|
|
executable.name,
|
|
)
|
|
|
|
|
|
def test_stage_runtime_rejects_non_binary_spawn_helper(tmp_path: Path) -> None:
|
|
executable = tmp_path / "dsh-jsonrpc-agent-pkg-macos-arm64"
|
|
executable.write_bytes(b"runtime")
|
|
executable.chmod(0o755)
|
|
spawn_helper = Path(f"{executable}-spawn-helper")
|
|
spawn_helper.write_bytes(b"helper")
|
|
spawn_helper.chmod(0o755)
|
|
|
|
with pytest.raises(ValueError, match="unsupported format or architecture"):
|
|
build_python_release.stage_runtime(
|
|
tmp_path / "staging",
|
|
"1.2.3",
|
|
executable,
|
|
executable.name,
|
|
)
|