def test_globbing(self):
repo = SimpleTree({
'spork': {'foon': ('1',)},
'spork2': {'foon': ('2',)}})
installed_repos = SimpleTree({'foo': {'bar': ('1',)}})
a = pmerge.parse_target(parse_match('*/foon'), repo, installed_repos)
assert len(a) == 2
def test_no_matches(self):
repo = SimpleTree({
'spork': {'foon': ('1',)},
'spork2': {'foon': ('2',)}})
installed_repos = SimpleTree({'foo': {'bar': ('1',)}})
with pytest.raises(pmerge.NoMatches):
pmerge.parse_target(parse_match("foo"), repo, installed_repos)
def test_ambiguous(self):
repo = SimpleTree({
'spork': {'foon': ('1',)},
'spork2': {'foon': ('2',)}})
installed_repos = SimpleTree({'foo': {'bar': ('1',)}})
with pytest.raises(pmerge.AmbiguousQuery):
pmerge.parse_target(parse_match("foon"), repo, installed_repos)
def setUp(self):
# we use an OrderedDict here specifically to trigger any sorter
# related bugs
d = {
"dev-util": {"diffball": ["1.0", "0.7"], "bsdiff": ["0.4.1", "0.4.2"]},
"dev-lib": {"fake": ["1.0", "1.0-r1"]}}
self.repo = SimpleTree(
OrderedDict((k, d[k]) for k in sorted(d, reverse=True)))
def test_collision_livefs(self):
# test pkg name collision between real and virtual pkgs on livefs
# repos, the real pkg will be selected over the virtual
installed_repos = SimpleTree({'foo': {'bar': ('1',)}, 'virtual': {'bar': ('0',)}})
repo = SimpleTree({'foo': {'bar': ('1',)}, 'virtual': {'bar': ('1',)}})
a = pmerge.parse_target(parse_match("bar"), repo, installed_repos)
assert len(a) == 1
assert a[0].key == 'foo/bar'
assert [x.key for x in repo.match(a[0])] == ['foo/bar']
def test_collision_livefs(self):
# test pkg name collision between real and virtual pkgs on livefs
# repos, the real pkg will be selected over the virtual
installed_repos = SimpleTree({'foo': {'bar': ('1',)}, 'virtual': {'bar': ('0',)}})
repo = SimpleTree({'foo': {'bar': ('1',)}, 'virtual': {'bar': ('1',)}})
a = pmerge.parse_target(parse_match("bar"), repo, installed_repos)
assert len(a) == 1
assert a[0].key == 'foo/bar'
assert [x.key for x in repo.match(a[0])] == ['foo/bar']
def test_base_targets(self):
repo = SimpleTree({'spork': {'foon': ('1', '1.0.1', '2')}})
installed_repos = SimpleTree({'foo': {'bar': ('1',)}})
for cat in ('', 'spork/'):
a = pmerge.parse_target(parse_match(f'={cat}foon-1'), repo, installed_repos)
assert len(a) == 1
assert a[0].key == 'spork/foon'
assert [x.fullver for x in repo.itermatch(a[0])] == ['1']
a = pmerge.parse_target(parse_match(f'{cat}foon'), repo, installed_repos)
assert len(a) == 1
assert a[0].key == 'spork/foon'
assert (
sorted(x.fullver for x in repo.itermatch(a[0])) ==
sorted(['1', '1.0.1', '2'])
)
def test_base_targets(self):
repo = SimpleTree({'spork': {'foon': ('1', '1.0.1', '2')}})
installed_repos = SimpleTree({'foo': {'bar': ('1',)}})
for cat in ('', 'spork/'):
a = pmerge.parse_target(parse_match(f'={cat}foon-1'), repo, installed_repos)
assert len(a) == 1
assert a[0].key == 'spork/foon'
assert [x.fullver for x in repo.itermatch(a[0])] == ['1']
a = pmerge.parse_target(parse_match(f'{cat}foon'), repo, installed_repos)
assert len(a) == 1
assert a[0].key == 'spork/foon'
assert (
sorted(x.fullver for x in repo.itermatch(a[0])) ==
sorted(['1', '1.0.1', '2'])
)
def setUp(self):
self.d1, self.d2 = {}, {}
for key, ver in self.tree1_pkgs:
cat, pkg = key.rsplit("/", 1)
self.d1.setdefault(cat, {}).setdefault(pkg, []).extend(ver)
for key, ver in self.tree2_pkgs:
cat, pkg = key.rsplit("/", 1)
self.d2.setdefault(cat, {}).setdefault(pkg, []).extend(ver)
self.d1 = OrderedDict((k, OrderedDict(self.d1[k].items()))
for k in sorted(self.d1, reverse=True))
self.d2 = OrderedDict((k, OrderedDict(self.d2[k].items()))
for k in sorted(self.d2, reverse=True))
self.tree1 = SimpleTree(self.d1)
self.tree2 = SimpleTree(self.d2)
self.ctree = self.kls(self.tree1, self.tree2)
def __init__(self, masked_use={}, forced_use={},
provides={}, masks=[], virtuals={}, arch='x86', name='none'):
self.provides_repo = SimpleTree(provides)
self.masked_use = {atom(k): v for k,v in masked_use.iteritems()}
self.forced_use = {atom(k): v for k,v in forced_use.iteritems()}
self.masks = tuple(map(atom, masks))
self.virtuals = SimpleTree(virtuals)
self.arch = arch
self.name = name
self.forced_data = collapsed_restrict_to_data(
[(AlwaysTrue, (self.arch,))],
self.forced_use.iteritems())
self.masked_data = collapsed_restrict_to_data(
[(AlwaysTrue, default_arches)],
self.masked_use.iteritems())
def setUp(self):
# we use an OrderedDict here specifically to trigger any sorter
# related bugs
d = {
"dev-util": {"diffball": ["1.0", "0.7"], "bsdiff": ["0.4.1", "0.4.2"]},
"dev-lib": {"fake": ["1.0", "1.0-r1"]}}
self.repo = SimpleTree(
OrderedDict((k, d[k]) for k in sorted(d, reverse=True)))
def setup_repos(self, restrictions=None):
repo = SimpleTree({
"dev-util": {"diffball": ["1.0", "0.7"], "bsdiff": ["0.4.1", "0.4.2"]},
"dev-lib": {"fake": ["1.0", "1.0-r1"]}})
if restrictions is None:
restrictions = atom("dev-util/diffball")
vrepo = filtered.tree(repo, restrictions)
return repo, vrepo
def _tmp_validate_args(parser, namespace):
tmpdir = namespace.domain.pm_tmpdir
dirs = ()
files = ()
if namespace.restrict and not namespace.wipe_all:
# create a fake repo from tmpdir entries and pull matches from it
pkg_map = {}
for pkg_build_dir in glob.glob(pjoin(tmpdir, '*', '*')):
try:
pkg = atom_mod.atom(
'=' + pkg_build_dir[len(tmpdir):].lstrip(os.path.sep))
except atom_mod.MalformedAtom:
continue
pkg_map.setdefault(pkg.category,
{}).setdefault(pkg.package,
[]).append(pkg.fullver)
repo = SimpleTree(pkg_map)
def _remove_dir_and_empty_parent(d):
"""Remove a given directory tree and its parent directory, if empty."""
shutil.rmtree(d)
try:
os.rmdir(os.path.dirname(d))
except OSError as e:
# POSIX specifies either ENOTEMPTY or EEXIST for non-empty dir
# in particular, Solaris uses EEXIST in that case.
if e.errno not in (errno.ENOTEMPTY, errno.EEXIST):
raise
removal_func = partial(_remove_dir_and_empty_parent)
dirs = ((removal_func, pjoin(tmpdir, pkg.cpvstr))
for pkg in repo.itermatch(namespace.restrict))
else:
# not in a configured repo dir, remove all tmpdir entries
dir_removal_func = partial(shutil.rmtree)
dirs = ((dir_removal_func, pjoin(tmpdir, d))
for d in listdir_dirs(tmpdir))
file_removal_func = partial(os.remove)
files = ((file_removal_func, pjoin(tmpdir, f))
for f in listdir_files(tmpdir))
namespace.remove = chain(dirs, files)
def __init__(self,
masked_use={},
stable_masked_use={},
forced_use={},
stable_forced_use={},
pkg_use={},
provides={},
iuse_effective=[],
use=[],
masks=[],
unmasks=[],
arch='x86',
name='none'):
self.provides_repo = SimpleTree(provides)
self.masked_use = ChunkedDataDict()
self.masked_use.update_from_stream(
chunked_data(atom(k), *split_negations(v))
for k, v in masked_use.items())
self.masked_use.freeze()
self.stable_masked_use = ChunkedDataDict()
self.stable_masked_use.update_from_stream(
chunked_data(atom(k), *split_negations(v))
for k, v in stable_masked_use.items())
self.stable_masked_use.freeze()
self.forced_use = ChunkedDataDict()
self.forced_use.update_from_stream(
chunked_data(atom(k), *split_negations(v))
for k, v in forced_use.items())
self.forced_use.freeze()
self.stable_forced_use = ChunkedDataDict()
self.stable_forced_use.update_from_stream(
chunked_data(atom(k), *split_negations(v))
for k, v in stable_forced_use.items())
self.stable_forced_use.freeze()
self.pkg_use = ChunkedDataDict()
self.pkg_use.update_from_stream(
chunked_data(atom(k), *split_negations(v))
for k, v in pkg_use.items())
self.pkg_use.freeze()
self.masks = tuple(map(atom, masks))
self.unmasks = tuple(map(atom, unmasks))
self.iuse_effective = set(iuse_effective)
self.use = set(use)
self.key = arch
self.name = name
vfilter = domain.generate_filter(self.masks, self.unmasks)
self.visible = vfilter.match
def test_iter(self):
fake_vdb = SimpleTree({"dev-util": {"diffball":["1.0"],
"bsdiff":["1.2", "1.3"]}}, pkg_klass=FakePkg)
ipkgset = installed.Installed([fake_vdb])
self.assertEqual(sorted(["dev-util/diffball",
"dev-util/bsdiff", "dev-util/bsdiff"]),
sorted(ipkgset))
vpkgset = installed.VersionedInstalled([fake_vdb])
self.assertEqual(sorted(["dev-util/diffball-1.0",
"dev-util/bsdiff-1.2", "dev-util/bsdiff-1.3"]),
sorted(vpkgset))
def __init__(self,
masked_use={},
stable_masked_use={},
forced_use={},
stable_forced_use={},
provides={},
iuse_effective=[],
masks=[],
unmasks=[],
arch='x86',
name='none'):
self.provides_repo = SimpleTree(provides)
self.masked_use = ChunkedDataDict()
self.masked_use.update_from_stream(
chunked_data(atom(k), *split_negations(v))
for k, v in masked_use.iteritems())
self.masked_use.freeze()
self.stable_masked_use = ChunkedDataDict()
self.stable_masked_use.update_from_stream(
chunked_data(atom(k), *split_negations(v))
for k, v in stable_masked_use.iteritems())
self.stable_masked_use.freeze()
self.forced_use = ChunkedDataDict()
self.forced_use.update_from_stream(
chunked_data(atom(k), *split_negations(v))
for k, v in forced_use.iteritems())
self.forced_use.freeze()
self.stable_forced_use = ChunkedDataDict()
self.stable_forced_use.update_from_stream(
chunked_data(atom(k), *split_negations(v))
for k, v in stable_forced_use.iteritems())
self.stable_forced_use.freeze()
self.masks = tuple(map(atom, masks))
self.unmasks = tuple(map(atom, unmasks))
self.iuse_effective = tuple(iuse_effective)
self.arch = arch
self.name = name
class TestPrototype:
def setup_method(self):
# we use an OrderedDict here specifically to trigger any sorter
# related bugs
d = {
"dev-util": {
"diffball": ["1.0", "0.7"],
"bsdiff": ["0.4.1", "0.4.2"]
},
"dev-lib": {
"fake": ["1.0", "1.0-r1"]
}
}
self.repo = SimpleTree(
OrderedDict((k, d[k]) for k in sorted(d, reverse=True)))
def test_concurrent_access(self):
iall = iter(self.repo)
self.repo.match(atom("dev-lib/fake"))
pkg = next(iall)
if pkg.category == 'dev-util':
self.repo.match(atom("dev-lib/fake"))
else:
self.repo.match(atom("dev-util/diffball"))
# should not explode...
list(iall)
def test_internal_lookups(self):
assert sorted(self.repo.categories) == sorted(["dev-lib", "dev-util"])
assert \
sorted(map("/".join, self.repo.versions)) == \
sorted([x for x in ["dev-util/diffball", "dev-util/bsdiff", "dev-lib/fake"]])
assert \
sorted(
f"{cp[0]}/{cp[1]}-{v}"
for cp, t in self.repo.versions.items() for v in t) == \
sorted([
"dev-util/diffball-1.0", "dev-util/diffball-0.7",
"dev-util/bsdiff-0.4.1", "dev-util/bsdiff-0.4.2",
"dev-lib/fake-1.0", "dev-lib/fake-1.0-r1"])
def test_simple_query(self):
a = atom("=dev-util/diffball-1.0")
self.repo.match(a)
assert self.repo.match(a)
assert not self.repo.match(atom("dev-util/monkeys_rule"))
def test_identify_candidates(self):
with pytest.raises(TypeError):
self.repo.match("asdf")
rc = packages.PackageRestriction("category",
values.StrExactMatch("dev-util"))
assert \
sorted(set(x.package for x in self.repo.itermatch(rc))) == \
sorted(["diffball", "bsdiff"])
rp = packages.PackageRestriction("package",
values.StrExactMatch("diffball"))
assert list(
x.version
for x in self.repo.itermatch(rp, sorter=sorted)) == ["0.7", "1.0"]
assert \
self.repo.match(packages.OrRestriction(rc, rp), sorter=sorted) == \
sorted(VersionedCPV(x) for x in (
"dev-util/diffball-0.7", "dev-util/diffball-1.0",
"dev-util/bsdiff-0.4.1", "dev-util/bsdiff-0.4.2"))
assert \
sorted(self.repo.itermatch(packages.AndRestriction(rc, rp))) == \
sorted(VersionedCPV(x) for x in (
"dev-util/diffball-0.7", "dev-util/diffball-1.0"))
assert sorted(self.repo) == self.repo.match(packages.AlwaysTrue,
sorter=sorted)
# mix/match cat/pkg to check that it handles that corner case
# properly for sorting.
assert \
sorted(self.repo, reverse=True) == \
self.repo.match(packages.OrRestriction(
rc, rp, packages.AlwaysTrue),
sorter=partial(sorted, reverse=True))
rc2 = packages.PackageRestriction("category",
values.StrExactMatch("dev-lib"))
assert sorted(self.repo.itermatch(packages.AndRestriction(rp,
rc2))) == []
# note this mixes a category level match, and a pkg level
# match. they *must* be treated as an or.
assert \
sorted(self.repo.itermatch(packages.OrRestriction(rp, rc2))) == \
sorted(VersionedCPV(x) for x in (
"dev-util/diffball-0.7", "dev-util/diffball-1.0",
"dev-lib/fake-1.0", "dev-lib/fake-1.0-r1"))
# this is similar to the test above, but mixes a cat/pkg
# candidate with a pkg candidate
rp2 = packages.PackageRestriction("package",
values.StrExactMatch("fake"))
r = packages.OrRestriction(atom("dev-util/diffball"), rp2)
assert \
sorted(self.repo.itermatch(r)) == \
sorted(VersionedCPV(x) for x in (
"dev-util/diffball-0.7", "dev-util/diffball-1.0",
"dev-lib/fake-1.0", "dev-lib/fake-1.0-r1"))
assert \
sorted(self.repo.itermatch(
packages.OrRestriction(packages.AlwaysTrue, rp2))) == \
sorted(VersionedCPV(x) for x in (
"dev-util/diffball-0.7", "dev-util/diffball-1.0",
"dev-util/bsdiff-0.4.1", "dev-util/bsdiff-0.4.2",
"dev-lib/fake-1.0", "dev-lib/fake-1.0-r1"))
assert \
sorted(self.repo.itermatch(packages.PackageRestriction(
'category', values.StrExactMatch('dev-util', negate=True)))) == \
sorted(VersionedCPV(x) for x in ("dev-lib/fake-1.0", "dev-lib/fake-1.0-r1"))
obj = malleable_obj(livefs=False)
pkg_cls = post_curry(MutatedPkg, {'repo': obj})
assert \
sorted(self.repo.itermatch(boolean.AndRestriction(boolean.OrRestriction(
packages.PackageRestriction(
"repo.livefs", values.EqualityMatch(False)),
packages.PackageRestriction(
"category", values.StrExactMatch("virtual"))),
atom("dev-lib/fake")),
pkg_cls=pkg_cls)) == \
sorted(VersionedCPV(x) for x in (
"dev-lib/fake-1.0", "dev-lib/fake-1.0-r1"))
assert \
sorted(self.repo.itermatch(packages.PackageRestriction(
'category', values.StrExactMatch('dev-lib', negate=True),
negate=True))) == \
sorted(VersionedCPV(x) for x in (
"dev-lib/fake-1.0", "dev-lib/fake-1.0-r1"))
assert \
sorted(self.repo.itermatch(packages.PackageRestriction(
'category', values.StrExactMatch('dev-lib', negate=True), negate=True))) == \
sorted(VersionedCPV(x) for x in (
"dev-lib/fake-1.0", "dev-lib/fake-1.0-r1"))
def test_iter(self):
expected = sorted(
VersionedCPV(x)
for x in ("dev-util/diffball-1.0", "dev-util/diffball-0.7",
"dev-util/bsdiff-0.4.1", "dev-util/bsdiff-0.4.2",
"dev-lib/fake-1.0", "dev-lib/fake-1.0-r1"))
assert sorted(self.repo) == expected
def test_notify_remove(self):
pkg = VersionedCPV("dev-util/diffball-1.0")
self.repo.notify_remove_package(pkg)
assert list(self.repo.versions[(pkg.category, pkg.package)]) == ["0.7"]
# test version being emptied, and package updated
pkg = VersionedCPV("dev-util/diffball-0.7")
self.repo.notify_remove_package(pkg)
assert (pkg.category, pkg.package) not in self.repo.versions
assert pkg.package not in self.repo.packages[pkg.category]
# test no remaining packages, category updated
pkg = VersionedCPV("dev-util/bsdiff-0.4.1")
self.repo.notify_remove_package(pkg)
pkg = VersionedCPV("dev-util/bsdiff-0.4.2")
self.repo.notify_remove_package(pkg)
assert (pkg.category, pkg.package) not in self.repo.versions
assert pkg.category not in self.repo.packages
assert pkg.category not in self.repo.categories
def test_notify_add(self):
pkg = VersionedCPV("dev-util/diffball-1.2")
self.repo.notify_add_package(pkg)
assert sorted(self.repo.versions[(pkg.category, pkg.package)]) == \
sorted(["1.0", "1.2", "0.7"])
pkg = VersionedCPV("foo/bar-1.0")
self.repo.notify_add_package(pkg)
assert pkg.category in self.repo.categories
assert pkg.category in self.repo.packages
ver_key = (pkg.category, pkg.package)
assert ver_key in self.repo.versions
assert list(self.repo.versions[ver_key]) == ["1.0"]
pkg = VersionedCPV("foo/cows-1.0")
self.repo.notify_add_package(pkg)
assert (pkg.category, pkg.package) in self.repo.versions
def _simple_redirect_test(self,
attr,
arg1='=dev-util/diffball-1.0',
arg2=None):
l = []
uniq_obj = object()
def f(*a, **kw):
a = a[1:-1]
l.extend((a, kw))
return uniq_obj
# if replace, override _replace since replace reflects to it
class my_ops(operations):
locals()[f'_cmd_implementation_{attr}'] = f
self.repo.operations_kls = my_ops
args = [self.repo.match(atom(arg1))]
if arg2:
args.append(VersionedCPV(arg2))
self.repo.frozen = False
op = getattr(self.repo.operations, attr)
def simple_check(op, args, **kw):
l[:] = []
assert op(*args, **kw) == uniq_obj
assert len(l) == 2
assert list(l[0]) == args
assert l
assert self.repo.operations.supports(attr)
simple_check(op, args)
assert not l[1]
simple_check(op, args)
assert 'force' not in l[1]
self.repo.frozen = True
assert not self.repo.operations.supports(attr)
assert not hasattr(self.repo.operations, attr)
test_replace = post_curry(_simple_redirect_test,
'replace',
arg2='dev-util/diffball-1.1')
test_uninstall = post_curry(_simple_redirect_test, 'uninstall')
test_install = post_curry(_simple_redirect_test, 'install')
class TestPrototype(TestCase):
def setUp(self):
# we use an OrderedDict here specifically to trigger any sorter
# related bugs
d = {
"dev-util": {"diffball": ["1.0", "0.7"], "bsdiff": ["0.4.1", "0.4.2"]},
"dev-lib": {"fake": ["1.0", "1.0-r1"]}}
self.repo = SimpleTree(
OrderedDict((k, d[k]) for k in sorted(d, reverse=True)))
def test_concurrent_access(self):
iall = iter(self.repo)
self.repo.match(atom("dev-lib/fake"))
pkg = iall.next()
if pkg.category == 'dev-util':
self.repo.match(atom("dev-lib/fake"))
else:
self.repo.match(atom("dev-util/diffball"))
# should not explode...
list(iall)
def test_internal_lookups(self):
self.assertEqual(
sorted(self.repo.categories),
sorted(["dev-lib", "dev-util"]))
self.assertEqual(
sorted(map("/".join, self.repo.versions)),
sorted([x for x in ["dev-util/diffball", "dev-util/bsdiff", "dev-lib/fake"]]))
self.assertEqual(
sorted(
"%s/%s-%s" % (cp[0], cp[1], v)
for cp, t in self.repo.versions.iteritems() for v in t),
sorted([
"dev-util/diffball-1.0", "dev-util/diffball-0.7",
"dev-util/bsdiff-0.4.1", "dev-util/bsdiff-0.4.2",
"dev-lib/fake-1.0", "dev-lib/fake-1.0-r1"]))
def test_simple_query(self):
a = atom("=dev-util/diffball-1.0")
self.repo.match(a)
self.assertTrue(self.repo.match(a))
self.assertFalse(self.repo.match(atom("dev-util/monkeys_rule")))
def test_identify_candidates(self):
self.assertRaises(TypeError, self.repo.match, ("asdf"))
rc = packages.PackageRestriction(
"category", values.StrExactMatch("dev-util"))
self.assertEqual(
sorted(set(x.package for x in self.repo.itermatch(rc))),
sorted(["diffball", "bsdiff"]))
rp = packages.PackageRestriction(
"package", values.StrExactMatch("diffball"))
self.assertEqual(
list(x.version for x in self.repo.itermatch(rp, sorter=sorted)),
["0.7", "1.0"])
self.assertEqual(
self.repo.match(packages.OrRestriction(rc, rp), sorter=sorted),
sorted(versioned_CPV(x) for x in (
"dev-util/diffball-0.7", "dev-util/diffball-1.0",
"dev-util/bsdiff-0.4.1", "dev-util/bsdiff-0.4.2")))
self.assertEqual(
sorted(self.repo.itermatch(packages.AndRestriction(rc, rp))),
sorted(versioned_CPV(x) for x in (
"dev-util/diffball-0.7", "dev-util/diffball-1.0")))
self.assertEqual(
sorted(self.repo),
self.repo.match(packages.AlwaysTrue, sorter=sorted))
# mix/match cat/pkg to check that it handles that corner case
# properly for sorting.
self.assertEqual(
sorted(self.repo, reverse=True),
self.repo.match(packages.OrRestriction(
rc, rp, packages.AlwaysTrue),
sorter=partial(sorted, reverse=True)))
rc2 = packages.PackageRestriction(
"category", values.StrExactMatch("dev-lib"))
self.assertEqual(
sorted(self.repo.itermatch(packages.AndRestriction(rp, rc2))),
sorted([]))
# note this mixes a category level match, and a pkg level
# match. they *must* be treated as an or.
self.assertEqual(
sorted(self.repo.itermatch(packages.OrRestriction(rp, rc2))),
sorted(versioned_CPV(x) for x in (
"dev-util/diffball-0.7", "dev-util/diffball-1.0",
"dev-lib/fake-1.0", "dev-lib/fake-1.0-r1")))
# this is similar to the test above, but mixes a cat/pkg
# candidate with a pkg candidate
rp2 = packages.PackageRestriction(
"package", values.StrExactMatch("fake"))
r = packages.OrRestriction(atom("dev-util/diffball"), rp2)
self.assertEqual(
sorted(self.repo.itermatch(r)),
sorted(versioned_CPV(x) for x in (
"dev-util/diffball-0.7", "dev-util/diffball-1.0",
"dev-lib/fake-1.0", "dev-lib/fake-1.0-r1")))
self.assertEqual(
sorted(self.repo.itermatch(
packages.OrRestriction(packages.AlwaysTrue, rp2))),
sorted(versioned_CPV(x) for x in (
"dev-util/diffball-0.7", "dev-util/diffball-1.0",
"dev-util/bsdiff-0.4.1", "dev-util/bsdiff-0.4.2",
"dev-lib/fake-1.0", "dev-lib/fake-1.0-r1")))
self.assertEqual(
sorted(self.repo.itermatch(
packages.PackageRestriction(
'category', values.StrExactMatch('dev-util', negate=True)))),
sorted(versioned_CPV(x) for x in ("dev-lib/fake-1.0", "dev-lib/fake-1.0-r1")))
obj = malleable_obj(livefs=False)
pkg_kls_override = post_curry(MutatedPkg, {'repo': obj})
self.assertEqual(
sorted(self.repo.itermatch(
boolean.AndRestriction(
boolean.OrRestriction(
packages.PackageRestriction(
"repo.livefs", values.EqualityMatch(False)),
packages.PackageRestriction(
"category", values.StrExactMatch("virtual"))),
atom("dev-lib/fake")),
pkg_klass_override=pkg_kls_override)),
sorted(versioned_CPV(x) for x in (
"dev-lib/fake-1.0", "dev-lib/fake-1.0-r1")))
self.assertEqual(
sorted(self.repo.itermatch(
packages.PackageRestriction(
'category', values.StrExactMatch('dev-lib', negate=True),
negate=True))),
sorted(versioned_CPV(x) for x in (
"dev-lib/fake-1.0", "dev-lib/fake-1.0-r1")))
self.assertEqual(
sorted(self.repo.itermatch(
packages.PackageRestriction(
'category', values.StrExactMatch('dev-lib', negate=True),
negate=True))),
sorted(versioned_CPV(x) for x in (
"dev-lib/fake-1.0", "dev-lib/fake-1.0-r1")))
def test_iter(self):
self.assertEqual(
sorted(self.repo),
sorted(versioned_CPV(x) for x in (
"dev-util/diffball-1.0", "dev-util/diffball-0.7",
"dev-util/bsdiff-0.4.1", "dev-util/bsdiff-0.4.2",
"dev-lib/fake-1.0", "dev-lib/fake-1.0-r1")))
def test_notify_remove(self):
pkg = versioned_CPV("dev-util/diffball-1.0")
self.repo.notify_remove_package(pkg)
self.assertEqual(list(self.repo.versions[
(pkg.category, pkg.package)]), ["0.7"])
# test version being emptied, and package updated
pkg = versioned_CPV("dev-util/diffball-0.7")
self.repo.notify_remove_package(pkg)
self.assertNotIn((pkg.category, pkg.package), self.repo.versions)
self.assertNotIn(pkg.package, self.repo.packages[pkg.category])
# test no remaining packages, category updated
pkg = versioned_CPV("dev-util/bsdiff-0.4.1")
self.repo.notify_remove_package(pkg)
pkg = versioned_CPV("dev-util/bsdiff-0.4.2")
self.repo.notify_remove_package(pkg)
self.assertNotIn((pkg.category, pkg.package), self.repo.versions)
self.assertNotIn(pkg.category, self.repo.packages)
self.assertNotIn(pkg.category, self.repo.categories)
def test_notify_add(self):
pkg = versioned_CPV("dev-util/diffball-1.2")
self.repo.notify_add_package(pkg)
self.assertEqual(sorted(self.repo.versions[
(pkg.category, pkg.package)]), sorted(["1.0", "1.2", "0.7"]))
pkg = versioned_CPV("foo/bar-1.0")
self.repo.notify_add_package(pkg)
self.assertIn(pkg.category, self.repo.categories)
self.assertIn(pkg.category, self.repo.packages)
ver_key = (pkg.category, pkg.package)
self.assertIn(ver_key, self.repo.versions)
self.assertEqual(list(self.repo.versions[ver_key]), ["1.0"])
pkg = versioned_CPV("foo/cows-1.0")
self.repo.notify_add_package(pkg)
self.assertIn((pkg.category, pkg.package), self.repo.versions)
def _simple_redirect_test(self, attr, arg1='=dev-util/diffball-1.0', arg2=None):
l = []
uniq_obj = object()
def f(*a, **kw):
a = a[1:-1]
l.extend((a, kw))
return uniq_obj
# if replace, override _replace since replace reflects to it
class my_ops(operations):
locals()['_cmd_implementation_%s' % attr] = f
self.repo.operations_kls = my_ops
args = [self.repo.match(atom(arg1))]
if arg2:
args.append(versioned_CPV(arg2))
self.repo.frozen = False
op = getattr(self.repo.operations, attr)
def simple_check(op, args, **kw):
l[:] = []
self.assertEqual(op(*args, **kw), uniq_obj)
self.assertEqual(len(l), 2)
self.assertEqual(list(l[0]), args)
self.assertTrue(l)
self.assertTrue(self.repo.operations.supports(attr))
simple_check(op, args)
self.assertFalse(l[1])
simple_check(op, args)
self.assertNotIn('force', l[1])
self.repo.frozen = True
self.assertFalse(self.repo.operations.supports(attr))
self.assertFalse(hasattr(self.repo.operations, attr))
test_replace = post_curry(_simple_redirect_test, 'replace', arg2='dev-util/diffball-1.1')
test_uninstall = post_curry(_simple_redirect_test, 'uninstall')
test_install = post_curry(_simple_redirect_test, 'install')
class TestPrototype(TestCase):
def setUp(self):
# we an orderreddict here specifically to trigger any sorter
# related bugs
d = {
"dev-util": {
"diffball": ["1.0", "0.7"],
"bsdiff": ["0.4.1", "0.4.2"]
},
"dev-lib": {
"fake": ["1.0", "1.0-r1"]
}
}
self.repo = SimpleTree(
OrderedDict((k, d[k]) for k in sorted(d, reverse=True)))
def test_concurrent_access(self):
iall = iter(self.repo)
self.repo.match(atom("dev-lib/fake"))
pkg = iall.next()
if pkg.category == 'dev-util':
self.repo.match(atom("dev-lib/fake"))
else:
self.repo.match(atom("dev-util/diffball"))
# should not explode...
list(iall)
def test_internal_lookups(self):
self.assertEqual(sorted(self.repo.categories),
sorted(["dev-lib", "dev-util"]))
self.assertEqual(
sorted(map("/".join, self.repo.versions)),
sorted([
x for x in
["dev-util/diffball", "dev-util/bsdiff", "dev-lib/fake"]
]))
self.assertEqual(
sorted("%s/%s-%s" % (cp[0], cp[1], v)
for cp, t in self.repo.versions.iteritems() for v in t),
sorted([
"dev-util/diffball-1.0", "dev-util/diffball-0.7",
"dev-util/bsdiff-0.4.1", "dev-util/bsdiff-0.4.2",
"dev-lib/fake-1.0", "dev-lib/fake-1.0-r1"
]))
def test_simple_query(self):
a = atom("=dev-util/diffball-1.0")
self.repo.match(a)
self.assertTrue(self.repo.match(a))
self.assertFalse(self.repo.match(atom("dev-util/monkeys_rule")))
def test_identify_candidates(self):
self.assertRaises(TypeError, self.repo.match, ("asdf"))
rc = packages.PackageRestriction("category",
values.StrExactMatch("dev-util"))
self.assertEqual(
sorted(set(x.package for x in self.repo.itermatch(rc))),
sorted(["diffball", "bsdiff"]))
rp = packages.PackageRestriction("package",
values.StrExactMatch("diffball"))
self.assertEqual(
list(x.version for x in self.repo.itermatch(rp, sorter=sorted)),
["0.7", "1.0"])
self.assertEqual(
self.repo.match(packages.OrRestriction(rc, rp), sorter=sorted),
sorted(
versioned_CPV(x)
for x in ("dev-util/diffball-0.7", "dev-util/diffball-1.0",
"dev-util/bsdiff-0.4.1", "dev-util/bsdiff-0.4.2")))
self.assertEqual(
sorted(self.repo.itermatch(packages.AndRestriction(rc, rp))),
sorted(
versioned_CPV(x)
for x in ("dev-util/diffball-0.7", "dev-util/diffball-1.0")))
self.assertEqual(sorted(self.repo),
self.repo.match(packages.AlwaysTrue, sorter=sorted))
# mix/match cat/pkg to check that it handles that corner case
# properly for sorting.
self.assertEqual(
sorted(self.repo, reverse=True),
self.repo.match(packages.OrRestriction(rc, rp,
packages.AlwaysTrue),
sorter=partial(sorted, reverse=True)))
rc2 = packages.PackageRestriction("category",
values.StrExactMatch("dev-lib"))
self.assertEqual(
sorted(self.repo.itermatch(packages.AndRestriction(rp, rc2))),
sorted([]))
# note this mixes a category level match, and a pkg level
# match. they *must* be treated as an or.
self.assertEqual(
sorted(self.repo.itermatch(packages.OrRestriction(rp, rc2))),
sorted(
versioned_CPV(x)
for x in ("dev-util/diffball-0.7", "dev-util/diffball-1.0",
"dev-lib/fake-1.0", "dev-lib/fake-1.0-r1")))
# this is similar to the test above, but mixes a cat/pkg
# candidate with a pkg candidate
rp2 = packages.PackageRestriction("package",
values.StrExactMatch("fake"))
r = packages.OrRestriction(atom("dev-util/diffball"), rp2)
self.assertEqual(
sorted(self.repo.itermatch(r)),
sorted(
versioned_CPV(x)
for x in ("dev-util/diffball-0.7", "dev-util/diffball-1.0",
"dev-lib/fake-1.0", "dev-lib/fake-1.0-r1")))
self.assertEqual(
sorted(
self.repo.itermatch(
packages.OrRestriction(packages.AlwaysTrue, rp2))),
sorted(
versioned_CPV(x)
for x in ("dev-util/diffball-0.7", "dev-util/diffball-1.0",
"dev-util/bsdiff-0.4.1", "dev-util/bsdiff-0.4.2",
"dev-lib/fake-1.0", "dev-lib/fake-1.0-r1")))
self.assertEqual(
sorted(
self.repo.itermatch(
packages.PackageRestriction(
'category',
values.StrExactMatch('dev-util', negate=True)))),
sorted(
versioned_CPV(x)
for x in ("dev-lib/fake-1.0", "dev-lib/fake-1.0-r1")))
obj = malleable_obj(livefs=False)
pkg_kls_override = post_curry(MutatedPkg, {'repo': obj})
self.assertEqual(
sorted(
self.repo.itermatch(boolean.AndRestriction(
boolean.OrRestriction(
packages.PackageRestriction(
"repo.livefs", values.EqualityMatch(False)),
packages.PackageRestriction(
"category", values.StrExactMatch("virtual"))),
atom("dev-lib/fake")),
pkg_klass_override=pkg_kls_override)),
sorted(
versioned_CPV(x)
for x in ("dev-lib/fake-1.0", "dev-lib/fake-1.0-r1")))
self.assertEqual(
sorted(
self.repo.itermatch(
packages.PackageRestriction('category',
values.StrExactMatch(
'dev-lib', negate=True),
negate=True))),
sorted(
versioned_CPV(x)
for x in ("dev-lib/fake-1.0", "dev-lib/fake-1.0-r1")))
self.assertEqual(
sorted(
self.repo.itermatch(
packages.PackageRestriction('category',
values.StrExactMatch(
'dev-lib', negate=True),
negate=True))),
sorted(
versioned_CPV(x)
for x in ("dev-lib/fake-1.0", "dev-lib/fake-1.0-r1")))
def test_iter(self):
self.assertEqual(
sorted(self.repo),
sorted(
versioned_CPV(x)
for x in ("dev-util/diffball-1.0", "dev-util/diffball-0.7",
"dev-util/bsdiff-0.4.1", "dev-util/bsdiff-0.4.2",
"dev-lib/fake-1.0", "dev-lib/fake-1.0-r1")))
def test_notify_remove(self):
pkg = versioned_CPV("dev-util/diffball-1.0")
self.repo.notify_remove_package(pkg)
self.assertEqual(list(self.repo.versions[(pkg.category, pkg.package)]),
["0.7"])
# test version being emptied, and package updated
pkg = versioned_CPV("dev-util/diffball-0.7")
self.repo.notify_remove_package(pkg)
self.assertNotIn((pkg.category, pkg.package), self.repo.versions)
self.assertNotIn(pkg.package, self.repo.packages[pkg.category])
# test no remaining packages, category updated
pkg = versioned_CPV("dev-util/bsdiff-0.4.1")
self.repo.notify_remove_package(pkg)
pkg = versioned_CPV("dev-util/bsdiff-0.4.2")
self.repo.notify_remove_package(pkg)
self.assertNotIn((pkg.category, pkg.package), self.repo.versions)
self.assertNotIn(pkg.category, self.repo.packages)
self.assertNotIn(pkg.category, self.repo.categories)
def test_notify_add(self):
pkg = versioned_CPV("dev-util/diffball-1.2")
self.repo.notify_add_package(pkg)
self.assertEqual(
sorted(self.repo.versions[(pkg.category, pkg.package)]),
sorted(["1.0", "1.2", "0.7"]))
pkg = versioned_CPV("foo/bar-1.0")
self.repo.notify_add_package(pkg)
self.assertIn(pkg.category, self.repo.categories)
self.assertIn(pkg.category, self.repo.packages)
ver_key = (pkg.category, pkg.package)
self.assertIn(ver_key, self.repo.versions)
self.assertEqual(list(self.repo.versions[ver_key]), ["1.0"])
pkg = versioned_CPV("foo/cows-1.0")
self.repo.notify_add_package(pkg)
self.assertIn((pkg.category, pkg.package), self.repo.versions)
def _simple_redirect_test(self,
attr,
arg1='=dev-util/diffball-1.0',
arg2=None):
l = []
uniq_obj = object()
def f(*a, **kw):
a = a[1:-1]
l.extend((a, kw))
return uniq_obj
# if replace, override _replace since replace reflects to it
class my_ops(operations):
locals()['_cmd_implementation_%s' % attr] = f
self.repo.operations_kls = my_ops
args = [self.repo.match(atom(arg1))]
if arg2:
args.append(versioned_CPV(arg2))
self.repo.frozen = False
op = getattr(self.repo.operations, attr)
def simple_check(op, args, **kw):
l[:] = []
self.assertEqual(op(*args, **kw), uniq_obj)
self.assertEqual(len(l), 2)
self.assertEqual(list(l[0]), args)
self.assertTrue(l)
self.assertTrue(self.repo.operations.supports(attr))
simple_check(op, args)
self.assertFalse(l[1])
simple_check(op, args) #, force=True)
self.assertNotIn('force', l[1])
self.repo.frozen = True
self.assertFalse(self.repo.operations.supports(attr))
self.assertFalse(hasattr(self.repo.operations, attr))
test_replace = post_curry(_simple_redirect_test,
'replace',
arg2='dev-util/diffball-1.1')
test_uninstall = post_curry(_simple_redirect_test, 'uninstall')
test_install = post_curry(_simple_redirect_test, 'install')
