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)