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)