f21e3a2d6d
RHEL has an older version of the library which does not backport all the assertions that we used. In order for the tests to pass there we need to use names that exist everywhere. Signed-off-by: Lubomír Sedlář <lsedlar@redhat.com>
469 lines
17 KiB
Python
469 lines
17 KiB
Python
#!/usr/bin/env python2
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import mock
|
|
import os
|
|
import sys
|
|
try:
|
|
import unittest2 as unittest
|
|
except ImportError:
|
|
import unittest
|
|
import json
|
|
import tempfile
|
|
|
|
sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
|
|
|
|
from pungi.phases.pkgset import pkgsets
|
|
from tests import helpers
|
|
|
|
|
|
class MockPathInfo(object):
|
|
def __init__(self, topdir):
|
|
self.topdir = topdir
|
|
|
|
def build(self, build_info):
|
|
return self.topdir
|
|
|
|
def get_filename(self, rpm_info):
|
|
return '{name}@{version}@{release}@{arch}'.format(**rpm_info)
|
|
|
|
def signed(self, rpm_info, sigkey):
|
|
return os.path.join('signed', sigkey, self.get_filename(rpm_info))
|
|
|
|
def rpm(self, rpm_info):
|
|
return os.path.join('rpms', self.get_filename(rpm_info))
|
|
|
|
|
|
class MockFile(object):
|
|
def __init__(self, path):
|
|
if path.startswith('/tmp'):
|
|
# Drop /tmp/something/ from path
|
|
path = path.split('/', 3)[-1]
|
|
self.file_path = path
|
|
self.file_name = os.path.basename(path)
|
|
self.name, self.version, self.release, self.arch = self.file_name.split('@')
|
|
self.sourcerpm = '{0.name}-{0.version}-{0.release}.{0.arch}'.format(self)
|
|
self.exclusivearch = []
|
|
self.excludearch = []
|
|
|
|
def __hash__(self):
|
|
return hash(self.file_path)
|
|
|
|
def __repr__(self):
|
|
return self.file_path
|
|
|
|
def __eq__(self, other):
|
|
try:
|
|
return self.file_path == other.file_path
|
|
except AttributeError:
|
|
return self.file_path == other
|
|
|
|
def __le__(self, other):
|
|
try:
|
|
return self.file_path < other.file_path
|
|
except AttributeError:
|
|
return self.file_path < other
|
|
|
|
def __lt__(self, other):
|
|
return self <= other and self != other
|
|
|
|
def __ge__(self, other):
|
|
return not (self <= other) or self == other
|
|
|
|
def __gt__(self, other):
|
|
return not (self <= other)
|
|
|
|
|
|
class MockFileCache(dict):
|
|
"""Mock for kobo.pkgset.FileCache.
|
|
It gets data from filename and does not touch filesystem.
|
|
"""
|
|
def __init__(self, _wrapper):
|
|
super(MockFileCache, self).__init__()
|
|
self.file_cache = self
|
|
|
|
def add(self, file_path):
|
|
obj = MockFile(file_path)
|
|
self[file_path] = obj
|
|
return obj
|
|
|
|
|
|
class FakePool(object):
|
|
"""This class will be substituted for ReaderPool.
|
|
It implements the same interface, but uses only the last added worker to
|
|
process all tasks sequentially.
|
|
"""
|
|
def __init__(self, package_set, logger=None):
|
|
self.queue = []
|
|
self.worker = None
|
|
self.package_set = package_set
|
|
|
|
def log_warning(self, *args, **kwargs):
|
|
pass
|
|
|
|
@property
|
|
def queue_total(self):
|
|
return len(self.queue)
|
|
|
|
def queue_put(self, item):
|
|
self.queue.append(item)
|
|
|
|
def add(self, worker):
|
|
self.worker = worker
|
|
|
|
def start(self):
|
|
for i, item in enumerate(self.queue):
|
|
self.worker.process(item, i)
|
|
|
|
def stop(self):
|
|
pass
|
|
|
|
|
|
class PkgsetCompareMixin(object):
|
|
def assertPkgsetEqual(self, actual, expected):
|
|
for k, v1 in expected.items():
|
|
self.assertIn(k, actual)
|
|
v2 = actual.pop(k)
|
|
self.assertItemsEqual(v1, v2)
|
|
self.assertEqual({}, actual)
|
|
|
|
|
|
@mock.patch('pungi.phases.pkgset.pkgsets.ReaderPool', new=FakePool)
|
|
@mock.patch('kobo.pkgset.FileCache', new=MockFileCache)
|
|
class TestKojiPkgset(PkgsetCompareMixin, helpers.PungiTestCase):
|
|
|
|
def setUp(self):
|
|
super(TestKojiPkgset, self).setUp()
|
|
with open(os.path.join(helpers.FIXTURE_DIR, 'tagged-rpms.json')) as f:
|
|
self.tagged_rpms = json.load(f)
|
|
|
|
self.path_info = MockPathInfo(self.topdir)
|
|
|
|
self.koji_wrapper = mock.Mock()
|
|
self.koji_wrapper.koji_proxy.listTaggedRPMS.return_value = self.tagged_rpms
|
|
self.koji_wrapper.koji_module.pathinfo = self.path_info
|
|
|
|
def _touch_files(self, filenames):
|
|
for filename in filenames:
|
|
helpers.touch(os.path.join(self.topdir, filename))
|
|
|
|
def assertPkgsetEqual(self, actual, expected):
|
|
for k, v1 in expected.items():
|
|
self.assertIn(k, actual)
|
|
v2 = actual.pop(k)
|
|
self.assertItemsEqual(v1, v2)
|
|
self.assertEqual({}, actual, msg='Some architectures were missing')
|
|
|
|
def test_all_arches(self):
|
|
self._touch_files([
|
|
'rpms/pungi@4.1.3@3.fc25@noarch',
|
|
'rpms/pungi@4.1.3@3.fc25@src',
|
|
'rpms/bash@4.3.42@4.fc24@i686',
|
|
'rpms/bash@4.3.42@4.fc24@x86_64',
|
|
'rpms/bash@4.3.42@4.fc24@src',
|
|
'rpms/bash-debuginfo@4.3.42@4.fc24@i686',
|
|
'rpms/bash-debuginfo@4.3.42@4.fc24@x86_64',
|
|
])
|
|
|
|
pkgset = pkgsets.KojiPackageSet(self.koji_wrapper, [None])
|
|
|
|
result = pkgset.populate('f25', logfile=self.topdir + '/pkgset.log')
|
|
|
|
self.assertEqual(
|
|
self.koji_wrapper.koji_proxy.mock_calls,
|
|
[mock.call.listTaggedRPMS('f25', event=None, inherit=True, latest=True)])
|
|
|
|
self.assertPkgsetEqual(result,
|
|
{'src': ['rpms/pungi@4.1.3@3.fc25@src',
|
|
'rpms/bash@4.3.42@4.fc24@src'],
|
|
'noarch': ['rpms/pungi@4.1.3@3.fc25@noarch'],
|
|
'i686': ['rpms/bash@4.3.42@4.fc24@i686',
|
|
'rpms/bash-debuginfo@4.3.42@4.fc24@i686'],
|
|
'x86_64': ['rpms/bash@4.3.42@4.fc24@x86_64',
|
|
'rpms/bash-debuginfo@4.3.42@4.fc24@x86_64']})
|
|
|
|
self.maxDiff = None
|
|
with open(self.topdir + '/pkgset.log') as f:
|
|
self.assertEqual(
|
|
sorted(f.read().strip().splitlines()),
|
|
sorted(['pungi-0:4.1.3-3.fc25.noarch: f25 [335]',
|
|
'pungi-0:4.1.3-3.fc25.src: f25 [335]',
|
|
'bash-0:4.3.42-4.fc24.i686: f25 [335]',
|
|
'bash-0:4.3.42-4.fc24.x86_64: f25 [335]',
|
|
'bash-0:4.3.42-4.fc24.src: f25 [335]',
|
|
'bash-debuginfo-0:4.3.42-4.fc24.i686: f25 [335]',
|
|
'bash-debuginfo-0:4.3.42-4.fc24.x86_64: f25 [335]']))
|
|
|
|
def test_only_one_arch(self):
|
|
self._touch_files([
|
|
'rpms/bash@4.3.42@4.fc24@x86_64',
|
|
'rpms/bash-debuginfo@4.3.42@4.fc24@x86_64',
|
|
])
|
|
|
|
pkgset = pkgsets.KojiPackageSet(self.koji_wrapper, [None], arches=['x86_64'])
|
|
|
|
result = pkgset.populate('f25')
|
|
|
|
self.assertEqual(
|
|
self.koji_wrapper.koji_proxy.mock_calls,
|
|
[mock.call.listTaggedRPMS('f25', event=None, inherit=True, latest=True)])
|
|
|
|
self.assertPkgsetEqual(result,
|
|
{'x86_64': ['rpms/bash-debuginfo@4.3.42@4.fc24@x86_64',
|
|
'rpms/bash@4.3.42@4.fc24@x86_64']})
|
|
|
|
def test_find_signed_with_preference(self):
|
|
self._touch_files([
|
|
'signed/cafebabe/bash@4.3.42@4.fc24@x86_64',
|
|
'signed/deadbeef/bash@4.3.42@4.fc24@x86_64',
|
|
'signed/deadbeef/bash-debuginfo@4.3.42@4.fc24@x86_64',
|
|
])
|
|
|
|
pkgset = pkgsets.KojiPackageSet(self.koji_wrapper, ['cafebabe', 'deadbeef'], arches=['x86_64'])
|
|
|
|
result = pkgset.populate('f25')
|
|
|
|
self.assertEqual(
|
|
self.koji_wrapper.koji_proxy.mock_calls,
|
|
[mock.call.listTaggedRPMS('f25', event=None, inherit=True, latest=True)])
|
|
|
|
self.assertPkgsetEqual(result,
|
|
{'x86_64': ['signed/cafebabe/bash@4.3.42@4.fc24@x86_64',
|
|
'signed/deadbeef/bash-debuginfo@4.3.42@4.fc24@x86_64']})
|
|
|
|
def test_find_signed_fallback_unsigned(self):
|
|
self._touch_files([
|
|
'signed/cafebabe/bash@4.3.42@4.fc24@x86_64',
|
|
'rpms/bash-debuginfo@4.3.42@4.fc24@x86_64',
|
|
])
|
|
|
|
pkgset = pkgsets.KojiPackageSet(self.koji_wrapper, ['cafebabe', None], arches=['x86_64'])
|
|
|
|
result = pkgset.populate('f25')
|
|
|
|
self.assertEqual(
|
|
self.koji_wrapper.koji_proxy.mock_calls,
|
|
[mock.call.listTaggedRPMS('f25', event=None, inherit=True, latest=True)])
|
|
|
|
self.assertPkgsetEqual(result,
|
|
{'x86_64': ['rpms/bash-debuginfo@4.3.42@4.fc24@x86_64',
|
|
'signed/cafebabe/bash@4.3.42@4.fc24@x86_64']})
|
|
|
|
def test_can_not_find_signed_package(self):
|
|
pkgset = pkgsets.KojiPackageSet(self.koji_wrapper, ['cafebabe'], arches=['x86_64'])
|
|
|
|
with self.assertRaises(RuntimeError) as ctx:
|
|
pkgset.populate('f25')
|
|
|
|
self.assertEqual(
|
|
self.koji_wrapper.koji_proxy.mock_calls,
|
|
[mock.call.listTaggedRPMS('f25', event=None, inherit=True, latest=True)])
|
|
|
|
self.assertRegexpMatches(
|
|
str(ctx.exception),
|
|
r'^RPM\(s\) not found for sigs: .+Check log for details.+')
|
|
|
|
def test_can_not_find_any_package(self):
|
|
pkgset = pkgsets.KojiPackageSet(self.koji_wrapper, ['cafebabe', None], arches=['x86_64'])
|
|
|
|
with self.assertRaises(RuntimeError) as ctx:
|
|
pkgset.populate('f25')
|
|
|
|
self.assertEqual(
|
|
self.koji_wrapper.koji_proxy.mock_calls,
|
|
[mock.call.listTaggedRPMS('f25', event=None, inherit=True, latest=True)])
|
|
|
|
self.assertRegexpMatches(
|
|
str(ctx.exception),
|
|
r'^RPM\(s\) not found for sigs: .+Check log for details.+')
|
|
|
|
def test_packages_attribute(self):
|
|
self._touch_files([
|
|
'rpms/pungi@4.1.3@3.fc25@noarch',
|
|
'rpms/pungi@4.1.3@3.fc25@src',
|
|
'rpms/bash@4.3.42@4.fc24@i686',
|
|
'rpms/bash@4.3.42@4.fc24@x86_64',
|
|
'rpms/bash@4.3.42@4.fc24@src',
|
|
'rpms/bash-debuginfo@4.3.42@4.fc24@i686',
|
|
'rpms/bash-debuginfo@4.3.42@4.fc24@x86_64',
|
|
])
|
|
|
|
pkgset = pkgsets.KojiPackageSet(self.koji_wrapper, [None],
|
|
packages=["bash"])
|
|
|
|
result = pkgset.populate('f25', logfile=self.topdir + '/pkgset.log')
|
|
|
|
self.assertEqual(
|
|
self.koji_wrapper.koji_proxy.mock_calls,
|
|
[mock.call.listTaggedRPMS('f25', event=None, inherit=True, latest=True)])
|
|
|
|
self.assertPkgsetEqual(result,
|
|
{'src': ['rpms/bash@4.3.42@4.fc24@src'],
|
|
'i686': ['rpms/bash@4.3.42@4.fc24@i686'],
|
|
'x86_64': ['rpms/bash@4.3.42@4.fc24@x86_64']})
|
|
|
|
|
|
@mock.patch('kobo.pkgset.FileCache', new=MockFileCache)
|
|
class TestMergePackageSets(PkgsetCompareMixin, unittest.TestCase):
|
|
def test_merge_in_another_arch(self):
|
|
first = pkgsets.PackageSetBase([None])
|
|
second = pkgsets.PackageSetBase([None])
|
|
|
|
for name in ['rpms/pungi@4.1.3@3.fc25@noarch', 'rpms/pungi@4.1.3@3.fc25@src']:
|
|
pkg = first.file_cache.add(name)
|
|
first.rpms_by_arch.setdefault(pkg.arch, []).append(pkg)
|
|
|
|
for name in ['rpms/bash@4.3.42@4.fc24@i686']:
|
|
pkg = second.file_cache.add(name)
|
|
second.rpms_by_arch.setdefault(pkg.arch, []).append(pkg)
|
|
|
|
first.merge(second, 'i386', ['i686'])
|
|
|
|
self.assertPkgsetEqual(first.rpms_by_arch,
|
|
{'src': ['rpms/pungi@4.1.3@3.fc25@src'],
|
|
'noarch': ['rpms/pungi@4.1.3@3.fc25@noarch'],
|
|
'i686': ['rpms/bash@4.3.42@4.fc24@i686']})
|
|
|
|
def test_merge_includes_noarch_with_different_exclude_arch(self):
|
|
first = pkgsets.PackageSetBase([None])
|
|
second = pkgsets.PackageSetBase([None])
|
|
|
|
pkg = first.file_cache.add('rpms/bash@4.3.42@4.fc24@i686')
|
|
first.rpms_by_arch.setdefault(pkg.arch, []).append(pkg)
|
|
|
|
pkg = second.file_cache.add('rpms/pungi@4.1.3@3.fc25@noarch')
|
|
pkg.excludearch = ['x86_64']
|
|
second.rpms_by_arch.setdefault(pkg.arch, []).append(pkg)
|
|
|
|
first.merge(second, 'i386', ['i686', 'noarch'])
|
|
|
|
self.assertPkgsetEqual(first.rpms_by_arch,
|
|
{'i686': ['rpms/bash@4.3.42@4.fc24@i686'],
|
|
'noarch': ['rpms/pungi@4.1.3@3.fc25@noarch']})
|
|
|
|
def test_merge_excludes_noarch_exclude_arch(self):
|
|
first = pkgsets.PackageSetBase([None])
|
|
second = pkgsets.PackageSetBase([None])
|
|
|
|
pkg = first.file_cache.add('rpms/bash@4.3.42@4.fc24@i686')
|
|
first.rpms_by_arch.setdefault(pkg.arch, []).append(pkg)
|
|
|
|
pkg = second.file_cache.add('rpms/pungi@4.1.3@3.fc25@noarch')
|
|
pkg.excludearch = ['i686']
|
|
second.rpms_by_arch.setdefault(pkg.arch, []).append(pkg)
|
|
|
|
first.merge(second, 'i386', ['i686', 'noarch'])
|
|
|
|
self.assertPkgsetEqual(first.rpms_by_arch,
|
|
{'i686': ['rpms/bash@4.3.42@4.fc24@i686'],
|
|
'noarch': []})
|
|
|
|
def test_merge_excludes_noarch_exclusive_arch(self):
|
|
first = pkgsets.PackageSetBase([None])
|
|
second = pkgsets.PackageSetBase([None])
|
|
|
|
pkg = first.file_cache.add('rpms/bash@4.3.42@4.fc24@i686')
|
|
first.rpms_by_arch.setdefault(pkg.arch, []).append(pkg)
|
|
|
|
pkg = second.file_cache.add('rpms/pungi@4.1.3@3.fc25@noarch')
|
|
pkg.exclusivearch = ['x86_64']
|
|
second.rpms_by_arch.setdefault(pkg.arch, []).append(pkg)
|
|
|
|
first.merge(second, 'i386', ['i686', 'noarch'])
|
|
|
|
self.assertPkgsetEqual(first.rpms_by_arch,
|
|
{'i686': ['rpms/bash@4.3.42@4.fc24@i686'],
|
|
'noarch': []})
|
|
|
|
def test_merge_includes_noarch_with_same_exclusive_arch(self):
|
|
first = pkgsets.PackageSetBase([None])
|
|
second = pkgsets.PackageSetBase([None])
|
|
|
|
pkg = first.file_cache.add('rpms/bash@4.3.42@4.fc24@i686')
|
|
first.rpms_by_arch.setdefault(pkg.arch, []).append(pkg)
|
|
|
|
pkg = second.file_cache.add('rpms/pungi@4.1.3@3.fc25@noarch')
|
|
pkg.exclusivearch = ['i686']
|
|
second.rpms_by_arch.setdefault(pkg.arch, []).append(pkg)
|
|
|
|
first.merge(second, 'i386', ['i686', 'noarch'])
|
|
|
|
self.assertPkgsetEqual(first.rpms_by_arch,
|
|
{'i686': ['rpms/bash@4.3.42@4.fc24@i686'],
|
|
'noarch': ['rpms/pungi@4.1.3@3.fc25@noarch']})
|
|
|
|
def test_merge_skips_package_in_cache(self):
|
|
first = pkgsets.PackageSetBase([None])
|
|
second = pkgsets.PackageSetBase([None])
|
|
|
|
pkg = first.file_cache.add('rpms/bash@4.3.42@4.fc24@i686')
|
|
first.rpms_by_arch.setdefault(pkg.arch, []).append(pkg)
|
|
|
|
pkg = second.file_cache.add('rpms/bash@4.3.42@4.fc24@i686')
|
|
second.rpms_by_arch.setdefault(pkg.arch, []).append(pkg)
|
|
|
|
first.merge(second, 'i386', ['i686'])
|
|
|
|
self.assertPkgsetEqual(first.rpms_by_arch,
|
|
{'i686': ['rpms/bash@4.3.42@4.fc24@i686']})
|
|
|
|
def test_merge_skips_src_without_binary(self):
|
|
first = pkgsets.PackageSetBase([None])
|
|
second = pkgsets.PackageSetBase([None])
|
|
|
|
pkg = first.file_cache.add('rpms/bash@4.3.42@4.fc24@i686')
|
|
first.rpms_by_arch.setdefault(pkg.arch, []).append(pkg)
|
|
|
|
pkg = second.file_cache.add('rpms/pungi@4.1.3@3.fc25@src')
|
|
second.rpms_by_arch.setdefault(pkg.arch, []).append(pkg)
|
|
|
|
first.merge(second, 'i386', ['i686', 'src'])
|
|
|
|
self.assertPkgsetEqual(first.rpms_by_arch,
|
|
{'i686': ['rpms/bash@4.3.42@4.fc24@i686'],
|
|
'src': [],
|
|
'nosrc': []})
|
|
|
|
|
|
@mock.patch('kobo.pkgset.FileCache', new=MockFileCache)
|
|
class TestSaveFileList(unittest.TestCase):
|
|
def setUp(self):
|
|
fd, self.tmpfile = tempfile.mkstemp()
|
|
os.close(fd)
|
|
|
|
def tearDown(self):
|
|
os.unlink(self.tmpfile)
|
|
|
|
def test_save_arches_alphabetically(self):
|
|
pkgset = pkgsets.PackageSetBase([None])
|
|
for name in ['rpms/pungi@4.1.3@3.fc25@x86_64',
|
|
'rpms/pungi@4.1.3@3.fc25@src',
|
|
'rpms/pungi@4.1.3@3.fc25@ppc64']:
|
|
pkg = pkgset.file_cache.add(name)
|
|
pkgset.rpms_by_arch.setdefault(pkg.arch, []).append(pkg)
|
|
|
|
pkgset.save_file_list(self.tmpfile)
|
|
|
|
with open(self.tmpfile) as f:
|
|
rpms = f.read().strip().split('\n')
|
|
self.assertEqual(rpms, ['rpms/pungi@4.1.3@3.fc25@ppc64',
|
|
'rpms/pungi@4.1.3@3.fc25@src',
|
|
'rpms/pungi@4.1.3@3.fc25@x86_64'])
|
|
|
|
def test_save_strip_prefix(self):
|
|
pkgset = pkgsets.PackageSetBase([None])
|
|
for name in ['rpms/pungi@4.1.3@3.fc25@noarch', 'rpms/pungi@4.1.3@3.fc25@src']:
|
|
pkg = pkgset.file_cache.add(name)
|
|
pkgset.rpms_by_arch.setdefault(pkg.arch, []).append(pkg)
|
|
|
|
pkgset.save_file_list(self.tmpfile, remove_path_prefix='rpms/')
|
|
|
|
with open(self.tmpfile) as f:
|
|
rpms = f.read().strip().split('\n')
|
|
self.assertItemsEqual(rpms, ['pungi@4.1.3@3.fc25@noarch',
|
|
'pungi@4.1.3@3.fc25@src'])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|