255 lines
10 KiB
Python
255 lines
10 KiB
Python
"""Tests for cli.piptool module."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import subprocess
|
|
from pathlib import Path
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
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
|
|
# ---------------------------------------------------------------------- #
|
|
class TestPipUninstall:
|
|
"""Test pip_uninstall function."""
|
|
|
|
def test_pip_uninstall_single_package(self) -> None:
|
|
"""Should uninstall single package."""
|
|
with patch("subprocess.run") as mock_run:
|
|
mock_run.return_value = MagicMock(returncode=0)
|
|
piptool.pip_uninstall(["numpy"])
|
|
assert mock_run.called
|
|
|
|
def test_pip_uninstall_multiple_packages(self) -> None:
|
|
"""Should uninstall multiple packages."""
|
|
with patch("subprocess.run") as mock_run:
|
|
mock_run.return_value = MagicMock(returncode=0)
|
|
piptool.pip_uninstall(["numpy", "pandas", "scipy"])
|
|
# Should call pip uninstall
|
|
assert mock_run.called
|
|
|
|
def test_pip_uninstall_with_wildcard(self) -> None:
|
|
"""Should handle wildcard in package name."""
|
|
with patch.object(piptool, "_expand_wildcard_packages", return_value=["numpy", "numpy-core"]), patch(
|
|
"subprocess.run"
|
|
) as mock_run:
|
|
mock_run.return_value = MagicMock(returncode=0)
|
|
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
|
|
# ---------------------------------------------------------------------- #
|
|
class TestPipReinstall:
|
|
"""Test pip_reinstall function."""
|
|
|
|
def test_pip_reinstall_single_package(self) -> None:
|
|
"""Should reinstall single package."""
|
|
with patch("subprocess.run") as mock_run:
|
|
mock_run.return_value = MagicMock(returncode=0)
|
|
piptool.pip_reinstall(["numpy"])
|
|
# Should call pip uninstall and pip install
|
|
assert mock_run.call_count == 2
|
|
|
|
def test_pip_reinstall_offline(self) -> None:
|
|
"""Should reinstall packages offline."""
|
|
with patch("subprocess.run") as mock_run:
|
|
mock_run.return_value = MagicMock(returncode=0)
|
|
piptool.pip_reinstall(["numpy"], offline=True)
|
|
# 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
|
|
# ---------------------------------------------------------------------- #
|
|
class TestPipDownload:
|
|
"""Test pip_download function."""
|
|
|
|
def test_pip_download_single_package(self) -> None:
|
|
"""Should download single package."""
|
|
with patch("subprocess.run") as mock_run:
|
|
mock_run.return_value = MagicMock(returncode=0)
|
|
piptool.pip_download(["numpy"])
|
|
assert mock_run.called
|
|
|
|
def test_pip_download_offline(self) -> None:
|
|
"""Should download packages offline."""
|
|
with patch("subprocess.run") as mock_run:
|
|
mock_run.return_value = MagicMock(returncode=0)
|
|
piptool.pip_download(["numpy"], offline=True)
|
|
# Should call pip download with offline flags
|
|
assert mock_run.called
|
|
|
|
|
|
# ---------------------------------------------------------------------- #
|
|
# pip_freeze
|
|
# ---------------------------------------------------------------------- #
|
|
class TestPipFreeze:
|
|
"""Test pip_freeze function."""
|
|
|
|
def test_pip_freeze(self, tmp_path: Path) -> None:
|
|
"""Should freeze dependencies."""
|
|
with patch("subprocess.run") as mock_run:
|
|
mock_run.return_value = MagicMock(stdout="numpy==1.0.0\npandas==2.0.0", returncode=0)
|
|
piptool.pip_freeze()
|
|
assert mock_run.called
|
|
|
|
|
|
# ---------------------------------------------------------------------- #
|
|
# main function
|
|
# ---------------------------------------------------------------------- #
|
|
class TestMain:
|
|
"""Test main function."""
|
|
|
|
def test_main_install_command(self) -> None:
|
|
"""main() should handle install command."""
|
|
with patch("sys.argv", ["piptool", "i", "numpy", "pandas"]), patch.object(px, "run") as mock_run:
|
|
piptool.main()
|
|
assert mock_run.called
|
|
|
|
def test_main_uninstall_command(self) -> None:
|
|
"""main() should handle uninstall command."""
|
|
with patch("sys.argv", ["piptool", "u", "numpy"]), patch.object(px, "run") as mock_run:
|
|
piptool.main()
|
|
assert mock_run.called
|
|
|
|
def test_main_reinstall_command(self) -> None:
|
|
"""main() should handle reinstall command."""
|
|
with patch("sys.argv", ["piptool", "r", "numpy"]), patch.object(px, "run") as mock_run:
|
|
piptool.main()
|
|
assert mock_run.called
|
|
|
|
def test_main_download_command(self) -> None:
|
|
"""main() should handle download command."""
|
|
with patch("sys.argv", ["piptool", "d", "numpy"]), patch.object(px, "run") as mock_run:
|
|
piptool.main()
|
|
assert mock_run.called
|
|
|
|
def test_main_upgrade_command(self) -> None:
|
|
"""main() should handle upgrade command."""
|
|
with patch("sys.argv", ["piptool", "up"]), patch.object(px, "run") as mock_run:
|
|
piptool.main()
|
|
assert mock_run.called
|
|
|
|
def test_main_freeze_command(self) -> None:
|
|
"""main() should handle freeze command."""
|
|
with patch("sys.argv", ["piptool", "f"]), patch.object(px, "run") as mock_run:
|
|
piptool.main()
|
|
assert mock_run.called
|
|
|
|
def test_main_with_no_args_shows_help(self) -> None:
|
|
"""main() with no args should show help."""
|
|
with patch("sys.argv", ["piptool"]):
|
|
piptool.main()
|
|
# Should print help and return
|