def create_payload_test(self):
"""Test PayloadFactory create method."""
self.assertIsInstance(PayloadFactory.create(PayloadType.DNF),
DNFModule)
self.assertIsInstance(PayloadFactory.create(PayloadType.LIVE_IMAGE),
LiveImageModule)
self.assertIsInstance(PayloadFactory.create(PayloadType.LIVE_OS),
LiveOSModule)
def create_payload_test(self):
"""Test PayloadFactory create method."""
payload = PayloadFactory.create(PayloadType.DNF)
self.assertIsInstance(payload, DNFModule)
self.assertEqual(payload.type, PayloadType.DNF)
payload = PayloadFactory.create(PayloadType.LIVE_IMAGE)
self.assertIsInstance(payload, LiveImageModule)
self.assertEqual(payload.type, PayloadType.LIVE_IMAGE)
payload = PayloadFactory.create(PayloadType.LIVE_OS)
self.assertIsInstance(payload, LiveOSModule)
self.assertEqual(payload.type, PayloadType.LIVE_OS)
def create_payload(self, payload_type):
"""Create payload based on the passed type.
:param payload_type: type of the desirable payload
:type payload_type: value of the payload.base.constants.PayloadType enum
"""
payload = PayloadFactory.create(payload_type)
self.set_payload(payload)
return payload