Files
OpenShell/python/release_tooling_test.py
2026-09-01 16:12:01 +00:00

95 lines
3.4 KiB
Python

# SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0
from __future__ import annotations
import importlib.util
import sys
from pathlib import Path
def _load_release_module():
path = Path(__file__).resolve().parents[1] / "tasks/scripts/release.py"
spec = importlib.util.spec_from_file_location("openshell_release_tooling", path)
assert spec is not None
assert spec.loader is not None
module = importlib.util.module_from_spec(spec)
sys.modules[spec.name] = module
spec.loader.exec_module(module)
return module
release = _load_release_module()
def test_exact_tag_versions_are_stable_release_versions() -> None:
versions = release._versions_from_parts((0, 0, 37), 0, "152d05940", "v0.0.37")
assert versions.python == "0.0.37"
assert versions.cargo == "0.0.37"
assert versions.docker == "0.0.37"
assert versions.deb == "0.0.37-1"
assert versions.rpm_version == "0.0.37"
assert versions.rpm_release == "1"
def test_dev_versions_share_one_build_identity() -> None:
versions = release._versions_from_parts((0, 0, 37), 108, "152d05940", "v0.0.37")
assert versions.python == "0.0.38.dev108+g152d05940"
assert versions.cargo == "0.0.38-dev.108+g152d05940"
assert versions.docker == "0.0.38-dev.108-g152d05940"
assert versions.deb == "0.0.38~dev.108+g152d05940-1"
assert versions.rpm_version == "0.0.38"
assert versions.rpm_release == "0.dev.108.g152d05940"
def test_prerelease_versions_share_tag_identity() -> None:
versions = release._versions_from_prerelease(
(0, 1, 0), 1, "152d05940", "v0.1.0-pre.1"
)
assert versions.python == "0.1.0rc1"
assert versions.cargo == "0.1.0-pre.1"
assert versions.npm == "0.1.0-pre.1"
assert versions.docker == "0.1.0-pre.1"
assert versions.deb == "0.1.0~pre.1-1"
assert versions.snap == "0.1.0-pre.1"
assert versions.rpm_version == "0.1.0"
assert versions.rpm_release == "0.pre.1"
def test_semver_tag_parser_excludes_vm_tags() -> None:
assert release._parse_semver_tag("v0.0.37") == (0, 0, 37)
assert release._parse_semver_tag("0.0.37") == (0, 0, 37)
assert release._parse_semver_tag("v0.1.0-pre.1") is None
assert release._parse_semver_tag("vm-runtime") is None
assert release._parse_semver_tag("vm-dev") is None
def test_prerelease_tag_parser_requires_positive_sequence() -> None:
assert release._parse_prerelease_tag("v0.1.0-pre.1") == (0, 1, 0, 1)
assert release._parse_prerelease_tag("0.1.0-pre.12") == (0, 1, 0, 12)
assert release._parse_prerelease_tag("v0.1.0") is None
assert release._parse_prerelease_tag("v0.1.0-pre.0") is None
assert release._parse_prerelease_tag("v0.1.0-rc.1") is None
def test_dev_versions_ignore_exact_prerelease_tag(monkeypatch) -> None:
def fake_git(args: list[str]) -> str:
if args == ["rev-parse", "--short=9", "HEAD"]:
return "152d05940"
if args == ["tag", "--merged", "HEAD", "--list", "v*.*.*"]:
return "v0.0.37\nv0.1.0-pre.1"
if args == ["rev-list", "v0.0.37..HEAD", "--count"]:
return "108"
raise AssertionError(f"unexpected git command: {args}")
monkeypatch.setattr(release, "_git", fake_git)
versions = release._compute_dev_versions()
assert versions.python == "0.0.38.dev108+g152d05940"
assert versions.cargo == "0.0.38-dev.108+g152d05940"
assert versions.git_tag == "v0.0.37"