#!/usr/bin/env python3 import unittest from pathlib import Path from utils import compare_versions, normalize_version, parse_version, validate_version class TestVersionUtils(unittest.TestCase): def test_normalize_version(self): """Test version normalization removes 'v' prefix.""" self.assertEqual(normalize_version("v0.5.3"), "0.5.3") self.assertEqual(normalize_version("0.5.3"), "0.5.3") self.assertEqual(normalize_version("v0.5.3rc0"), "0.5.3rc0") self.assertEqual(normalize_version("0.5.3.post1"), "0.5.3.post1") def test_validate_version(self): """Test version format validation.""" # Valid formats self.assertTrue(validate_version("0.5.3")) self.assertTrue(validate_version("0.5.3rc0")) self.assertTrue(validate_version("0.5.3rc1")) self.assertTrue(validate_version("0.5.3rc999")) self.assertTrue(validate_version("0.5.3.post1")) self.assertTrue(validate_version("0.5.3.post10")) self.assertTrue(validate_version("1.2.3")) self.assertTrue(validate_version("10.20.30")) # Invalid formats self.assertFalse(validate_version("0.5")) self.assertFalse(validate_version("0.5.3.")) self.assertFalse(validate_version("0.5.3rc")) self.assertFalse(validate_version("0.5.3post1")) self.assertFalse(validate_version("0.5.3-rc0")) self.assertFalse(validate_version("v0.5.3")) self.assertFalse(validate_version("0.5.3beta1")) self.assertFalse(validate_version("0.5.3.rc0")) def test_parse_version_stable(self): """Test parsing stable version.""" self.assertEqual(parse_version("0.5.3"), (0, 5, 3, 0, 0)) self.assertEqual(parse_version("1.2.3"), (1, 2, 3, 0, 0)) self.assertEqual(parse_version("10.20.30"), (10, 20, 30, 0, 0)) def test_parse_version_rc(self): """Test parsing release candidate versions.""" self.assertEqual(parse_version("0.5.3rc0"), (0, 5, 3, -1000, 0)) self.assertEqual(parse_version("0.5.3rc1"), (0, 5, 3, -999, 0)) self.assertEqual(parse_version("0.5.3rc2"), (0, 5, 3, -998, 0)) self.assertEqual(parse_version("0.5.3rc10"), (0, 5, 3, -990, 0)) def test_parse_version_post(self): """Test parsing post-release versions.""" self.assertEqual(parse_version("0.5.3.post1"), (0, 5, 3, 0, 1)) self.assertEqual(parse_version("0.5.3.post2"), (0, 5, 3, 0, 2)) self.assertEqual(parse_version("0.5.3.post10"), (0, 5, 3, 0, 10)) def test_parse_version_invalid(self): """Test parsing invalid versions raises error.""" with self.assertRaises(ValueError): parse_version("0.5") with self.assertRaises(ValueError): parse_version("invalid") with self.assertRaises(ValueError): parse_version("v0.5.3") def test_compare_versions_equal(self): """Test comparing equal versions.""" self.assertEqual(compare_versions("0.5.3", "0.5.3"), 0) self.assertEqual(compare_versions("0.5.3rc0", "0.5.3rc0"), 0) self.assertEqual(compare_versions("0.5.3.post1", "0.5.3.post1"), 0) def test_compare_versions_rc_ordering(self): """Test release candidate ordering: rc0 < rc1 < rc2 < stable.""" # rc0 < rc1 self.assertEqual(compare_versions("0.5.3rc0", "0.5.3rc1"), -1) self.assertEqual(compare_versions("0.5.3rc1", "0.5.3rc0"), 1) # rc1 < rc2 self.assertEqual(compare_versions("0.5.3rc1", "0.5.3rc2"), -1) self.assertEqual(compare_versions("0.5.3rc2", "0.5.3rc1"), 1) # rc < stable self.assertEqual(compare_versions("0.5.3rc0", "0.5.3"), -1) self.assertEqual(compare_versions("0.5.3rc1", "0.5.3"), -1) self.assertEqual(compare_versions("0.5.3", "0.5.3rc0"), 1) def test_compare_versions_post_ordering(self): """Test post-release ordering: stable < post1 < post2.""" # stable < post1 self.assertEqual(compare_versions("0.5.3", "0.5.3.post1"), -1) self.assertEqual(compare_versions("0.5.3.post1", "0.5.3"), 1) # post1 < post2 self.assertEqual(compare_versions("0.5.3.post1", "0.5.3.post2"), -1) self.assertEqual(compare_versions("0.5.3.post2", "0.5.3.post1"), 1) def test_compare_versions_full_ordering(self): """Test complete version ordering: rc < stable < post.""" # rc < stable < post self.assertEqual(compare_versions("0.5.3rc0", "0.5.3"), -1) self.assertEqual(compare_versions("0.5.3", "0.5.3.post1"), -1) self.assertEqual(compare_versions("0.5.3rc0", "0.5.3.post1"), -1) # Verify transitivity: rc0 < rc1 < stable < post1 < post2 versions = [ "0.5.3rc0", "0.5.3rc1", "0.5.3", "0.5.3.post1", "0.5.3.post2", ] for i in range(len(versions) - 1): self.assertEqual( compare_versions(versions[i], versions[i + 1]), -1, f"{versions[i]} should be less than {versions[i + 1]}", ) def test_compare_versions_different_patch(self): """Test comparing versions with different patch numbers.""" # 0.5.3 < 0.5.4 self.assertEqual(compare_versions("0.5.3", "0.5.4"), -1) self.assertEqual(compare_versions("0.5.4", "0.5.3"), 1) # rc of higher patch > stable of lower patch self.assertEqual(compare_versions("0.5.4rc0", "0.5.3"), 1) self.assertEqual(compare_versions("0.5.3.post1", "0.5.4rc0"), -1) def test_compare_versions_different_minor(self): """Test comparing versions with different minor numbers.""" self.assertEqual(compare_versions("0.4.9", "0.5.0"), -1) self.assertEqual(compare_versions("0.5.0", "0.4.9"), 1) def test_compare_versions_different_major(self): """Test comparing versions with different major numbers.""" self.assertEqual(compare_versions("0.9.9", "1.0.0"), -1) self.assertEqual(compare_versions("1.0.0", "0.9.9"), 1) def test_real_world_scenarios(self): """Test real-world version bump scenarios.""" # Scenario 1: RC progression self.assertEqual(compare_versions("0.5.3rc0", "0.5.3rc1"), -1) # Scenario 2: RC to stable release self.assertEqual(compare_versions("0.5.3rc2", "0.5.3"), -1) # Scenario 3: Stable to post-release hotfix self.assertEqual(compare_versions("0.5.3", "0.5.3.post1"), -1) # Scenario 4: Post-release to next RC self.assertEqual(compare_versions("0.5.3.post1", "0.5.4rc0"), -1) # Scenario 5: Next stable version self.assertEqual(compare_versions("0.5.3", "0.5.4"), -1) if __name__ == "__main__": unittest.main()