Add Git utilities, semantic version helpers, and unit tests
Ref: https://chatgpt.com/share/6936c64b-ae80-800f-ae7e-ed0a692ec3fc
This commit is contained in:
Kevin Veen-Birkenbach
124
tests/unit/pkgmgr/test_git_utils.py
Normal file
124
tests/unit/pkgmgr/test_git_utils.py
Normal file
@@ -0,0 +1,124 @@
|
||||
#!/usr/bin/env python3
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
import subprocess
|
||||
import unittest
|
||||
from types import SimpleNamespace
|
||||
from unittest.mock import patch
|
||||
|
||||
from pkgmgr.git_utils import (
|
||||
GitError,
|
||||
run_git,
|
||||
get_tags,
|
||||
get_head_commit,
|
||||
get_current_branch,
|
||||
)
|
||||
|
||||
|
||||
class TestGitUtils(unittest.TestCase):
|
||||
@patch("pkgmgr.git_utils.subprocess.run")
|
||||
def test_run_git_success(self, mock_run):
|
||||
mock_run.return_value = SimpleNamespace(
|
||||
stdout="ok\n",
|
||||
stderr="",
|
||||
returncode=0,
|
||||
)
|
||||
|
||||
output = run_git(["status"], cwd="/tmp/repo")
|
||||
|
||||
self.assertEqual(output, "ok")
|
||||
mock_run.assert_called_once()
|
||||
# basic sanity: command prefix should be 'git'
|
||||
args, kwargs = mock_run.call_args
|
||||
self.assertEqual(args[0][0], "git")
|
||||
self.assertEqual(kwargs.get("cwd"), "/tmp/repo")
|
||||
|
||||
@patch("pkgmgr.git_utils.subprocess.run")
|
||||
def test_run_git_failure_raises_giterror(self, mock_run):
|
||||
mock_run.side_effect = subprocess.CalledProcessError(
|
||||
returncode=1,
|
||||
cmd=["git", "status"],
|
||||
output="bad\n",
|
||||
stderr="error\n",
|
||||
)
|
||||
|
||||
with self.assertRaises(GitError) as ctx:
|
||||
run_git(["status"], cwd="/tmp/repo")
|
||||
|
||||
msg = str(ctx.exception)
|
||||
self.assertIn("Git command failed", msg)
|
||||
self.assertIn("Exit code: 1", msg)
|
||||
self.assertIn("bad", msg)
|
||||
self.assertIn("error", msg)
|
||||
|
||||
@patch("pkgmgr.git_utils.subprocess.run")
|
||||
def test_get_tags_empty(self, mock_run):
|
||||
mock_run.return_value = SimpleNamespace(
|
||||
stdout="",
|
||||
stderr="",
|
||||
returncode=0,
|
||||
)
|
||||
|
||||
tags = get_tags(cwd="/tmp/repo")
|
||||
self.assertEqual(tags, [])
|
||||
|
||||
@patch("pkgmgr.git_utils.subprocess.run")
|
||||
def test_get_tags_non_empty(self, mock_run):
|
||||
mock_run.return_value = SimpleNamespace(
|
||||
stdout="v1.0.0\nv1.1.0\n",
|
||||
stderr="",
|
||||
returncode=0,
|
||||
)
|
||||
|
||||
tags = get_tags(cwd="/tmp/repo")
|
||||
self.assertEqual(tags, ["v1.0.0", "v1.1.0"])
|
||||
|
||||
@patch("pkgmgr.git_utils.subprocess.run")
|
||||
def test_get_head_commit_success(self, mock_run):
|
||||
mock_run.return_value = SimpleNamespace(
|
||||
stdout="abc123\n",
|
||||
stderr="",
|
||||
returncode=0,
|
||||
)
|
||||
|
||||
commit = get_head_commit(cwd="/tmp/repo")
|
||||
self.assertEqual(commit, "abc123")
|
||||
|
||||
@patch("pkgmgr.git_utils.subprocess.run")
|
||||
def test_get_head_commit_failure_returns_none(self, mock_run):
|
||||
mock_run.side_effect = subprocess.CalledProcessError(
|
||||
returncode=1,
|
||||
cmd=["git", "rev-parse", "HEAD"],
|
||||
output="",
|
||||
stderr="error\n",
|
||||
)
|
||||
|
||||
commit = get_head_commit(cwd="/tmp/repo")
|
||||
self.assertIsNone(commit)
|
||||
|
||||
@patch("pkgmgr.git_utils.subprocess.run")
|
||||
def test_get_current_branch_success(self, mock_run):
|
||||
mock_run.return_value = SimpleNamespace(
|
||||
stdout="main\n",
|
||||
stderr="",
|
||||
returncode=0,
|
||||
)
|
||||
|
||||
branch = get_current_branch(cwd="/tmp/repo")
|
||||
self.assertEqual(branch, "main")
|
||||
|
||||
@patch("pkgmgr.git_utils.subprocess.run")
|
||||
def test_get_current_branch_failure_returns_none(self, mock_run):
|
||||
mock_run.side_effect = subprocess.CalledProcessError(
|
||||
returncode=1,
|
||||
cmd=["git", "rev-parse", "--abbrev-ref", "HEAD"],
|
||||
output="",
|
||||
stderr="error\n",
|
||||
)
|
||||
|
||||
branch = get_current_branch(cwd="/tmp/repo")
|
||||
self.assertIsNone(branch)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
120
tests/unit/pkgmgr/test_versioning.py
Normal file
120
tests/unit/pkgmgr/test_versioning.py
Normal file
@@ -0,0 +1,120 @@
|
||||
#!/usr/bin/env python3
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
import unittest
|
||||
|
||||
from pkgmgr.versioning import (
|
||||
SemVer,
|
||||
is_semver_tag,
|
||||
extract_semver_from_tags,
|
||||
find_latest_version,
|
||||
bump_major,
|
||||
bump_minor,
|
||||
bump_patch,
|
||||
)
|
||||
|
||||
|
||||
class TestSemVer(unittest.TestCase):
|
||||
def test_semver_parse_basic(self):
|
||||
ver = SemVer.parse("1.2.3")
|
||||
self.assertEqual(ver.major, 1)
|
||||
self.assertEqual(ver.minor, 2)
|
||||
self.assertEqual(ver.patch, 3)
|
||||
|
||||
def test_semver_parse_with_v_prefix(self):
|
||||
ver = SemVer.parse("v10.20.30")
|
||||
self.assertEqual(ver.major, 10)
|
||||
self.assertEqual(ver.minor, 20)
|
||||
self.assertEqual(ver.patch, 30)
|
||||
|
||||
def test_semver_parse_invalid(self):
|
||||
invalid_values = ["", "1", "1.2", "1.2.3.4", "a.b.c", "v1.2.x"]
|
||||
for value in invalid_values:
|
||||
with self.subTest(value=value):
|
||||
with self.assertRaises(ValueError):
|
||||
SemVer.parse(value)
|
||||
|
||||
def test_semver_to_tag_and_str(self):
|
||||
ver = SemVer(1, 2, 3)
|
||||
self.assertEqual(ver.to_tag(), "v1.2.3")
|
||||
self.assertEqual(ver.to_tag(with_prefix=False), "1.2.3")
|
||||
self.assertEqual(str(ver), "1.2.3")
|
||||
|
||||
def test_is_semver_tag(self):
|
||||
cases = [
|
||||
("1.2.3", True),
|
||||
("v1.2.3", True),
|
||||
("v0.0.0", True),
|
||||
("1.2", False),
|
||||
("foo", False),
|
||||
("v1.2.x", False),
|
||||
]
|
||||
for tag, expected in cases:
|
||||
with self.subTest(tag=tag):
|
||||
self.assertEqual(is_semver_tag(tag), expected)
|
||||
|
||||
def test_extract_semver_from_tags_all(self):
|
||||
tags = ["v1.2.3", "1.0.0", "not-a-tag", "v2.0.1"]
|
||||
result = extract_semver_from_tags(tags)
|
||||
tag_strings = [t for (t, _v) in result]
|
||||
|
||||
self.assertIn("v1.2.3", tag_strings)
|
||||
self.assertIn("1.0.0", tag_strings)
|
||||
self.assertIn("v2.0.1", tag_strings)
|
||||
self.assertNotIn("not-a-tag", tag_strings)
|
||||
|
||||
def test_extract_semver_from_tags_filter_major(self):
|
||||
tags = ["v1.2.3", "v1.3.0", "v2.0.0", "v2.1.0"]
|
||||
result = extract_semver_from_tags(tags, major=1)
|
||||
tag_strings = {t for (t, _v) in result}
|
||||
self.assertEqual(tag_strings, {"v1.2.3", "v1.3.0"})
|
||||
|
||||
def test_extract_semver_from_tags_filter_major_and_minor(self):
|
||||
tags = ["v1.2.3", "v1.2.4", "v1.3.0", "v2.2.0"]
|
||||
result = extract_semver_from_tags(tags, major=1, minor=2)
|
||||
tag_strings = {t for (t, _v) in result}
|
||||
self.assertEqual(tag_strings, {"v1.2.3", "v1.2.4"})
|
||||
|
||||
def test_find_latest_version_simple(self):
|
||||
tags = ["v1.2.3", "v1.2.4", "v2.0.0"]
|
||||
latest = find_latest_version(tags)
|
||||
self.assertIsNotNone(latest)
|
||||
tag, ver = latest
|
||||
self.assertEqual(tag, "v2.0.0")
|
||||
self.assertEqual((ver.major, ver.minor, ver.patch), (2, 0, 0))
|
||||
|
||||
def test_find_latest_version_filtered_major(self):
|
||||
tags = ["v1.2.3", "v1.4.0", "v2.0.0", "v2.1.0"]
|
||||
tag, ver = find_latest_version(tags, major=1)
|
||||
self.assertEqual(tag, "v1.4.0")
|
||||
self.assertEqual((ver.major, ver.minor, ver.patch), (1, 4, 0))
|
||||
|
||||
def test_find_latest_version_filtered_major_minor(self):
|
||||
tags = ["v1.2.3", "v1.2.4", "v1.3.0"]
|
||||
tag, ver = find_latest_version(tags, major=1, minor=2)
|
||||
self.assertEqual(tag, "v1.2.4")
|
||||
self.assertEqual((ver.major, ver.minor, ver.patch), (1, 2, 4))
|
||||
|
||||
def test_find_latest_version_no_match_returns_none(self):
|
||||
tags = ["not-semver", "also-bad"]
|
||||
latest = find_latest_version(tags)
|
||||
self.assertIsNone(latest)
|
||||
|
||||
def test_bump_major(self):
|
||||
ver = SemVer(1, 2, 3)
|
||||
bumped = bump_major(ver)
|
||||
self.assertEqual((bumped.major, bumped.minor, bumped.patch), (2, 0, 0))
|
||||
|
||||
def test_bump_minor(self):
|
||||
ver = SemVer(1, 2, 3)
|
||||
bumped = bump_minor(ver)
|
||||
self.assertEqual((bumped.major, bumped.minor, bumped.patch), (1, 3, 0))
|
||||
|
||||
def test_bump_patch(self):
|
||||
ver = SemVer(1, 2, 3)
|
||||
bumped = bump_patch(ver)
|
||||
self.assertEqual((bumped.major, bumped.minor, bumped.patch), (1, 2, 4))
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
Reference in New Issue
Block a user