Add a test for compressed_listing
@@ -4,7 +4,11 @@ import pytest
from mock import Mock
import pip.req.req_uninstall
-from pip.req.req_uninstall import UninstallPathSet, uninstallation_paths
+from pip.req.req_uninstall import (
+ UninstallPathSet, compact, compress_for_output_listing,
+ uninstallation_paths
+)
+from tests.lib import create_file
# Pretend all files are local, so UninstallPathSet accepts files in the tmpdir,
@@ -40,6 +44,55 @@ def test_uninstallation_paths():
assert paths2 == paths
+def test_compressed_listing(tmpdir):
+ def in_tmpdir(paths):
+ li = []
+ for path in paths:
+ li.append(str(os.path.join(tmpdir, path.replace("/", os.path.sep))))
+ return li
+
+ sample = in_tmpdir([
+ "lib/mypkg.dist-info/METADATA",
+ "lib/mypkg.dist-info/PKG-INFO",
+ "lib/mypkg/would_be_removed.txt",
+ "lib/mypkg/would_be_skipped.skip.txt",
+ "lib/mypkg/__init__.py",
+ "lib/mypkg/my_awesome_code.py",
+ "lib/mypkg/__pycache__/my_awesome_code-magic.pyc",
+ "lib/mypkg/support/support_file.py",
+ "lib/mypkg/support/more_support.py",
+ "lib/mypkg/support/would_be_skipped.skip.py",
+ "lib/mypkg/support/__pycache__/support_file-magic.pyc",
+ "lib/random_other_place/file_without_a_dot_pyc",
+ "bin/mybin",
+ ])
+
+ # Create the required files
+ for fname in sample:
+ create_file(fname, "random blub")
+
+ # Remove the files to be skipped from the paths
+ for path in sample[:]:
+ if ".skip." in path:
+ sample.remove(path)
+
+ expected_remove = in_tmpdir([
+ "bin/mybin",
+ "lib/mypkg.dist-info/*",
+ "lib/mypkg/*",
+ "lib/random_other_place/file_without_a_dot_pyc",
+ ])
+
+ expected_skip = in_tmpdir([
+ "lib/mypkg/would_be_skipped.skip.txt",
+ "lib/mypkg/support/would_be_skipped.skip.py",
+ ])
+
+ will_remove, will_skip = compress_for_output_listing(sample)
+ assert sorted(expected_skip) == sorted(compact(will_skip))
+ assert sorted(expected_remove) == sorted(compact(will_remove))
+
+
class TestUninstallPathSet(object):
def test_add(self, tmpdir, monkeypatch):
monkeypatch.setattr(pip.req.req_uninstall, 'is_local', mock_is_local)
@@ -80,7 +133,7 @@ class TestUninstallPathSet(object):
ups = UninstallPathSet(dist=Mock())
ups.add(short_path)
ups.add(os.path.join(short_path, 'longer'))
- assert ups.compact(ups.paths) == set([short_path])
+ assert compact(ups.paths) == set([short_path])
@pytest.mark.skipif("sys.platform == 'win32'")
def test_detect_symlink_dirs(self, monkeypatch, tmpdir):
