def test_reboot_setting(self, reboot_setting_in_api='Never', reboot_setting_in_code=Constants.REBOOT_NEVER):
argument_composer = ArgumentComposer()
argument_composer.reboot_setting = reboot_setting_in_api
runtime = RuntimeCompositor(argument_composer.get_composed_arguments(), True, Constants.YUM)
reboot_manager = runtime.reboot_manager
self.assertEqual(reboot_manager.is_setting(reboot_setting_in_code), True)
runtime.stop()
def test_reboot_setting_default_config(self):
argument_composer = ArgumentComposer()
argument_composer.reboot_setting = ""
runtime = RuntimeCompositor(argument_composer.get_composed_arguments(), True, Constants.YUM)
reboot_manager = runtime.reboot_manager
self.assertEqual(reboot_manager.is_setting(Constants.REBOOT_IF_REQUIRED), True)
runtime.stop()
def test_reboot_always_runs_only_once_if_no_reboot_is_required(self):
reboot_setting_in_api = 'Always'
argument_composer = ArgumentComposer()
argument_composer.reboot_setting = reboot_setting_in_api
runtime = RuntimeCompositor(argument_composer.get_composed_arguments(),
True, Constants.YUM)
reboot_manager = runtime.reboot_manager
# Validate single reboot scenario
runtime.status_handler.is_reboot_pending = True
self.assertEqual(
reboot_manager.start_reboot_if_required_and_time_available(20),
True)
# mock completing the reboot once, with no reboot required
runtime.status_handler.set_installation_reboot_status(
Constants.RebootStatus.REQUIRED)
runtime.status_handler.set_installation_reboot_status(
Constants.RebootStatus.STARTED)
runtime.status_handler.is_reboot_pending = False
runtime.status_handler.set_installation_reboot_status(
Constants.RebootStatus.COMPLETED)
# no further reboot should be required
self.assertEqual(
reboot_manager.start_reboot_if_required_and_time_available(20),
False)
runtime.stop()
def test_reboot_always_time_not_available(self):
reboot_setting_in_api = 'Always'
argument_composer = ArgumentComposer()
argument_composer.reboot_setting = reboot_setting_in_api
runtime = RuntimeCompositor(argument_composer.get_composed_arguments(), True, Constants.YUM)
reboot_manager = runtime.reboot_manager
self.assertEqual(reboot_manager.start_reboot_if_required_and_time_available(10), False)
runtime.stop()