~tests

tests/cli/test_piptool.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+import subprocess
 from pathlib import Path
 from unittest.mock import MagicMock, patch
 
@@ -9,6 +10,95 @@ import pyflowx as px
 from pyflowx.cli import piptool
 
 
+# ---------------------------------------------------------------------- #
+# _get_installed_packages
+# ---------------------------------------------------------------------- #
+class TestGetInstalledPackages:
+    """Test _get_installed_packages function."""
+
+    def test_get_installed_packages_success(self) -> None:
+        """Should get installed packages."""
+        with patch("subprocess.run") as mock_run:
+            mock_run.return_value = MagicMock(stdout="numpy==1.0.0\npandas==2.0.0\n", returncode=0)
+            result = piptool._get_installed_packages()
+            assert "numpy" in result
+            assert "pandas" in result
+
+    def test_get_installed_packages_empty(self) -> None:
+        """Should handle empty output."""
+        with patch("subprocess.run") as mock_run:
+            mock_run.return_value = MagicMock(stdout="", returncode=0)
+            result = piptool._get_installed_packages()
+            assert result == []
+
+    def test_get_installed_packages_error(self) -> None:
+        """Should handle subprocess error."""
+        with patch("subprocess.run", side_effect=subprocess.SubprocessError):
+            result = piptool._get_installed_packages()
+            assert result == []
+
+    def test_get_installed_packages_oserror(self) -> None:
+        """Should handle OSError."""
+        with patch("subprocess.run", side_effect=OSError):
+            result = piptool._get_installed_packages()
+            assert result == []
+
+
+# ---------------------------------------------------------------------- #
+# _expand_wildcard_packages
+# ---------------------------------------------------------------------- #
+class TestExpandWildcardPackages:
+    """Test _expand_wildcard_packages function."""
+
+    def test_expand_wildcard_no_pattern(self) -> None:
+        """Should return package name when no wildcard."""
+        result = piptool._expand_wildcard_packages("numpy")
+        assert result == ["numpy"]
+
+    def test_expand_wildcard_with_star(self) -> None:
+        """Should expand wildcard with star."""
+        with patch.object(piptool, "_get_installed_packages", return_value=["numpy", "numpy-core", "pandas"]):
+            result = piptool._expand_wildcard_packages("numpy*")
+            assert "numpy" in result
+            assert "numpy-core" in result
+
+    def test_expand_wildcard_with_question(self) -> None:
+        """Should expand wildcard with question mark."""
+        with patch.object(piptool, "_get_installed_packages", return_value=["numpy", "numba"]):
+            result = piptool._expand_wildcard_packages("num??")
+            assert len(result) > 0
+
+    def test_expand_wildcard_no_match(self) -> None:
+        """Should return empty list when no match."""
+        with patch.object(piptool, "_get_installed_packages", return_value=["pandas", "scipy"]):
+            result = piptool._expand_wildcard_packages("numpy*")
+            assert result == []
+
+
+# ---------------------------------------------------------------------- #
+# _filter_protected_packages
+# ---------------------------------------------------------------------- #
+class TestFilterProtectedPackages:
+    """Test _filter_protected_packages function."""
+
+    def test_filter_protected_packages_normal(self) -> None:
+        """Should filter protected packages."""
+        result = piptool._filter_protected_packages(["numpy", "pandas", "pyflowx"])
+        assert "numpy" in result
+        assert "pandas" in result
+        assert "pyflowx" not in result
+
+    def test_filter_protected_packages_all_protected(self) -> None:
+        """Should filter all protected packages."""
+        result = piptool._filter_protected_packages(["pyflowx", "bitool"])
+        assert result == []
+
+    def test_filter_protected_packages_case_insensitive(self) -> None:
+        """Should filter case insensitive."""
+        result = piptool._filter_protected_packages(["PyFlowX", "BITOOL"])
+        assert result == []
+
+
 # ---------------------------------------------------------------------- #
 # pip_uninstall
 # ---------------------------------------------------------------------- #
@@ -39,6 +129,17 @@ class TestPipUninstall:
             piptool.pip_uninstall(["numpy*"])
             assert mock_run.called
 
+    def test_pip_uninstall_empty_packages(self) -> None:
+        """Should handle empty packages list."""
+        with patch.object(piptool, "_expand_wildcard_packages", return_value=[]):
+            piptool.pip_uninstall(["nonexistent*"])
+            # Should not call subprocess.run
+
+    def test_pip_uninstall_all_protected(self) -> None:
+        """Should handle all protected packages."""
+        piptool.pip_uninstall(["pyflowx"])
+        # Should not call subprocess.run
+
 
 # ---------------------------------------------------------------------- #
 # pip_reinstall
@@ -62,6 +163,11 @@ class TestPipReinstall:
             # Should call pip install with offline flags
             assert mock_run.called
 
+    def test_pip_reinstall_all_protected(self) -> None:
+        """Should handle all protected packages."""
+        piptool.pip_reinstall(["pyflowx"])
+        # Should not call subprocess.run
+
 
 # ---------------------------------------------------------------------- #
 # pip_download