mirror of https://github.com/pypa/pip
271 lines
8.5 KiB
Python
271 lines
8.5 KiB
Python
import os
|
|
|
|
import pytest
|
|
from mock import Mock
|
|
|
|
import pip._internal.req.req_uninstall
|
|
from pip._internal.req.req_uninstall import (
|
|
StashedUninstallPathSet, UninstallPathSet, compact,
|
|
compress_for_output_listing, compress_for_rename, uninstallation_paths,
|
|
)
|
|
from tests.lib import create_file
|
|
|
|
|
|
# Pretend all files are local, so UninstallPathSet accepts files in the tmpdir,
|
|
# outside the virtualenv
|
|
def mock_is_local(path):
|
|
return True
|
|
|
|
|
|
def test_uninstallation_paths():
|
|
class dist(object):
|
|
def get_metadata_lines(self, record):
|
|
return ['file.py,,',
|
|
'file.pyc,,',
|
|
'file.so,,',
|
|
'nopyc.py']
|
|
location = ''
|
|
|
|
d = dist()
|
|
|
|
paths = list(uninstallation_paths(d))
|
|
|
|
expected = ['file.py',
|
|
'file.pyc',
|
|
'file.pyo',
|
|
'file.so',
|
|
'nopyc.py',
|
|
'nopyc.pyc',
|
|
'nopyc.pyo']
|
|
|
|
assert paths == expected
|
|
|
|
# Avoid an easy 'unique generator' bug
|
|
paths2 = list(uninstallation_paths(d))
|
|
|
|
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
|
|
sample = [path for path in sample if ".skip." not in 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",
|
|
])
|
|
|
|
expected_rename = in_tmpdir([
|
|
"bin/",
|
|
"lib/mypkg.dist-info/",
|
|
"lib/mypkg/would_be_removed.txt",
|
|
"lib/mypkg/__init__.py",
|
|
"lib/mypkg/my_awesome_code.py",
|
|
"lib/mypkg/__pycache__/",
|
|
"lib/mypkg/support/support_file.py",
|
|
"lib/mypkg/support/more_support.py",
|
|
"lib/mypkg/support/__pycache__/",
|
|
"lib/random_other_place/",
|
|
])
|
|
|
|
will_remove, will_skip = compress_for_output_listing(sample)
|
|
will_rename = compress_for_rename(sample)
|
|
assert sorted(expected_skip) == sorted(compact(will_skip))
|
|
assert sorted(expected_remove) == sorted(compact(will_remove))
|
|
assert sorted(expected_rename) == sorted(compact(will_rename))
|
|
|
|
|
|
class TestUninstallPathSet(object):
|
|
def test_add(self, tmpdir, monkeypatch):
|
|
monkeypatch.setattr(pip._internal.req.req_uninstall, 'is_local',
|
|
mock_is_local)
|
|
# Fix case for windows tests
|
|
file_extant = os.path.normcase(os.path.join(tmpdir, 'foo'))
|
|
file_nonexistent = os.path.normcase(
|
|
os.path.join(tmpdir, 'nonexistent'))
|
|
with open(file_extant, 'w'):
|
|
pass
|
|
|
|
ups = UninstallPathSet(dist=Mock())
|
|
assert ups.paths == set()
|
|
ups.add(file_extant)
|
|
assert ups.paths == {file_extant}
|
|
|
|
ups.add(file_nonexistent)
|
|
assert ups.paths == {file_extant}
|
|
|
|
@pytest.mark.skipif("sys.platform == 'win32'")
|
|
def test_add_symlink(self, tmpdir, monkeypatch):
|
|
monkeypatch.setattr(pip._internal.req.req_uninstall, 'is_local',
|
|
mock_is_local)
|
|
f = os.path.join(tmpdir, 'foo')
|
|
with open(f, 'w'):
|
|
pass
|
|
foo_link = os.path.join(tmpdir, 'foo_link')
|
|
os.symlink(f, foo_link)
|
|
|
|
ups = UninstallPathSet(dist=Mock())
|
|
ups.add(foo_link)
|
|
assert ups.paths == {foo_link}
|
|
|
|
def test_compact_shorter_path(self, monkeypatch):
|
|
monkeypatch.setattr(pip._internal.req.req_uninstall, 'is_local',
|
|
lambda p: True)
|
|
monkeypatch.setattr('os.path.exists', lambda p: True)
|
|
# This deals with nt/posix path differences
|
|
short_path = os.path.normcase(os.path.abspath(
|
|
os.path.join(os.path.sep, 'path')))
|
|
ups = UninstallPathSet(dist=Mock())
|
|
ups.add(short_path)
|
|
ups.add(os.path.join(short_path, 'longer'))
|
|
assert compact(ups.paths) == {short_path}
|
|
|
|
@pytest.mark.skipif("sys.platform == 'win32'")
|
|
def test_detect_symlink_dirs(self, monkeypatch, tmpdir):
|
|
monkeypatch.setattr(pip._internal.req.req_uninstall, 'is_local',
|
|
lambda p: True)
|
|
|
|
# construct 2 paths:
|
|
# tmpdir/dir/file
|
|
# tmpdir/dirlink/file (where dirlink is a link to dir)
|
|
d = tmpdir.joinpath('dir')
|
|
d.mkdir()
|
|
dlink = tmpdir.joinpath('dirlink')
|
|
os.symlink(d, dlink)
|
|
d.joinpath('file').touch()
|
|
path1 = str(d.joinpath('file'))
|
|
path2 = str(dlink.joinpath('file'))
|
|
|
|
ups = UninstallPathSet(dist=Mock())
|
|
ups.add(path1)
|
|
ups.add(path2)
|
|
assert ups.paths == {path1}
|
|
|
|
|
|
class TestStashedUninstallPathSet(object):
|
|
WALK_RESULT = [
|
|
("A", ["B", "C"], ["a.py"]),
|
|
("A/B", ["D"], ["b.py"]),
|
|
("A/B/D", [], ["c.py"]),
|
|
("A/C", [], ["d.py", "e.py"]),
|
|
("A/E", ["F"], ["f.py"]),
|
|
("A/E/F", [], []),
|
|
("A/G", ["H"], ["g.py"]),
|
|
("A/G/H", [], ["h.py"]),
|
|
]
|
|
|
|
@classmethod
|
|
def mock_walk(cls, root):
|
|
for dirname, subdirs, files in cls.WALK_RESULT:
|
|
dirname = os.path.sep.join(dirname.split("/"))
|
|
if dirname.startswith(root):
|
|
yield dirname[len(root) + 1:], subdirs, files
|
|
|
|
def test_compress_for_rename(self, monkeypatch):
|
|
paths = [os.path.sep.join(p.split("/")) for p in [
|
|
"A/B/b.py",
|
|
"A/B/D/c.py",
|
|
"A/C/d.py",
|
|
"A/E/f.py",
|
|
"A/G/g.py",
|
|
]]
|
|
|
|
expected_paths = [os.path.sep.join(p.split("/")) for p in [
|
|
"A/B/", # selected everything below A/B
|
|
"A/C/d.py", # did not select everything below A/C
|
|
"A/E/", # only empty folders remain under A/E
|
|
"A/G/g.py", # non-empty folder remains under A/G
|
|
]]
|
|
|
|
monkeypatch.setattr('os.walk', self.mock_walk)
|
|
|
|
actual_paths = compress_for_rename(paths)
|
|
assert set(expected_paths) == set(actual_paths)
|
|
|
|
@classmethod
|
|
def make_stash(cls, tmpdir, paths):
|
|
for dirname, subdirs, files in cls.WALK_RESULT:
|
|
root = os.path.join(tmpdir, *dirname.split("/"))
|
|
if not os.path.exists(root):
|
|
os.mkdir(root)
|
|
for d in subdirs:
|
|
os.mkdir(os.path.join(root, d))
|
|
for f in files:
|
|
with open(os.path.join(root, f), "wb"):
|
|
pass
|
|
|
|
pathset = StashedUninstallPathSet()
|
|
|
|
paths = [os.path.join(tmpdir, *p.split('/')) for p in paths]
|
|
stashed_paths = [(p, pathset.stash(p)) for p in paths]
|
|
|
|
return pathset, stashed_paths
|
|
|
|
def test_stash(self, tmpdir):
|
|
pathset, stashed_paths = self.make_stash(tmpdir, [
|
|
"A/B/", "A/C/d.py", "A/E/", "A/G/g.py",
|
|
])
|
|
|
|
for old_path, new_path in stashed_paths:
|
|
assert not os.path.exists(old_path)
|
|
assert os.path.exists(new_path)
|
|
|
|
assert stashed_paths == pathset._moves
|
|
|
|
def test_commit(self, tmpdir):
|
|
pathset, stashed_paths = self.make_stash(tmpdir, [
|
|
"A/B/", "A/C/d.py", "A/E/", "A/G/g.py",
|
|
])
|
|
|
|
pathset.commit()
|
|
|
|
for old_path, new_path in stashed_paths:
|
|
assert not os.path.exists(old_path)
|
|
assert not os.path.exists(new_path)
|
|
|
|
def test_rollback(self, tmpdir):
|
|
pathset, stashed_paths = self.make_stash(tmpdir, [
|
|
"A/B/", "A/C/d.py", "A/E/", "A/G/g.py",
|
|
])
|
|
|
|
pathset.rollback()
|
|
|
|
for old_path, new_path in stashed_paths:
|
|
assert os.path.exists(old_path)
|
|
assert not os.path.exists(new_path)
|