def setUp(self):
self.db_dir = tempfile.mkdtemp()
productid.DatabaseDirectory.PATH = self.db_dir
self.pm = productid.ProductManager()
cert1 = stubs.StubEntitlementCertificate(
stubs.StubProduct('product1'),
start_date=datetime.datetime(2010, 1, 1),
end_date=datetime.datetime(2050, 1, 1))
cert2 = stubs.StubEntitlementCertificate(
stubs.StubProduct('product2'),
start_date=datetime.datetime(2010, 1, 1),
end_date=datetime.datetime(2060, 1, 1))
self.pm.pdir = stubs.StubProductDirectory([cert1, cert2])
sys.stdout = stubs.MockStdout()
self.yb = YumBase()
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_get_releases_rhel_no_content(self):
stub_content_5 = stubs.StubContent("c5", required_tags="AwesomeOS",
gpg=None, enabled="1")
stub_product = stubs.StubProduct("rhel-6")
stub_entitlement_certs = [stubs.StubEntitlementCertificate(stub_product,
content=[stub_content_5])]
self.ent_dir = stubs.StubEntitlementDirectory(stub_entitlement_certs)
rb = self._get_rb()
releases = rb.get_releases()
self.assertEquals([], releases)
def test_get_releases_rhel_no_enabled_content(self):
stub_content_6 = stubs.StubContent("c6",
required_tags="rhel-6",
gpg=None,
enabled="0")
stub_product = stubs.StubProduct("rhel-6")
stub_entitlement_certs = [
stubs.StubEntitlementCertificate(stub_product,
content=[stub_content_6])
]
self.ent_dir = stubs.StubEntitlementDirectory(stub_entitlement_certs)
cdn_rv_provider = self._get_cdn_rv_provider()
releases = cdn_rv_provider.get_releases()
self.assertEquals([], releases)
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)
def test_autobind_all_products_covered(self):
self.stub_backend.entitlement_dir = stubs.StubEntitlementDirectory(
[stubs.StubEntitlementCertificate(self.stub_product)])
autobind_controller = self._get_autobind_controller()
self.assertRaises(autobind.AllProductsCoveredException,
autobind_controller.load)
