def _create_cert(self, product_id, label, version, provided_tags):
cert = stubs.StubProductCertificate(
stubs.StubProduct(product_id, label, version=version,
provided_tags=provided_tags))
cert.delete = Mock()
cert.write = Mock()
return cert
def test_workstation_skips_desktop(self):
desktop_cert = self._create_desktop_cert()
workstation_cert = self._create_workstation_cert()
self.prod_dir.certs.append(workstation_cert)
some_other_cert = stubs.StubProductCertificate(
stubs.StubProduct("8127", "Some Other Product"))
some_other_cert.delete = Mock()
some_other_cert.write = Mock()
self.prod_repo_map = {'71': ["repo2"],
'68': ["repo1"],
'8127': ["repo3"]}
self.prod_db_mock.find_repos = Mock(side_effect=self.find_repos_side_effect)
# Workstation comes first in this scenario:
enabled = [
(workstation_cert, 'repo2'),
(desktop_cert, 'repo1'),
(some_other_cert, 'repo3'),
]
self.prod_mgr.update_installed(enabled, ['repo1', 'repo2', 'repo3'])
self.assertFalse(workstation_cert.delete.called)
self.assertFalse(desktop_cert.write.called)
self.assertFalse(desktop_cert.delete.called)
# Testing a bug where desktop cert skipping ended the whole process:
self.assertTrue(some_other_cert.write.called)
self.assertFalse(some_other_cert.delete.called)
self.assertFalse(self.prod_db_mock.delete.called)
def test_get_releases_no_rhel(self):
self.prod_dir = stubs.StubProductDirectory(
[stubs.StubProductCertificate(
stubs.StubProduct("rhel-6-something",
provided_tags="rhel-6-something,rhel-6-stub"),)])
rb = self._get_rb()
releases = rb.get_releases()
self.assertEquals([], releases)
def setUp(self):
stub_content = stubs.StubContent("c1",
required_tags='rhel-6',
gpg=None,
enabled="1")
# this content should be ignored since it's not enabled
stub_content_2 = stubs.StubContent("c2",
required_tags='rhel-6',
gpg=None,
enabled="0")
# this should be ignored because of required_tag isn't what we
# are looking for
stub_content_3 = stubs.StubContent("c3",
required_tags="NotAwesomeOS",
gpg=None,
enabled="1")
# this should be ignored because of required_tag isn't what we
# are looking for, and it is not enabled
stub_content_4 = stubs.StubContent("c4",
required_tags="NotAwesomeOS",
gpg=None,
enabled="0")
stub_contents = [
stub_content, stub_content_2, stub_content_3, stub_content_4
]
stub_product = stubs.StubProduct("rhel-6")
stub_entitlement_certs = [
stubs.StubEntitlementCertificate(stub_product,
content=stub_contents)
]
stub_entitlement_dir = stubs.StubEntitlementDirectory(
stub_entitlement_certs)
stub_product_dir = stubs.StubProductDirectory([
stubs.StubProductCertificate(
stubs.StubProduct("rhel-6", provided_tags="rhel-6-stub"), )
])
def get_versions(dummy):
return versions
stub_content_connection = stubs.StubContentConnection()
stub_content_connection.get_versions = get_versions
self.rb = release.ReleaseBackend(
ent_dir=stub_entitlement_dir,
prod_dir=stub_product_dir,
content_connection=stub_content_connection)
def _getPrefDialog(self):
stub_backend = stubs.StubBackend()
stub_backend.cp_provider.consumer_auth_cp.setConsumer(CONSUMER_DATA)
stub_backend.product_dir = stubs.StubCertificateDirectory([stubs.StubProductCertificate(stubs.StubProduct("rhel-6"))])
stub_backend.entitlement_dir = stubs.StubEntitlementDirectory([stubs.StubEntitlementCertificate(stubs.StubProduct("rhel-6"))])
self.preferences_dialog = preferences.PreferencesDialog(backend=stub_backend,
parent=None)
self.preferences_dialog.release_backend.facts = stubs.StubFacts()
self.preferences_dialog.release_backend.get_releases = get_releases
self.preferences_dialog.async_updater = stubs.StubAsyncUpdater(self.preferences_dialog)
def setUp(self):
fixture.SubManFixture.setUp(self)
stub_content = stubs.StubContent("c1",
required_tags='rhel-6',
gpg=None,
enabled="1")
# this content should be ignored since it's not enabled
stub_content_2 = stubs.StubContent("c2",
required_tags='rhel-6',
gpg=None,
enabled="0")
# this should be ignored because of required_tag isn't what we
# are looking for
stub_content_3 = stubs.StubContent("c3",
required_tags="NotAwesomeOS",
gpg=None,
enabled="1")
# this should be ignored because of required_tag isn't what we
# are looking for, and it is not enabled
stub_content_4 = stubs.StubContent("c4",
required_tags="NotAwesomeOS",
gpg=None,
enabled="0")
stub_contents = [
stub_content, stub_content_2, stub_content_3, stub_content_4
]
stub_product = stubs.StubProduct("rhel-6")
stub_entitlement_certs = [
stubs.StubEntitlementCertificate(stub_product,
content=stub_contents)
]
# the fixtures ent_dir
self.ent_dir = stubs.StubEntitlementDirectory(stub_entitlement_certs)
# fixtures prod_dir
self.prod_dir = stubs.StubProductDirectory([
stubs.StubProductCertificate(
stubs.StubProduct("rhel-6",
provided_tags="rhel-6,rhel-6-stub"), )
])
# FIXME: should just mock this
# fixture knows to stub this for contentConnection
self._release_versions = versions
def test_autobind_load_provided_products(self):
self.stub_product = stubs.StubProduct("some_random_product")
self.stub_pool = modelhelpers.create_pool(
product_id=self.stub_product.getHash(),
product_name=self.stub_product.getName(),
provided_products=['rhel-6'])
self.stub_backend.uep.stub_pool = self.stub_pool
self.stub_installed_product = stubs.StubProduct("rhel-6")
self.stub_backend.product_dir = stubs.StubCertificateDirectory(
[stubs.StubProductCertificate(self.stub_installed_product)])
autobind_controller = self._get_autobind_controller()
autobind_controller.load()
self.assertTrue('Pro' in autobind_controller.suitable_slas)
def setUp(self):
self.stub_backend = stubs.StubBackend()
self.stub_consumer = stubs.StubConsumer()
self.stub_facts = stubs.StubFacts()
self.stub_backend.uep.getConsumer = self._getConsumerData
self.stub_product = stubs.StubProduct("rhel-6")
self.stub_pool = modelhelpers.create_pool(
product_id=self.stub_product.getHash(),
product_name=self.stub_product.getName())
self.stub_backend.uep.stub_pool = self.stub_pool
self.stub_backend.product_dir = stubs.StubCertificateDirectory(
[stubs.StubProductCertificate(self.stub_product)])
self.stub_backend.entitlement_dir = stubs.StubEntitlementDirectory([])
self.stub_backend.uep.dryRunBind = self._dryRunBind
def _getPrefDialog(self):
stub_backend = stubs.StubBackend()
stub_backend.uep.getConsumer = getConsumerData
if self._getConsumerData:
stub_backend.uep.getConsumer = self._getConsumerData
self.consumer = StubConsumer()
if self._getConsumer:
self.consumer = self._getConsumer()
stub_backend.product_dir = stubs.StubCertificateDirectory(
[stubs.StubProductCertificate(stubs.StubProduct("rhel-6"))])
stub_backend.entitlement_dir = stubs.StubEntitlementDirectory(
[stubs.StubEntitlementCertificate(stubs.StubProduct("rhel-6"))])
self.preferences_dialog = preferences.PreferencesDialog(
backend=stub_backend, consumer=self.consumer, parent=None)
self.preferences_dialog.release_backend.facts = stubs.StubFacts()
self.preferences_dialog.release_backend.get_releases = get_releases
def setUp(self):
SubManFixture.setUp(self)
# we have to have a reference to the patchers
#self.patcher2 = mock.patch.object(entcertlib.EntCertUpdateAction, '_get_consumer_id')
#self.entcertlib_updateaction_getconsumerid = self.patcher2.start()
self.patcher3 = mock.patch.object(repolib.RepoUpdateActionCommand,
'perform')
self.repolib_updateaction_perform = self.patcher3.start()
self.patcher6 = mock.patch(
'subscription_manager.managerlib.persist_consumer_cert')
self.managerlib_persist_consumer_cert = self.patcher6.start()
# mock out all hardware fetching... we may need to fake socket counts
self.hwprobe_getall_patcher = mock.patch.object(
hwprobe.Hardware, 'get_all')
self.hwprobe_getall_mock = self.hwprobe_getall_patcher.start()
self.hwprobe_getall_mock.return_value = {}
self.patcher_entcertlib_writer = mock.patch(
"subscription_manager.entcertlib.Writer")
self.entcertlib_writer = self.patcher_entcertlib_writer.start()
self.patcher_entcertlib_action_syslogreport = mock.patch.object(
entcertlib.EntCertUpdateAction, 'syslog_results')
self.update_action_syslog_mock = self.patcher_entcertlib_action_syslogreport.start(
)
# some stub certs
stub_product = stubs.StubProduct('stub_product')
self.stub_ent1 = stubs.StubEntitlementCertificate(stub_product)
self.stub_ent2 = stubs.StubEntitlementCertificate(stub_product)
self.stub_ent_expires_tomorrow = \
stubs.StubEntitlementCertificate(stub_product,
end_date=datetime.now() + timedelta(days=1))
self.stub_ent_expires_tomorrow_ent_dir = \
stubs.StubEntitlementDirectory([self.stub_ent_expires_tomorrow])
self.local_ent_certs = [self.stub_ent1, self.stub_ent2]
self.stub_entitled_proddir = \
stubs.StubProductDirectory([stubs.StubProductCertificate(stub_product)])
# local entitlement dir
self.stub_ent_dir = stubs.StubEntitlementDirectory(
self.local_ent_certs)
inj.provide(inj.ENT_DIR, self.stub_ent_dir)
self.mock_uep = mock.Mock()
self.mock_uep.getCertificateSerials = mock.Mock(
return_value=[{
'serial': self.stub_ent1.serial
}, {
'serial': self.stub_ent2.serial
}])
self.mock_uep.getConsumer = mock.Mock(return_value=CONSUMER_DATA)
self.set_consumer_auth_cp(self.mock_uep)
stub_release = {'releaseVer': '6.4'}
self.mock_uep.getRelease = mock.Mock(return_value=stub_release)
# we need to mock the consumers uuid with the mocked GoneExceptions
# uuid
self._inject_mock_valid_consumer(uuid="234234")
self.repolib_updateaction_perform.return_value = 0
# Setup a mock cert sorter to initiate the behaviour we want to test.
# Must use a non-callable mock for our features dep injection
# framework.
self.mock_cert_sorter = mock.NonCallableMock()
# TODO: need to provide return for "getRelease" for repolib stuff
injection.provide(injection.CERT_SORTER, self.mock_cert_sorter)
def setUp(self):
SubManFixture.setUp(self)
# we have to have a reference to the patchers
self.patcher2 = mock.patch.object(certlib.UpdateAction,
'_get_consumer_id')
self.certlib_updateaction_getconsumerid = self.patcher2.start()
self.patcher3 = mock.patch.object(repolib.UpdateAction, 'perform')
self.repolib_updateaction_perform = self.patcher3.start()
self.patcher4 = mock.patch(
'subscription_manager.factlib.ConsumerIdentity')
self.factlib_consumeridentity = self.patcher4.start()
self.patcher5 = mock.patch(
'subscription_manager.certlib.ConsumerIdentity')
self.certlib_consumeridentity = self.patcher5.start()
self.patcher6 = mock.patch(
'subscription_manager.managerlib.persist_consumer_cert')
self.managerlib_persist_consumer_cert = self.patcher6.start()
self.patcher8 = mock.patch.object(facts.Facts, 'get_last_update')
self.facts_getlastupdate = self.patcher8.start()
# mock out all hardware fetching... we may need to fake socket counts
self.hwprobe_getall_patcher = mock.patch.object(
hwprobe.Hardware, 'get_all')
self.hwprobe_getall_mock = self.hwprobe_getall_patcher.start()
self.hwprobe_getall_mock.return_value = {}
self.patcher_certlib_writer = mock.patch(
"subscription_manager.certlib.Writer")
self.certlib_writer = self.patcher_certlib_writer.start()
self.patcher_certlib_action_syslogreport = mock.patch.object(
certlib.UpdateAction, 'syslog_results')
self.update_action_syslog_mock = self.patcher_certlib_action_syslogreport.start(
)
# some stub certs
stub_product = stubs.StubProduct('stub_product')
self.stub_ent1 = stubs.StubEntitlementCertificate(stub_product)
self.stub_ent2 = stubs.StubEntitlementCertificate(stub_product)
self.stub_ent_expires_tomorrow = \
stubs.StubEntitlementCertificate(stub_product,
end_date=datetime.now() + timedelta(days=1))
self.stub_ent_expires_tomorrow_entdir = \
stubs.StubEntitlementDirectory([self.stub_ent_expires_tomorrow])
self.local_ent_certs = [self.stub_ent1, self.stub_ent2]
self.stub_entitled_proddir = \
stubs.StubProductDirectory([stubs.StubProductCertificate(stub_product)])
# local entitlement dir
self.stub_entdir = stubs.StubEntitlementDirectory(self.local_ent_certs)
inj.provide(inj.ENT_DIR, self.stub_entdir)
self.mock_uep = mock.Mock()
self.mock_uep.getCertificateSerials = mock.Mock(
return_value=[{
'serial': self.stub_ent1.serial
}, {
'serial': self.stub_ent2.serial
}])
self.mock_uep.getConsumer = mock.Mock(return_value=CONSUMER_DATA)
self.certlib_updateaction_getconsumerid.return_value = "234234"
self.repolib_updateaction_perform.return_value = 0
self.facts_getlastupdate.return_value = None
self.factlib_consumeridentity.read.return_value = stubs.StubConsumerIdentity(
"sdfsdf", "sdfsdf")
self.certlib_consumeridentity.read.return_value = stubs.StubConsumerIdentity(
"sdfsdf", "sdfsdf")
# Setup a mock cert sorter to initiate the behaviour we want to test.
# Must use a non-callable mock for our features dep injection
# framework.
self.mock_cert_sorter = mock.NonCallableMock()
injection.provide(injection.CERT_SORTER, self.mock_cert_sorter)
