def test_install_units(self):
repo_id = "test_repo_id"
errata_obj = self.get_test_errata_object()
errata_unit = Unit(TYPE_ID_ERRATA, {"id":errata_obj["id"]}, errata_obj, None)
existing_units = [errata_unit]
test_repo = profiler_mocks.get_repo(repo_id)
conduit = profiler_mocks.get_profiler_conduit(existing_units=existing_units,
repo_bindings=[test_repo])
example_errata = {"unit_key":errata_unit.unit_key, "type_id":TYPE_ID_ERRATA}
prof = YumProfiler()
translated_units = prof.install_units(self.test_consumer, [example_errata], None, None,
conduit)
# check repo_id passed to the conduit get_units()
self.assertEqual(conduit.get_units.call_args[0][0].id, repo_id)
# check unit association criteria passed to the conduit get_units()
self.assertEqual(conduit.get_units.call_args[0][1].type_ids, [TYPE_ID_ERRATA])
self.assertEqual(conduit.get_units.call_args[0][1].unit_filters, errata_unit.unit_key)
# validate translated units
self.assertEqual(len(translated_units), 2)
expected = []
for r in prof._get_rpms_from_errata(errata_unit):
expected_name = "%s-%s:%s-%s.%s" % (r["name"], r["epoch"], r["version"], r["release"],
r["arch"])
expected.append(expected_name)
for u in translated_units:
rpm_name = u["unit_key"]["name"]
self.assertTrue(rpm_name in expected)
def test_unit_applicable_false(self):
rpm_unit_key = self.create_profile_entry("bla-bla", 0, "0.1", "2", "x86_64", "Test Vendor")
rpm_unit = Unit(TYPE_ID_RPM, rpm_unit_key, {}, None)
test_repo = profiler_mocks.get_repo("test_repo_id")
conduit = profiler_mocks.get_profiler_conduit(repo_units=[rpm_unit], repo_bindings=[test_repo])
prof = YumProfiler()
unit_profile = self.test_consumer.profiles[TYPE_ID_RPM]
bound_repo_id = "test_repo_id"
report_list = prof.calculate_applicable_units(unit_profile, bound_repo_id, None, conduit)
self.assertEqual(report_list, {TYPE_ID_RPM: [], TYPE_ID_ERRATA: []})
def test_unit_applicable_updated_rpm_already_installed(self):
# Errata refers to RPMs already installed, i.e. the consumer has these exact NEVRA already
errata_obj = self.get_test_errata_object()
errata_unit = Unit(TYPE_ID_ERRATA, {"id":errata_obj["id"]}, errata_obj, None)
existing_units = [errata_unit]
test_repo = profiler_mocks.get_repo("test_repo_id")
conduit = profiler_mocks.get_profiler_conduit(existing_units=existing_units, repo_bindings=[test_repo])
example_errata = {"unit_key":errata_unit.unit_key, "type_id":TYPE_ID_ERRATA}
prof = RPMErrataProfiler()
report = prof.unit_applicable(self.test_consumer_been_updated, example_errata, None, conduit)
self.assertFalse(report.applicable)
def test_unit_applicable_false(self):
# Errata refers to RPMs which are NOT part of our test consumer's profile
errata_obj = self.get_test_errata_object_unrelated()
errata_unit = Unit(TYPE_ID_ERRATA, {"id":errata_obj["id"]}, errata_obj, None)
existing_units = [errata_unit]
test_repo = profiler_mocks.get_repo("test_repo_id")
conduit = profiler_mocks.get_profiler_conduit(existing_units=existing_units, repo_bindings=[test_repo])
example_errata = {"unit_key":errata_unit.unit_key, "type_id":TYPE_ID_ERRATA}
prof = RPMErrataProfiler()
report = prof.unit_applicable(self.test_consumer, example_errata, None, conduit)
self.assertFalse(report.applicable)
def test_unit_applicable_true(self):
# Errata refers to RPMs which ARE part of our test consumer's profile
errata_obj = self.get_test_errata_object()
errata_unit = Unit(TYPE_ID_ERRATA, {"id":errata_obj["id"]}, errata_obj, None)
existing_units = [errata_unit]
test_repo = profiler_mocks.get_repo("test_repo_id")
conduit = profiler_mocks.get_profiler_conduit(existing_units=existing_units, repo_bindings=[test_repo])
example_errata = [errata_unit.unit_key]
prof = RPMErrataProfiler()
report_list = prof.units_applicable(self.test_consumer, ["test_repo_id"], TYPE_ID_ERRATA, example_errata, None, conduit)
self.assertFalse(report_list == [])
def test_unit_applicable_same_name_diff_arch(self):
# Errata refers to RPMs that are x86_64, the test consumer is i386
# the rpms installed share the same name as the errata, but the client arch is different
# so this errata is marked as unapplicable
errata_obj = self.get_test_errata_object()
errata_unit = Unit(TYPE_ID_ERRATA, {"id":errata_obj["id"]}, errata_obj, None)
existing_units = [errata_unit]
test_repo = profiler_mocks.get_repo("test_repo_id")
conduit = profiler_mocks.get_profiler_conduit(existing_units=existing_units, repo_bindings=[test_repo])
example_errata = {"unit_key":errata_unit.unit_key, "type_id":TYPE_ID_ERRATA}
prof = RPMErrataProfiler()
report = prof.unit_applicable(self.test_consumer_i386, example_errata, None, conduit)
self.assertFalse(report.applicable)
def test_translate(self):
# Setup test data
errata_obj = self.get_test_errata_object()
errata_unit = Unit(TYPE_ID_ERRATA, {"id":errata_obj["id"]}, errata_obj, None)
existing_units = [errata_unit]
test_repo = profiler_mocks.get_repo("test_repo_id")
conduit = profiler_mocks.get_profiler_conduit(existing_units=existing_units, repo_bindings=[test_repo])
example_errata = {"unit_key":errata_unit.unit_key, "type_id":TYPE_ID_ERRATA}
# Test
prof = RPMErrataProfiler()
applicable_rpms, upgrade_details = prof.translate(example_errata, self.test_consumer, conduit)
self.assertEqual(len(applicable_rpms), 2)
self.assertEqual(len(upgrade_details), 2)
def test_unit_applicable_true(self):
rpm_unit_key = self.create_profile_entry("emoticons", 0, "0.1", "2", "x86_64", "Test Vendor")
rpm_unit = Unit(TYPE_ID_RPM, rpm_unit_key, {}, None)
# Let's give it an id, so we can assert for it later
rpm_unit.id = "a_test_id"
test_repo = profiler_mocks.get_repo("test_repo_id")
conduit = profiler_mocks.get_profiler_conduit(repo_units=[rpm_unit], repo_bindings=[test_repo])
prof = YumProfiler()
unit_profile = self.test_consumer.profiles[TYPE_ID_RPM]
bound_repo_id = "test_repo_id"
report_list = prof.calculate_applicable_units(unit_profile, bound_repo_id, None, conduit)
self.assertEqual(report_list, {TYPE_ID_RPM: [rpm_unit.id], TYPE_ID_ERRATA: []})
def test_unit_applicable_false(self):
rpm_unit_key = self.create_profile_entry("bla-bla", 0, "0.1", "2", "x86_64", "Test Vendor")
rpm_unit = Unit(TYPE_ID_RPM, rpm_unit_key, {}, None)
existing_units = [rpm_unit]
test_repo = profiler_mocks.get_repo("test_repo_id")
conduit = profiler_mocks.get_profiler_conduit(existing_units=existing_units, repo_bindings=[test_repo])
example_rpms = [rpm_unit.unit_key]
prof = RPMPkgProfiler()
report_list = prof.units_applicable(
self.test_consumer_i386, ["test_repo_id"], TYPE_ID_RPM, example_rpms, None, conduit
)
self.assertTrue(report_list == [])
def test_unit_applicable_updated_rpm_already_installed(self):
# Errata refers to RPMs already installed, i.e. the consumer has these exact NEVRA already
errata_obj = self.get_test_errata_object()
errata_unit = Unit(TYPE_ID_ERRATA, {"id": errata_obj["id"]}, errata_obj, None)
test_repo = profiler_mocks.get_repo("test_repo_id")
prof = YumProfiler()
errata_rpms = prof._get_rpms_from_errata(errata_unit)
conduit = profiler_mocks.get_profiler_conduit(repo_units=[errata_unit],
repo_bindings=[test_repo],
errata_rpms=errata_rpms)
unit_profile = self.test_consumer_been_updated.profiles[TYPE_ID_RPM]
bound_repo_id = "test_repo_id"
report_list = prof.calculate_applicable_units(unit_profile, bound_repo_id, None, conduit)
self.assertEqual(report_list, {TYPE_ID_RPM: [], TYPE_ID_ERRATA: []})
def test_unit_applicable_false(self):
# Errata refers to RPMs which are NOT part of our test consumer's profile
errata_obj = self.get_test_errata_object_unrelated()
errata_unit = Unit(TYPE_ID_ERRATA, {"id": errata_obj["id"]}, errata_obj, None)
test_repo = profiler_mocks.get_repo("test_repo_id")
prof = YumProfiler()
errata_rpms = prof._get_rpms_from_errata(errata_unit)
conduit = profiler_mocks.get_profiler_conduit(repo_units=[errata_unit],
repo_bindings=[test_repo],
errata_rpms=errata_rpms)
unit_profile = self.test_consumer.profiles[TYPE_ID_RPM]
bound_repo_id = "test_repo_id"
report_list = prof.calculate_applicable_units(unit_profile, bound_repo_id, None, conduit)
self.assertEqual(report_list, {TYPE_ID_ERRATA: [], TYPE_ID_RPM: []})
def test_install_units(self):
errata_obj = self.get_test_errata_object()
errata_unit = Unit(TYPE_ID_ERRATA, {"id":errata_obj["id"]}, errata_obj, None)
existing_units = [errata_unit]
test_repo = profiler_mocks.get_repo("test_repo_id")
conduit = profiler_mocks.get_profiler_conduit(existing_units=existing_units, repo_bindings=[test_repo])
example_errata = {"unit_key":errata_unit.unit_key, "type_id":TYPE_ID_ERRATA}
prof = RPMErrataProfiler()
translated_units = prof.install_units(self.test_consumer, [example_errata], None, None, conduit)
self.assertEqual(len(translated_units), 2)
expected = []
for r in self.test_consumer.profiles[TYPE_ID_RPM]:
expected_name = "%s.%s" % (r["name"], r["arch"])
expected.append(expected_name)
for u in translated_units:
rpm_name = u["unit_key"]["name"]
self.assertTrue(rpm_name in expected)
def test_unit_applicable_same_name_diff_arch(self):
# Errata refers to RPMs that are x86_64, the test consumer is i386
# the rpms installed share the same name as the errata, but the client arch is different
# so this errata is marked as unapplicable
errata_obj = self.get_test_errata_object()
errata_unit = Unit(TYPE_ID_ERRATA, {"id": errata_obj["id"]}, errata_obj, None)
test_repo = profiler_mocks.get_repo("test_repo_id")
prof = YumProfiler()
errata_rpms = prof._get_rpms_from_errata(errata_unit)
conduit = profiler_mocks.get_profiler_conduit(repo_units=[errata_unit],
repo_bindings=[test_repo],
errata_rpms=errata_rpms)
unit_profile = self.test_consumer_i386.profiles[TYPE_ID_RPM]
bound_repo_id = "test_repo_id"
report_list = prof.calculate_applicable_units(unit_profile, bound_repo_id, None, conduit)
self.assertEqual(report_list, {TYPE_ID_RPM: [], TYPE_ID_ERRATA: []})
def test_install_units(self):
"""
Verify that all available packages in the erratum are installed
In this test, there are two packages in the erratum, and both are
available to the consumer. Thus, both should be installed.
"""
repo_id = "test_repo_id"
errata_obj = self.get_test_errata_object()
errata_unit = Unit(TYPE_ID_ERRATA, {"id": errata_obj["id"]}, errata_obj, None)
existing_units = [errata_unit]
test_repo = profiler_mocks.get_repo(repo_id)
# create two RPM units that match what is in the erratum
rpm_units = []
rpm_unit_key_1 = self.create_profile_entry("emoticons", 0, "0.1", "2", "x86_64",
"Test Vendor")
rpm_units.append(Unit(TYPE_ID_RPM, rpm_unit_key_1, {}, None))
rpm_unit_key_2 = self.create_profile_entry("patb", 0, "0.1", "2", "x86_64", "Test Vendor")
rpm_units.append(Unit(TYPE_ID_RPM, rpm_unit_key_2, {}, None))
existing_units += rpm_units
conduit = profiler_mocks.get_profiler_conduit(existing_units=existing_units,
repo_bindings=[test_repo],
repo_units=rpm_units)
example_errata = {"unit_key": errata_unit.unit_key, "type_id": TYPE_ID_ERRATA}
prof = YumProfiler()
translated_units = prof.install_units(self.test_consumer, [example_errata], None, None,
conduit)
# check repo_id passed to the conduit get_units()
self.assertEqual(conduit.get_units.call_args[0][0].id, repo_id)
# check unit association criteria passed to the conduit get_units()
self.assertEqual(conduit.get_units.call_args_list[0][0][1].type_ids, [TYPE_ID_ERRATA])
self.assertEqual(conduit.get_units.call_args_list[0][0][1].unit_filters,
errata_unit.unit_key)
# validate translated units
self.assertEqual(len(translated_units), 2)
expected = prof._get_rpms_from_errata(errata_unit)
for u in translated_units:
rpm_unit_key = u["unit_key"]
self.assertTrue(rpm_unit_key in expected)
def test_unit_applicable(self):
# Errata refers to RPMs which ARE part of our test consumer's profile,
# AND in the repo.
errata_obj = self.get_test_errata_object()
errata_unit = Unit(TYPE_ID_ERRATA, {"id": errata_obj["id"]}, errata_obj, None)
errata_unit.id = "an_errata"
rpm_unit_key = self.create_profile_entry("emoticons", 0, "0.1", "2", "x86_64", "Test Vendor")
rpm_unit = Unit(TYPE_ID_RPM, rpm_unit_key, {}, None)
# Let's give it an id, so we can assert for it later
rpm_unit.id = "a_test_id"
test_repo = profiler_mocks.get_repo("test_repo_id")
prof = YumProfiler()
errata_rpms = prof._get_rpms_from_errata(errata_unit)
conduit = profiler_mocks.get_profiler_conduit(
repo_units=[errata_unit, rpm_unit], repo_bindings=[test_repo], errata_rpms=errata_rpms
)
unit_profile = self.test_consumer.profiles[TYPE_ID_RPM]
bound_repo_id = "test_repo_id"
report_list = prof.calculate_applicable_units(unit_profile, bound_repo_id, None, conduit)
self.assertEqual(report_list, {TYPE_ID_RPM: ["a_test_id"], TYPE_ID_ERRATA: ["an_errata"]})
def test_install_units_with_superseding_versions(self):
"""
Verify that errata installing multiple versions of a package installs the latest package
In this test, there are three errata to install. Two provide the same package, but one
errata has a higher version. The third errata installs two unrelated packages.
Only the most recent package versions should be installed.
"""
profiler = YumProfiler()
# "older" errata is associated with lower version rpm
errata_obj_old = self.get_test_errata_object(eid='grinder_test_2')
errata_old = Unit(TYPE_ID_ERRATA, {"errata_id": errata_obj_old["id"]},
errata_obj_old, None)
rpm_key_old = self.create_profile_entry("grinder_test_package", 0,
"2.0", "1.fc14", "noarch",
"Test Vendor")
# "newer" errata is associated with higher version rpm
errata_obj_new = self.get_test_errata_object(eid='grinder_test_3')
errata_new = Unit(TYPE_ID_ERRATA, {"errata_id": errata_obj_new["id"]},
errata_obj_new, None)
rpm_key_new = self.create_profile_entry("grinder_test_package", 0,
"3.0", "1.fc14", "noarch",
"Test Vendor")
# "unrelated" errata is a different package, a control
errata_obj_unr = self.get_test_errata_object(eid='RHEA-2010:9999')
errata_unr = Unit(TYPE_ID_ERRATA, {"errata_id": errata_obj_unr["id"]},
errata_obj_unr, None)
rpm_key_unr_1 = self.create_profile_entry("emoticons", 0, "0.1", "2",
"x86_64", "Test Vendor")
rpm_key_unr_2 = self.create_profile_entry("patb", 0, "0.1", "2",
"x86_64", "Test Vendor")
# units list for the conduit
rpm_keys = [rpm_key_old, rpm_key_new, rpm_key_unr_1, rpm_key_unr_2]
rpm_units = [
Unit(TYPE_ID_RPM, rpm_key, {}, None) for rpm_key in rpm_keys
]
existing_units = [errata_old, errata_new, errata_unr] + rpm_units
# Set the profile to indicate that the errata apply by including fake unit keys of
# installed packages with lower versions that need to be upgraded
rpm_keys_installed = [
self.create_profile_entry("grinder_test_package", 0, "1.0",
"1.fc14", "noarch", "Test Vendor"),
self.create_profile_entry("emoticons", 0, "0.0", "1", "x86_64",
"Test Vendor"),
self.create_profile_entry("patb", 0, "0.0", "2", "x86_64",
"Test Vendor")
]
profiles = {TYPE_ID_RPM: rpm_keys_installed}
# put together all the args for install_units and call it
consumer = Consumer(self.consumer_id, profiles)
errata = [
{
"unit_key": errata_old.unit_key,
"type_id": TYPE_ID_ERRATA
},
{
"unit_key": errata_new.unit_key,
"type_id": TYPE_ID_ERRATA
},
{
"unit_key": errata_unr.unit_key,
"type_id": TYPE_ID_ERRATA
},
]
test_repo = profiler_mocks.get_repo("test_repo_id")
conduit = profiler_mocks.get_profiler_conduit(
existing_units=existing_units,
repo_bindings=[test_repo],
repo_units=rpm_units)
translated_units = profiler.install_units(consumer, errata, None, None,
conduit)
# validate translated units:
# - the unrelated rpms should still be present
# - the rpm from the "newer" errata should still be present
# - the rpm from the "older" errata should have been removed
# - the total number of units present is 3
translated_filenames = [
u['unit_key']['filename'] for u in translated_units
]
self.assertTrue('emoticons-0.1-2.x86_64.rpm' in translated_filenames)
self.assertTrue('patb-0.1-2.x86_64.rpm' in translated_filenames)
self.assertTrue('grinder_test_package-3.0-1.fc14.noarch.rpm' in
translated_filenames)
self.assertTrue('grinder_test_package-2.0-1.fc14.noarch.rpm' not in
translated_filenames)
self.assertEqual(len(translated_units), 3)
def test_install_units_with_superseding_versions(self):
"""
Verify that errata installing multiple versions of a package installs the latest package
In this test, there are three errata to install. Two provide the same package, but one
errata has a higher version. The third errata installs two unrelated packages.
Only the most recent package versions should be installed.
"""
profiler = YumProfiler()
# "older" errata is associated with lower version rpm
errata_obj_old = self.get_test_errata_object(eid='grinder_test_2')
errata_old = Unit(TYPE_ID_ERRATA, {"errata_id": errata_obj_old["id"]}, errata_obj_old, None)
rpm_key_old = self.create_profile_entry(
"grinder_test_package", 0, "2.0", "1.fc14", "noarch", "Test Vendor")
# "newer" errata is associated with higher version rpm
errata_obj_new = self.get_test_errata_object(eid='grinder_test_3')
errata_new = Unit(TYPE_ID_ERRATA, {"errata_id": errata_obj_new["id"]}, errata_obj_new, None)
rpm_key_new = self.create_profile_entry(
"grinder_test_package", 0, "3.0", "1.fc14", "noarch", "Test Vendor")
# "unrelated" errata is a different package, a control
errata_obj_unr = self.get_test_errata_object(eid='RHEA-2010:9999')
errata_unr = Unit(TYPE_ID_ERRATA, {"errata_id": errata_obj_unr["id"]}, errata_obj_unr, None)
rpm_key_unr_1 = self.create_profile_entry(
"emoticons", 0, "0.1", "2", "x86_64", "Test Vendor")
rpm_key_unr_2 = self.create_profile_entry(
"patb", 0, "0.1", "2", "x86_64", "Test Vendor")
# units list for the conduit
rpm_keys = [rpm_key_old, rpm_key_new, rpm_key_unr_1, rpm_key_unr_2]
rpm_units = [Unit(TYPE_ID_RPM, rpm_key, {}, None) for rpm_key in rpm_keys]
existing_units = [errata_old, errata_new, errata_unr] + rpm_units
# Set the profile to indicate that the errata apply by including fake unit keys of
# installed packages with lower versions that need to be upgraded
rpm_keys_installed = [
self.create_profile_entry(
"grinder_test_package", 0, "1.0", "1.fc14", "noarch", "Test Vendor"),
self.create_profile_entry(
"emoticons", 0, "0.0", "1", "x86_64", "Test Vendor"),
self.create_profile_entry(
"patb", 0, "0.0", "2", "x86_64", "Test Vendor")
]
profiles = {TYPE_ID_RPM: rpm_keys_installed}
# put together all the args for install_units and call it
consumer = Consumer(self.consumer_id, profiles)
errata = [
{"unit_key": errata_old.unit_key, "type_id": TYPE_ID_ERRATA},
{"unit_key": errata_new.unit_key, "type_id": TYPE_ID_ERRATA},
{"unit_key": errata_unr.unit_key, "type_id": TYPE_ID_ERRATA},
]
test_repo = profiler_mocks.get_repo("test_repo_id")
conduit = profiler_mocks.get_profiler_conduit(existing_units=existing_units,
repo_bindings=[test_repo],
repo_units=rpm_units)
translated_units = profiler.install_units(consumer, errata, None, None, conduit)
# validate translated units:
# - the unrelated rpms should still be present
# - the rpm from the "newer" errata should still be present
# - the rpm from the "older" errata should have been removed
# - the total number of units present is 3
translated_filenames = [u['unit_key']['filename'] for u in translated_units]
self.assertTrue('emoticons-0.1-2.x86_64.rpm' in translated_filenames)
self.assertTrue('patb-0.1-2.x86_64.rpm' in translated_filenames)
self.assertTrue('grinder_test_package-3.0-1.fc14.noarch.rpm' in translated_filenames)
self.assertTrue('grinder_test_package-2.0-1.fc14.noarch.rpm' not in translated_filenames)
self.assertEqual(len(translated_units), 3)
def test_install_units_unit_not_in_repo(self):
"""
This tests that if an erratum unit is requested to be installed, we do
not attempt to install any RPM units that are not available in repos.
For example, if an erratum contains packages for RHEL6 and RHEL7, we do
not want to ask a RHEL6 consumer to install RHEL7 packages that are
unavailable on that host.
This is a related issue to errata applicability but is slightly
different since the API caller wants to install a particular erratum, and is
not trying to determine which errata are applicable.
Note also that RHEA-2014:9999 has emoticons-0.1 and patb-0.1 in
different package collections; this is atypical and would likely not be
seen in the wild. I set it up like this to ensure the package list from
the erratum was being flattened during comparisons.
More detail is available in https://pulp.plan.io/issues/770
"""
repo_id = "test_repo_id"
# this erratum has four RPMs but only two are available
errata_obj = self.get_test_errata_object(eid='RHEA-2014:9999')
errata_unit = Unit(TYPE_ID_ERRATA, {"errata_id": errata_obj["id"]},
errata_obj, None)
existing_units = [errata_unit]
test_repo = profiler_mocks.get_repo(repo_id)
# create two RPM units that match what is in the erratum. There are
# higher versioned RPMs in the erratum that are not available; these
# should not be installed.
rpm_units = []
rpm_unit_key_1 = self.create_profile_entry("emoticons", 0, "0.1", "2",
"x86_64", "Test Vendor")
rpm_units.append(Unit(TYPE_ID_RPM, rpm_unit_key_1, {}, None))
rpm_unit_key_2 = self.create_profile_entry("patb", 0, "0.1", "2",
"x86_64", "Test Vendor")
rpm_units.append(Unit(TYPE_ID_RPM, rpm_unit_key_2, {}, None))
existing_units += rpm_units
conduit = profiler_mocks.get_profiler_conduit(
existing_units=existing_units,
repo_bindings=[test_repo],
repo_units=rpm_units)
def mocked_get_units(repo_id, criteria=None):
"""
Override the default get_units in profiler_mocks.
This method is specific to this particular unit test. The default
get_units() in profiler_mocks only checks the criteria's type_id and not any
other fields.
:param repo_id: repo ID (unused)
:type repo_id: not used
:param criteria: unit association criteria
:type criteria: pulp.server.db.model.criteria.UnitAssociationCriteria
"""
if TYPE_ID_ERRATA in criteria.type_ids:
return [errata_unit]
elif criteria['unit_filters']['name'] == 'emoticons' and \
criteria['unit_filters']['version'] == '0.1':
return [rpm_units[0]]
elif criteria['unit_filters']['name'] == 'patb' and \
criteria['unit_filters']['version'] == '0.1':
return [rpm_units[1]]
else:
return []
conduit.get_units.side_effect = mocked_get_units
example_errata = {
"unit_key": errata_unit.unit_key,
"type_id": TYPE_ID_ERRATA
}
prof = YumProfiler()
translated_units = prof.install_units(self.test_consumer,
[example_errata], None, None,
conduit)
# check repo_id passed to the conduit get_units()
self.assertEqual(conduit.get_units.call_args_list[0][0][0].id, repo_id)
# validate translated units
self.assertEqual(len(translated_units), 2)
self.assertEqual(translated_units[0]['unit_key']['filename'],
'patb-0.1-2.x86_64.rpm')
self.assertEqual(translated_units[1]['unit_key']['filename'],
'emoticons-0.1-2.x86_64.rpm')
expected = prof._get_rpms_from_errata(errata_unit)
for u in translated_units:
rpm_unit_key = u["unit_key"]
self.assertTrue(rpm_unit_key in expected)
def test_install_units_unit_not_in_repo(self):
"""
This tests that if an erratum unit is requested to be installed, we do
not attempt to install any RPM units that are not available in repos.
For example, if an erratum contains packages for RHEL6 and RHEL7, we do
not want to ask a RHEL6 consumer to install RHEL7 packages that are
unavailable on that host.
This is a related issue to errata applicability but is slightly
different since the API caller wants to install a particular erratum, and is
not trying to determine which errata are applicable.
Note also that RHEA-2014:9999 has emoticons-0.1 and patb-0.1 in
different package collections; this is atypical and would likely not be
seen in the wild. I set it up like this to ensure the package list from
the erratum was being flattened during comparisons.
More detail is available in https://pulp.plan.io/issues/770
"""
repo_id = "test_repo_id"
# this erratum has four RPMs but only two are available
errata_obj = self.get_test_errata_object(eid='RHEA-2014:9999')
errata_unit = Unit(TYPE_ID_ERRATA, {"id": errata_obj["id"]}, errata_obj, None)
existing_units = [errata_unit]
test_repo = profiler_mocks.get_repo(repo_id)
# create two RPM units that match what is in the erratum. There are
# higher versioned RPMs in the erratum that are not available; these
# should not be installed.
rpm_units = []
rpm_unit_key_1 = self.create_profile_entry("emoticons", 0, "0.1", "2", "x86_64",
"Test Vendor")
rpm_units.append(Unit(TYPE_ID_RPM, rpm_unit_key_1, {}, None))
rpm_unit_key_2 = self.create_profile_entry("patb", 0, "0.1", "2", "x86_64", "Test Vendor")
rpm_units.append(Unit(TYPE_ID_RPM, rpm_unit_key_2, {}, None))
existing_units += rpm_units
conduit = profiler_mocks.get_profiler_conduit(existing_units=existing_units,
repo_bindings=[test_repo],
repo_units=rpm_units)
def mocked_get_units(repo_id, criteria=None):
"""
Override the default get_units in profiler_mocks.
This method is specific to this particular unit test. The default
get_units() in profiler_mocks only checks the criteria's type_id and not any
other fields.
:param repo_id: repo ID (unused)
:type repo_id: not used
:param criteria: unit association criteria
:type criteria: pulp.server.db.model.criteria.UnitAssociationCriteria
"""
if TYPE_ID_ERRATA in criteria.type_ids:
return [errata_unit]
elif criteria['unit_filters']['name'] == 'emoticons' and \
criteria['unit_filters']['version'] == '0.1':
return [rpm_units[0]]
elif criteria['unit_filters']['name'] == 'patb' and \
criteria['unit_filters']['version'] == '0.1':
return [rpm_units[1]]
else:
return []
conduit.get_units.side_effect = mocked_get_units
example_errata = {"unit_key": errata_unit.unit_key, "type_id": TYPE_ID_ERRATA}
prof = YumProfiler()
translated_units = prof.install_units(self.test_consumer, [example_errata], None, None,
conduit)
# check repo_id passed to the conduit get_units()
self.assertEqual(conduit.get_units.call_args_list[0][0][0].id, repo_id)
# validate translated units
self.assertEqual(len(translated_units), 2)
self.assertEqual(translated_units[0]['unit_key']['filename'], 'patb-0.1-2.x86_64.rpm')
self.assertEqual(translated_units[1]['unit_key']['filename'], 'emoticons-0.1-2.x86_64.rpm')
expected = prof._get_rpms_from_errata(errata_unit)
for u in translated_units:
rpm_unit_key = u["unit_key"]
self.assertTrue(rpm_unit_key in expected)
