def create_handler_test(self): """Test HandlerFactory create method.""" self.assertIsInstance(HandlerFactory.create(HandlerType.DNF), DNFHandlerModule) self.assertIsInstance(HandlerFactory.create(HandlerType.LIVE_IMAGE), LiveImageHandlerModule) self.assertIsInstance(HandlerFactory.create(HandlerType.LIVE_OS), LiveOSHandlerModule)
def create_handler(self, handler_type): """Create handler based on the passed type. :param handler_type: type of the desirable handler :type handler_type: value of the payload.base.constants.HandlerType enum """ handler = HandlerFactory.create(handler_type) self._initialize_handler(handler) return self._payload_handler_path
def create_handler_from_ks_test(self): """Test HandlerFactory 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(HandlerFactory.create_from_ks_data(data), LiveImageHandlerModule) data.liveimg.seen = False data.packages.seen = True self.assertIsInstance(HandlerFactory.create_from_ks_data(data), DNFHandlerModule) data.liveimg.seen = False data.packages.seen = False self.assertIsNone(HandlerFactory.create_from_ks_data(data))
def process_kickstart(self, data): """Process the kickstart data.""" log.debug("Processing kickstart data...") # create handler if no handler is set already if not self.is_handler_set(): handler = HandlerFactory.create_from_ks_data(data) if not handler: log.warning("No handler was created. Kickstart data passed in are lost.") return handler.process_kickstart(data) self._initialize_handler(handler)