~bumpversion

tests/cli/test_bumpversion.py

@@ -2,105 +2,253 @@
 
 from __future__ import annotations
 
-from unittest.mock import MagicMock, patch
+from pathlib import Path
 
 import pytest
 
-import pyflowx as px
 from pyflowx.cli import bumpversion
 
 
 # ---------------------------------------------------------------------- #
-# bump_version
+# bump_file_version
 # ---------------------------------------------------------------------- #
-class TestBumpVersion:
-    """Test bump_version function."""
+class TestBumpFileVersion:
+    """Test bump_file_version function."""
 
-    def test_bump_version_patch(self) -> None:
-        """Should bump patch version."""
-        with patch("subprocess.run") as mock_run:
-            mock_run.return_value = MagicMock(returncode=0)
-            bumpversion.bump_version("patch")
-            assert mock_run.called
+    def test_bump_patch_version(self, tmp_path: Path) -> None:
+        """Should bump patch version correctly."""
+        test_file = tmp_path / "test.txt"
+        test_file.write_text("version = 1.2.3", encoding="utf-8")
 
-    def test_bump_version_minor(self) -> None:
-        """Should bump minor version."""
-        with patch("subprocess.run") as mock_run:
-            mock_run.return_value = MagicMock(returncode=0)
-            bumpversion.bump_version("minor")
-            assert mock_run.called
+        result = bumpversion.bump_file_version(test_file, "patch")
 
-    def test_bump_version_major(self) -> None:
-        """Should bump major version."""
-        with patch("subprocess.run") as mock_run:
-            mock_run.return_value = MagicMock(returncode=0)
-            bumpversion.bump_version("major")
-            assert mock_run.called
+        assert result == "1.2.4"
+        assert test_file.read_text(encoding="utf-8") == "version = 1.2.4"
 
-    def test_bump_version_with_tag(self) -> None:
-        """Should bump version with tag."""
-        with patch("subprocess.run") as mock_run:
-            mock_run.return_value = MagicMock(returncode=0, stdout="v1.0.0")
-            bumpversion.bump_version("patch", tag=True)
-            assert mock_run.called
+    def test_bump_minor_version(self, tmp_path: Path) -> None:
+        """Should bump minor version correctly."""
+        test_file = tmp_path / "test.txt"
+        test_file.write_text("version = 1.2.3", encoding="utf-8")
 
-    def test_bump_version_with_commit(self) -> None:
-        """Should bump version with commit."""
-        with patch("subprocess.run") as mock_run:
-            mock_run.return_value = MagicMock(returncode=0)
-            bumpversion.bump_version("patch", commit=True)
-            assert mock_run.called
+        result = bumpversion.bump_file_version(test_file, "minor")
 
