|
|
import functools |
|
|
import inspect |
|
|
import re |
|
|
import textwrap |
|
|
|
|
|
import pytest |
|
|
|
|
|
import pkg_resources |
|
|
|
|
|
from .test_resources import Metadata |
|
|
|
|
|
|
|
|
def strip_comments(s): |
|
|
return '\n'.join( |
|
|
line |
|
|
for line in s.split('\n') |
|
|
if line.strip() and not line.strip().startswith('#') |
|
|
) |
|
|
|
|
|
|
|
|
def parse_distributions(s): |
|
|
""" |
|
|
Parse a series of distribution specs of the form: |
|
|
{project_name}-{version} |
|
|
[optional, indented requirements specification] |
|
|
|
|
|
Example: |
|
|
|
|
|
foo-0.2 |
|
|
bar-1.0 |
|
|
foo>=3.0 |
|
|
[feature] |
|
|
baz |
|
|
|
|
|
yield 2 distributions: |
|
|
- project_name=foo, version=0.2 |
|
|
- project_name=bar, version=1.0, |
|
|
requires=['foo>=3.0', 'baz; extra=="feature"'] |
|
|
""" |
|
|
s = s.strip() |
|
|
for spec in re.split(r'\n(?=[^\s])', s): |
|
|
if not spec: |
|
|
continue |
|
|
fields = spec.split('\n', 1) |
|
|
assert 1 <= len(fields) <= 2 |
|
|
name, version = fields.pop(0).rsplit('-', 1) |
|
|
if fields: |
|
|
requires = textwrap.dedent(fields.pop(0)) |
|
|
metadata = Metadata(('requires.txt', requires)) |
|
|
else: |
|
|
metadata = None |
|
|
dist = pkg_resources.Distribution( |
|
|
project_name=name, version=version, metadata=metadata |
|
|
) |
|
|
yield dist |
|
|
|
|
|
|
|
|
class FakeInstaller: |
|
|
def __init__(self, installable_dists) -> None: |
|
|
self._installable_dists = installable_dists |
|
|
|
|
|
def __call__(self, req): |
|
|
return next( |
|
|
iter(filter(lambda dist: dist in req, self._installable_dists)), None |
|
|
) |
|
|
|
|
|
|
|
|
def parametrize_test_working_set_resolve(*test_list): |
|
|
idlist = [] |
|
|
argvalues = [] |
|
|
for test in test_list: |
|
|
( |
|
|
name, |
|
|
installed_dists, |
|
|
installable_dists, |
|
|
requirements, |
|
|
expected1, |
|
|
expected2, |
|
|
) = ( |
|
|
strip_comments(s.lstrip()) |
|
|
for s in textwrap.dedent(test).lstrip().split('\n\n', 5) |
|
|
) |
|
|
installed_dists = list(parse_distributions(installed_dists)) |
|
|
installable_dists = list(parse_distributions(installable_dists)) |
|
|
requirements = list(pkg_resources.parse_requirements(requirements)) |
|
|
for id_, replace_conflicting, expected in ( |
|
|
(name, False, expected1), |
|
|
(name + '_replace_conflicting', True, expected2), |
|
|
): |
|
|
idlist.append(id_) |
|
|
expected = strip_comments(expected.strip()) |
|
|
if re.match(r'\w+$', expected): |
|
|
expected = getattr(pkg_resources, expected) |
|
|
assert issubclass(expected, Exception) |
|
|
else: |
|
|
expected = list(parse_distributions(expected)) |
|
|
argvalues.append( |
|
|
pytest.param( |
|
|
installed_dists, |
|
|
installable_dists, |
|
|
requirements, |
|
|
replace_conflicting, |
|
|
expected, |
|
|
) |
|
|
) |
|
|
return pytest.mark.parametrize( |
|
|
'installed_dists,installable_dists,' |
|
|
'requirements,replace_conflicting,' |
|
|
'resolved_dists_or_exception', |
|
|
argvalues, |
|
|
ids=idlist, |
|
|
) |
|
|
|
|
|
|
|
|
@parametrize_test_working_set_resolve( |
|
|
""" |
|
|
# id |
|
|
noop |
|
|
|
|
|
# installed |
|
|
|
|
|
# installable |
|
|
|
|
|
# wanted |
|
|
|
|
|
# resolved |
|
|
|
|
|
# resolved [replace conflicting] |
|
|
""", |
|
|
""" |
|
|
# id |
|
|
already_installed |
|
|
|
|
|
# installed |
|
|
foo-3.0 |
|
|
|
|
|
# installable |
|
|
|
|
|
# wanted |
|
|
foo>=2.1,!=3.1,<4 |
|
|
|
|
|
# resolved |
|
|
foo-3.0 |
|
|
|
|
|
# resolved [replace conflicting] |
|
|
foo-3.0 |
|
|
""", |
|
|
""" |
|
|
# id |
|
|
installable_not_installed |
|
|
|
|
|
# installed |
|
|
|
|
|
# installable |
|
|
foo-3.0 |
|
|
foo-4.0 |
|
|
|
|
|
# wanted |
|
|
foo>=2.1,!=3.1,<4 |
|
|
|
|
|
# resolved |
|
|
foo-3.0 |
|
|
|
|
|
# resolved [replace conflicting] |
|
|
foo-3.0 |
|
|
""", |
|
|
""" |
|
|
# id |
|
|
not_installable |
|
|
|
|
|
# installed |
|
|
|
|
|
# installable |
|
|
|
|
|
# wanted |
|
|
foo>=2.1,!=3.1,<4 |
|
|
|
|
|
# resolved |
|
|
DistributionNotFound |
|
|
|
|
|
# resolved [replace conflicting] |
|
|
DistributionNotFound |
|
|
""", |
|
|
""" |
|
|
# id |
|
|
no_matching_version |
|
|
|
|
|
# installed |
|
|
|
|
|
# installable |
|
|
foo-3.1 |
|
|
|
|
|
# wanted |
|
|
foo>=2.1,!=3.1,<4 |
|
|
|
|
|
# resolved |
|
|
DistributionNotFound |
|
|
|
|
|
# resolved [replace conflicting] |
|
|
DistributionNotFound |
|
|
""", |
|
|
""" |
|
|
# id |
|
|
installable_with_installed_conflict |
|
|
|
|
|
# installed |
|
|
foo-3.1 |
|
|
|
|
|
# installable |
|
|
foo-3.5 |
|
|
|
|
|
# wanted |
|
|
foo>=2.1,!=3.1,<4 |
|
|
|
|
|
# resolved |
|
|
VersionConflict |
|
|
|
|
|
# resolved [replace conflicting] |
|
|
foo-3.5 |
|
|
""", |
|
|
""" |
|
|
# id |
|
|
not_installable_with_installed_conflict |
|
|
|
|
|
# installed |
|
|
foo-3.1 |
|
|
|
|
|
# installable |
|
|
|
|
|
# wanted |
|
|
foo>=2.1,!=3.1,<4 |
|
|
|
|
|
# resolved |
|
|
VersionConflict |
|
|
|
|
|
# resolved [replace conflicting] |
|
|
DistributionNotFound |
|
|
""", |
|
|
""" |
|
|
# id |
|
|
installed_with_installed_require |
|
|
|
|
|
# installed |
|
|
foo-3.9 |
|
|
baz-0.1 |
|
|
foo>=2.1,!=3.1,<4 |
|
|
|
|
|
# installable |
|
|
|
|
|
# wanted |
|
|
baz |
|
|
|
|
|
# resolved |
|
|
foo-3.9 |
|
|
baz-0.1 |
|
|
|
|
|
# resolved [replace conflicting] |
|
|
foo-3.9 |
|
|
baz-0.1 |
|
|
""", |
|
|
""" |
|
|
# id |
|
|
installed_with_conflicting_installed_require |
|
|
|
|
|
# installed |
|
|
foo-5 |
|
|
baz-0.1 |
|
|
foo>=2.1,!=3.1,<4 |
|
|
|
|
|
# installable |
|
|
|
|
|
# wanted |
|
|
baz |
|
|
|
|
|
# resolved |
|
|
VersionConflict |
|
|
|
|
|
# resolved [replace conflicting] |
|
|
DistributionNotFound |
|
|
""", |
|
|
""" |
|
|
# id |
|
|
installed_with_installable_conflicting_require |
|
|
|
|
|
# installed |
|
|
foo-5 |
|
|
baz-0.1 |
|
|
foo>=2.1,!=3.1,<4 |
|
|
|
|
|
# installable |
|
|
foo-2.9 |
|
|
|
|
|
# wanted |
|
|
baz |
|
|
|
|
|
# resolved |
|
|
VersionConflict |
|
|
|
|
|
# resolved [replace conflicting] |
|
|
baz-0.1 |
|
|
foo-2.9 |
|
|
""", |
|
|
""" |
|
|
# id |
|
|
installed_with_installable_require |
|
|
|
|
|
# installed |
|
|
baz-0.1 |
|
|
foo>=2.1,!=3.1,<4 |
|
|
|
|
|
# installable |
|
|
foo-3.9 |
|
|
|
|
|
# wanted |
|
|
baz |
|
|
|
|
|
# resolved |
|
|
foo-3.9 |
|
|
baz-0.1 |
|
|
|
|
|
# resolved [replace conflicting] |
|
|
foo-3.9 |
|
|
baz-0.1 |
|
|
""", |
|
|
""" |
|
|
# id |
|
|
installable_with_installed_require |
|
|
|
|
|
# installed |
|
|
foo-3.9 |
|
|
|
|
|
# installable |
|
|
baz-0.1 |
|
|
foo>=2.1,!=3.1,<4 |
|
|
|
|
|
# wanted |
|
|
baz |
|
|
|
|
|
# resolved |
|
|
foo-3.9 |
|
|
baz-0.1 |
|
|
|
|
|
# resolved [replace conflicting] |
|
|
foo-3.9 |
|
|
baz-0.1 |
|
|
""", |
|
|
""" |
|
|
# id |
|
|
installable_with_installable_require |
|
|
|
|
|
# installed |
|
|
|
|
|
# installable |
|
|
foo-3.9 |
|
|
baz-0.1 |
|
|
foo>=2.1,!=3.1,<4 |
|
|
|
|
|
# wanted |
|
|
baz |
|
|
|
|
|
# resolved |
|
|
foo-3.9 |
|
|
baz-0.1 |
|
|
|
|
|
# resolved [replace conflicting] |
|
|
foo-3.9 |
|
|
baz-0.1 |
|
|
""", |
|
|
""" |
|
|
# id |
|
|
installable_with_conflicting_installable_require |
|
|
|
|
|
# installed |
|
|
foo-5 |
|
|
|
|
|
# installable |
|
|
foo-2.9 |
|
|
baz-0.1 |
|
|
foo>=2.1,!=3.1,<4 |
|
|
|
|
|
# wanted |
|
|
baz |
|
|
|
|
|
# resolved |
|
|
VersionConflict |
|
|
|
|
|
# resolved [replace conflicting] |
|
|
baz-0.1 |
|
|
foo-2.9 |
|
|
""", |
|
|
""" |
|
|
# id |
|
|
conflicting_installables |
|
|
|
|
|
# installed |
|
|
|
|
|
# installable |
|
|
foo-2.9 |
|
|
foo-5.0 |
|
|
|
|
|
# wanted |
|
|
foo>=2.1,!=3.1,<4 |
|
|
foo>=4 |
|
|
|
|
|
# resolved |
|
|
VersionConflict |
|
|
|
|
|
# resolved [replace conflicting] |
|
|
VersionConflict |
|
|
""", |
|
|
""" |
|
|
# id |
|
|
installables_with_conflicting_requires |
|
|
|
|
|
# installed |
|
|
|
|
|
# installable |
|
|
foo-2.9 |
|
|
dep==1.0 |
|
|
baz-5.0 |
|
|
dep==2.0 |
|
|
dep-1.0 |
|
|
dep-2.0 |
|
|
|
|
|
# wanted |
|
|
foo |
|
|
baz |
|
|
|
|
|
# resolved |
|
|
VersionConflict |
|
|
|
|
|
# resolved [replace conflicting] |
|
|
VersionConflict |
|
|
""", |
|
|
""" |
|
|
# id |
|
|
installables_with_conflicting_nested_requires |
|
|
|
|
|
# installed |
|
|
|
|
|
# installable |
|
|
foo-2.9 |
|
|
dep1 |
|
|
dep1-1.0 |
|
|
subdep<1.0 |
|
|
baz-5.0 |
|
|
dep2 |
|
|
dep2-1.0 |
|
|
subdep>1.0 |
|
|
subdep-0.9 |
|
|
subdep-1.1 |
|
|
|
|
|
# wanted |
|
|
foo |
|
|
baz |
|
|
|
|
|
# resolved |
|
|
VersionConflict |
|
|
|
|
|
# resolved [replace conflicting] |
|
|
VersionConflict |
|
|
""", |
|
|
""" |
|
|
# id |
|
|
wanted_normalized_name_installed_canonical |
|
|
|
|
|
# installed |
|
|
foo.bar-3.6 |
|
|
|
|
|
# installable |
|
|
|
|
|
# wanted |
|
|
foo-bar==3.6 |
|
|
|
|
|
# resolved |
|
|
foo.bar-3.6 |
|
|
|
|
|
# resolved [replace conflicting] |
|
|
foo.bar-3.6 |
|
|
""", |
|
|
) |
|
|
def test_working_set_resolve( |
|
|
installed_dists, |
|
|
installable_dists, |
|
|
requirements, |
|
|
replace_conflicting, |
|
|
resolved_dists_or_exception, |
|
|
): |
|
|
ws = pkg_resources.WorkingSet([]) |
|
|
list(map(ws.add, installed_dists)) |
|
|
resolve_call = functools.partial( |
|
|
ws.resolve, |
|
|
requirements, |
|
|
installer=FakeInstaller(installable_dists), |
|
|
replace_conflicting=replace_conflicting, |
|
|
) |
|
|
if inspect.isclass(resolved_dists_or_exception): |
|
|
with pytest.raises(resolved_dists_or_exception): |
|
|
resolve_call() |
|
|
else: |
|
|
assert sorted(resolve_call()) == sorted(resolved_dists_or_exception) |
|
|
|
