class EntitlementRegenerationTests(unittest.TestCase):
def setUp(self):
self.cp = UEPConnection(username="******", password="******", insecure=True)
self.consumer = self.cp.registerConsumer("test-consumer", "system", owner="admin")
self.consumer_uuid = self.consumer['uuid']
# This product is present in the Candlepin test data
self.cp.bindByProduct(self.consumer_uuid, ["awesomeos-instancebased"])
entitlements = self.cp.getEntitlementList(self.consumer_uuid)
self.assertTrue(len(entitlements) > 0)
self.entitlement = entitlements[0]
self.entitlement_id = self.entitlement['id']
def test_regenerate_entitlements_default(self):
result = self.cp.regenEntitlementCertificates(self.consumer_uuid)
self.assertTrue(result)
def test_regenerate_entitlements_lazy(self):
result = self.cp.regenEntitlementCertificates(self.consumer_uuid, True)
self.assertTrue(result)
def test_regenerate_entitlements_eager(self):
result = self.cp.regenEntitlementCertificates(self.consumer_uuid, False)
self.assertTrue(result)
def test_regenerate_entitlements_bad_uuid(self):
result = self.cp.regenEntitlementCertificates("bad_consumer_uuid")
self.assertFalse(result)
def test_regenerate_entitlement_default(self):
result = self.cp.regenEntitlementCertificate(self.consumer_uuid, self.entitlement_id)
self.assertTrue(result)
def test_regenerate_entitlement_lazy(self):
result = self.cp.regenEntitlementCertificate(self.consumer_uuid, self.entitlement_id, True)
self.assertTrue(result)
def test_regenerate_entitlement_eager(self):
result = self.cp.regenEntitlementCertificate(self.consumer_uuid, self.entitlement_id, False)
self.assertTrue(result)
def test_regenerate_entitlement_bad_consumer_uuid(self):
result = self.cp.regenEntitlementCertificate("bad_consumer_uuid", self.entitlement_id)
self.assertFalse(result)
def test_regenerate_entitlement_bad_entitlement_id(self):
result = self.cp.regenEntitlementCertificate(self.consumer_uuid, "bad_entitlement_id")
self.assertFalse(result)
def tearDown(self):
self.cp.unregisterConsumer(self.consumer_uuid)