-    def test_bump_version_file_not_found(self) -> None:
-        """Should handle FileNotFoundError."""
-        with patch("subprocess.run", side_effect=FileNotFoundError), pytest.raises(FileNotFoundError):
-            bumpversion.bump_version("patch")
+        assert result == "1.3.0"
+        assert test_file.read_text(encoding="utf-8") == "version = 1.3.0"
+
+    def test_bump_major_version(self, tmp_path: Path) -> None:
+        """Should bump major version correctly."""
+        test_file = tmp_path / "test.txt"
+        test_file.write_text("version = 1.2.3", encoding="utf-8")
+
+        result = bumpversion.bump_file_version(test_file, "major")
+
+        assert result == "2.0.0"
+        assert test_file.read_text(encoding="utf-8") == "version = 2.0.0"
+
+    def test_version_pattern_with_prerelease(self, tmp_path: Path) -> None:
+        """Should handle version with prerelease suffix."""
+        test_file = tmp_path / "test.txt"
+        test_file.write_text("version = 1.2.3-alpha.1", encoding="utf-8")
+
+        result = bumpversion.bump_file_version(test_file, "patch")
+
+        assert result == "1.2.4"
+        # 预发布版本应该被清除
+        content = test_file.read_text(encoding="utf-8")
+        assert "alpha" not in content
+
+    def test_version_pattern_with_build_metadata(self, tmp_path: Path) -> None:
+        """Should handle version with build metadata."""
+        test_file = tmp_path / "test.txt"
+        test_file.write_text("version = 1.2.3+build.123", encoding="utf-8")
+
+        result = bumpversion.bump_file_version(test_file, "patch")
+
+        assert result == "1.2.4"
+        # 构建元数据应该被清除
+        content = test_file.read_text(encoding="utf-8")
+        assert "build" not in content
+
+    def test_no_version_found(self, tmp_path: Path, capsys) -> None:
+        """Should return None when no version pattern found."""
+        test_file = tmp_path / "test.txt"
+        test_file.write_text("no version here", encoding="utf-8")
+
+        result = bumpversion.bump_file_version(test_file, "patch")
+
+        assert result is None
+        captured = capsys.readouterr()
+        assert "未找到版本号模式" in captured.out
+
+    def test_utf8_encoding(self, tmp_path: Path) -> None:
+        """Should handle UTF-8 encoded files correctly."""
+        test_file = tmp_path / "test.txt"
+        test_file.write_text("版本 = 1.2.3", encoding="utf-8")
+
+        result = bumpversion.bump_file_version(test_file, "patch")
+
+        assert result == "1.2.4"
+        assert test_file.read_text(encoding="utf-8") == "版本 = 1.2.4"
+
+    def test_pyproject_toml_format(self, tmp_path: Path) -> None:
+        """Should handle pyproject.toml format correctly."""
+        test_file = tmp_path / "pyproject.toml"
+        content = """
+[project]
+name = "test"
+version = "0.1.0"
+description = "Test project"
+"""
+        test_file.write_text(content, encoding="utf-8")
+
+        result = bumpversion.bump_file_version(test_file, "minor")
+
+        assert result == "0.2.0"
+        updated = test_file.read_text(encoding="utf-8")
+        assert 'version = "0.2.0"' in updated
+        assert 'name = "test"' in updated
+
+    def test_init_py_format(self, tmp_path: Path) -> None:
+        """Should handle __init__.py format correctly."""
+        test_file = tmp_path / "__init__.py"
+        content = '''"""Package info."""
+
+__version__ = "1.0.0"
+'''
+        test_file.write_text(content, encoding="utf-8")
+
+        result = bumpversion.bump_file_version(test_file, "major")
+
+        assert result == "2.0.0"
+        updated = test_file.read_text(encoding="utf-8")
+        assert '__version__ = "2.0.0"' in updated
+
+    def test_multiple_versions_in_file(self, tmp_path: Path) -> None:
+        """Should only bump the first version occurrence."""
+        test_file = tmp_path / "test.txt"
+        test_file.write_text("version1 = 1.0.0\nversion2 = 2.0.0", encoding="utf-8")
+
+        result = bumpversion.bump_file_version(test_file, "patch")
+
+        assert result == "1.0.1"
+        content = test_file.read_text(encoding="utf-8")
+        assert "version1 = 1.0.1" in content
+        assert "version2 = 2.0.0" in content
+
+    def test_file_read_error(self, tmp_path: Path, capsys) -> None:
+        """Should handle file read errors."""
+        # 创建一个目录而不是文件
+        test_file = tmp_path / "test_dir"
+        test_file.mkdir()
+
+        with pytest.raises(Exception):  # noqa: B017
+            bumpversion.bump_file_version(test_file, "patch")
+
+    def test_file_write_error(self, tmp_path: Path, capsys) -> None:
+        """Should handle file write errors."""
+        # 在只读目录中创建文件（这个测试在某些系统上可能不适用）
+        test_file = tmp_path / "readonly.txt"
+        test_file.write_text("version = 1.0.0", encoding="utf-8")
+        # 设置为只读
+        test_file.chmod(0o444)
+
+        try:
+            with pytest.raises(Exception):
+                bumpversion.bump_file_version(test_file, "patch")
+        finally:
+            # 恢复权限以便清理
+            test_file.chmod(0o644)
 
 
 # ---------------------------------------------------------------------- #
-# bump_version_alpha
+# Version pattern tests
 # ---------------------------------------------------------------------- #
-class TestBumpVersionAlpha:
-    """Test bump_version_alpha function."""
+class TestVersionPattern:
+    """Test version pattern matching."""
 
