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_from_ks_test(self):
"""Test PayloadFactory create from KS method."""
# Live OS can't be detected from the KS data so it is not tested here
data = Mock()
data.liveimg.seen = True
data.packages.seen = False
self.assertIsInstance(PayloadFactory.create_from_ks_data(data), LiveImageModule)
data.liveimg.seen = False
data.packages.seen = True
self.assertIsInstance(PayloadFactory.create_from_ks_data(data), DNFModule)
data.liveimg.seen = False
data.packages.seen = False
self.assertIsNone(PayloadFactory.create_from_ks_data(data))
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
def process_kickstart(self, data):
"""Process the kickstart data."""
log.debug("Processing kickstart data...")
# create payload if no payload is set already
if not self.is_payload_set():
payload = PayloadFactory.create_from_ks_data(data)
if not payload:
log.warning(
"No payload was created. Kickstart data passed in are lost."
)
return
payload.process_kickstart(data)
self._initialize_payload(payload)