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)