-    def test_bump_version_alpha_patch(self) -> None:
-        """Should bump alpha patch version."""
-        with patch("subprocess.run") as mock_run:
-            mock_run.return_value = MagicMock(returncode=0)
-            bumpversion.bump_version_alpha("patch")
-            assert mock_run.called
+    def test_simple_version(self, tmp_path: Path) -> None:
+        """Should match simple version."""
+        test_file = tmp_path / "test.txt"
+        test_file.write_text("1.0.0", encoding="utf-8")
+
+        result = bumpversion.bump_file_version(test_file, "patch")
+
+        assert result == "1.0.1"
+
+    def test_version_with_zeros(self, tmp_path: Path) -> None:
+        """Should handle versions with zeros correctly."""
+        test_file = tmp_path / "test.txt"
+        test_file.write_text("0.0.0", encoding="utf-8")
+
+        result = bumpversion.bump_file_version(test_file, "patch")
+
+        assert result == "0.0.1"
+
+    def test_large_version_numbers(self, tmp_path: Path) -> None:
+        """Should handle large version numbers."""
+        test_file = tmp_path / "test.txt"
+        test_file.write_text("10.20.30", encoding="utf-8")
+
+        result = bumpversion.bump_file_version(test_file, "minor")
+
+        assert result == "10.21.0"
+
+    def test_version_in_url(self, tmp_path: Path) -> None:
+        """Should only match first version in complex text."""
+        test_file = tmp_path / "test.txt"
+        test_file.write_text("https://example.com/v1.2.3/download", encoding="utf-8")
+
+        result = bumpversion.bump_file_version(test_file, "patch")
+
+        assert result == "1.2.4"
 
 
 # ---------------------------------------------------------------------- #
-# TaskSpec definitions
+# Edge cases
 # ---------------------------------------------------------------------- #
-class TestTaskSpecDefinitions:
-    """Test that all TaskSpec definitions are valid."""
+class TestEdgeCases:
+    """Test edge cases and error handling."""
 
-    def test_bump_patch_spec(self) -> None:
-        """bump_patch spec should be properly defined."""
-        assert bumpversion.bump_patch.name == "bump_patch"
-        assert bumpversion.bump_patch.fn is not None
+    def test_empty_file(self, tmp_path: Path, capsys) -> None:
+        """Should handle empty file."""
+        test_file = tmp_path / "empty.txt"
+        test_file.write_text("", encoding="utf-8")
 
-    def test_bump_minor_spec(self) -> None:
-        """bump_minor spec should be properly defined."""
-        assert bumpversion.bump_minor.name == "bump_minor"
-        assert bumpversion.bump_minor.fn is not None
+        result = bumpversion.bump_file_version(test_file, "patch")
 
-    def test_bump_major_spec(self) -> None:
-        """bump_major spec should be properly defined."""
-        assert bumpversion.bump_major.name == "bump_major"
-        assert bumpversion.bump_major.fn is not None
+        assert result is None
+        captured = capsys.readouterr()
+        assert "未找到版本号模式" in captured.out
 
+    def test_file_with_special_chars(self, tmp_path: Path) -> None:
+        """Should handle file with special characters."""
+        test_file = tmp_path / "test.txt"
+        content = "# 中文注释\nversion = 1.0.0\n# 特殊字符: @#$%"
+        test_file.write_text(content, encoding="utf-8")
 
-# ---------------------------------------------------------------------- #
-# main function
-# ---------------------------------------------------------------------- #
-class TestMain:
-    """Test main function."""
+        result = bumpversion.bump_file_version(test_file, "patch")
 
-    def test_main_calls_run_cli(self) -> None:
-        """main() should create a CliRunner and call run_cli()."""
-        with patch.object(px.CliRunner, "run_cli") as mock_run_cli:
-            bumpversion.main()
-            assert mock_run_cli.called
+        assert result == "1.0.1"
+        updated = test_file.read_text(encoding="utf-8")
+        assert "# 中文注释" in updated
+        assert "# 特殊字符: @#$%" in updated
+
+    def test_consecutive_bumps(self, tmp_path: Path) -> None:
+        """Should handle consecutive version bumps correctly."""
+        test_file = tmp_path / "test.txt"
+        test_file.write_text("version = 1.0.0", encoding="utf-8")
+
+        # 第一次 bump
+        result1 = bumpversion.bump_file_version(test_file, "patch")
+        assert result1 == "1.0.1"
+
+        # 第二次 bump
+        result2 = bumpversion.bump_file_version(test_file, "minor")
+        assert result2 == "1.1.0"
+
+        # 第三次 bump
+        result3 = bumpversion.bump_file_version(test_file, "major")
+        assert result3 == "2.0.0"
+
+        # 验证最终结果
+        assert test_file.read_text(encoding="utf-8") == "version = 2.0.0"