def test_run_backup_for_all_engines(self):
upgrader = UpgradeManager(**self.default_args(
upgraders=[mock.Mock(), mock.Mock()], ))
upgrader.run()
self.called_once(upgrader._upgraders[0].backup)
self.called_once(upgrader._upgraders[1].backup)
def test_upgrade_run_on_success_methods(self):
engines = [mock.Mock(), mock.Mock(), mock.Mock()]
upgrader = UpgradeManager(**self.default_args(upgraders=engines))
upgrader.run()
for engine in engines:
self.called_once(engine.on_success)
def run_upgrade(args):
"""Run upgrade on master node
:param args: argparse object
"""
# Get admin password
if not args.password:
args.password = getpass.getpass('Admin Password: '******'Configuration data: %s', config)
# Initialize upgrade engines
upgraders_to_use = [
SUPPORTED_SYSTEMS[system](config)
for system in args.systems]
# Initialize checkers
if not args.no_checker:
checker_manager = CheckerManager(upgraders_to_use, config)
checker_manager.check()
# Initialize pre upgrade hook manager
hook_manager = PreUpgradeHookManager(upgraders_to_use, config)
hook_manager.run()
# Initialize upgrade manager with engines and checkers
upgrade_manager = UpgradeManager(
upgraders_to_use, config, args.no_rollback)
upgrade_manager.run()
def run_upgrade(args):
"""Run upgrade on master node
:param args: argparse object
"""
# Get admin password
if not args.password:
args.password = getpass.getpass('Admin Password: '******'Configuration data: %s', config)
# Initialize upgrade engines
upgraders_to_use = [
SUPPORTED_SYSTEMS[system](config) for system in args.systems
]
# Initialize checkers
if not args.no_checker:
checker_manager = CheckerManager(upgraders_to_use, config)
checker_manager.check()
# Initialize pre upgrade hook manager
hook_manager = PreUpgradeHookManager(upgraders_to_use, config)
hook_manager.run()
# Initialize upgrade manager with engines and checkers
upgrade_manager = UpgradeManager(upgraders_to_use, args.no_rollback)
upgrade_manager.run()
def test_upgrade_run_on_success_methods(self):
engines = [mock.Mock(), mock.Mock(), mock.Mock()]
upgrader = UpgradeManager(**self.default_args(upgraders=engines))
upgrader.run()
for engine in engines:
self.called_once(engine.on_success)
def test_upgrade_succed(self):
upgrader = UpgradeManager(**self.default_args())
engine_mock = upgrader._upgraders[0]
upgrader.run()
engine_mock.upgrade.assert_called_once_with()
self.method_was_not_called(engine_mock.rollback)
def test_upgrade_succed(self):
upgrader = UpgradeManager(**self.default_args())
engine_mock = upgrader._upgraders[0]
upgrader.run()
engine_mock.upgrade.assert_called_once_with()
self.method_was_not_called(engine_mock.rollback)
def run_upgrade(args):
"""Run upgrade on master node
:param args: argparse object
"""
# Initialize config
config = build_config(args.src)
logger.debug('Configuration data: {0}'.format(config))
# Initialize upgrade engines
upgraders_to_use = [
SUPPORTED_SYSTEMS[system](config) for system in args.systems
]
# Initialize checkers
if not args.no_checker:
checker_manager = CheckerManager(upgraders_to_use, config)
checker_manager.check()
# Initialize pre upgrade hook manager
hook_manager = PreUpgradeHookManager(upgraders_to_use, config)
hook_manager.run()
# Initialize upgrade manager with engines and checkers
upgrade_manager = UpgradeManager(upgraders_to_use, args.no_rollback)
upgrade_manager.run()
def run_upgrade(args):
"""Run upgrade on master node
:param args: argparse object
"""
# Initialize config
config = build_config(args.src)
logger.debug('Configuration data: {0}'.format(config))
# Initialize upgrade engines
upgraders_to_use = [
SUPPORTED_SYSTEMS[system](config)
for system in args.systems]
# Initialize checkers
if not args.no_checker:
checker_manager = CheckerManager(upgraders_to_use, config)
checker_manager.check()
# Initialize pre upgrade hook manager
hook_manager = PreUpgradeHookManager(upgraders_to_use, config)
hook_manager.run()
# Initialize upgrade manager with engines and checkers
upgrade_manager = UpgradeManager(upgraders_to_use, args.no_rollback)
upgrade_manager.run()
def test_run_backup_for_all_engines(self):
upgrader = UpgradeManager(**self.default_args(
upgraders=[mock.Mock(), mock.Mock()],
))
upgrader.run()
self.called_once(upgrader._upgraders[0].backup)
self.called_once(upgrader._upgraders[1].backup)
def test_upgrade_does_not_fail_if_on_success_method_raise_error(self):
error_engine = mock.Mock()
error_engine.on_success.side_effect = Exception('error')
engines = [mock.Mock(), error_engine, mock.Mock()]
upgrader = UpgradeManager(**self.default_args(upgraders=engines))
upgrader.run()
for engine in engines:
self.called_once(engine.on_success)
def test_upgrade_does_not_fail_if_on_success_method_raise_error(self):
error_engine = mock.Mock()
error_engine.on_success.side_effect = Exception('error')
engines = [mock.Mock(), error_engine, mock.Mock()]
upgrader = UpgradeManager(**self.default_args(upgraders=engines))
upgrader.run()
for engine in engines:
self.called_once(engine.on_success)
def test_run_upgrade_for_all_engines(self):
upgrader = UpgradeManager(**self.default_args(
upgraders=[mock.Mock(), mock.Mock()], ))
upgrader.run()
self.called_once(upgrader._upgraders[0].upgrade)
self.method_was_not_called(upgrader._upgraders[0].rollback)
self.called_once(upgrader._upgraders[1].upgrade)
self.method_was_not_called(upgrader._upgraders[1].rollback)
def test_run_upgrade_for_all_engines(self):
upgrader = UpgradeManager(**self.default_args(
upgraders=[mock.Mock(), mock.Mock()],
))
upgrader.run()
self.called_once(upgrader._upgraders[0].upgrade)
self.method_was_not_called(upgrader._upgraders[0].rollback)
self.called_once(upgrader._upgraders[1].upgrade)
self.method_was_not_called(upgrader._upgraders[1].rollback)
def test_upgrade_succed(self):
upgrader = UpgradeManager(**self.default_args())
engine_mock = upgrader._upgraders[0]
upgrader.run()
engine_mock.upgrade.assert_called_once_with()
self.method_was_not_called(engine_mock.rollback)
self.called_once(self.version_mock.save_current)
self.called_once(self.version_mock.switch_to_new)
self.method_was_not_called(self.version_mock.switch_to_previous)
def test_upgrade_succed(self):
upgrader = UpgradeManager(**self.default_args())
engine_mock = upgrader._upgraders[0]
upgrader.run()
engine_mock.upgrade.assert_called_once_with()
self.method_was_not_called(engine_mock.rollback)
self.called_once(self.version_mock.save_current)
self.called_once(self.version_mock.switch_to_new)
self.method_was_not_called(self.version_mock.switch_to_previous)
def test_upgrade_does_not_fail_if_on_success_method_raise_error(self):
upgrader = UpgradeManager(**self.default_args())
upgrader._on_success = mock.Mock()
upgrader._on_success.side_effect = Exception('error')
upgrader.run()
def test_before_upgrade_is_called(self):
upgrader = UpgradeManager(**self.default_args())
upgrader.before_upgrade = mock.Mock()
upgrader.run()
self.called_once(upgrader.before_upgrade)
def test_upgrade_run_on_success_methods(self):
upgrader = UpgradeManager(**self.default_args())
upgrader._on_success = mock.Mock()
upgrader.run()
self.called_once(upgrader._on_success)
def test_upgrade_does_not_fail_if_on_success_method_raise_error(self):
upgrader = UpgradeManager(**self.default_args())
upgrader._on_success = mock.Mock()
upgrader._on_success.side_effect = Exception('error')
upgrader.run()
def test_upgrade_run_on_success_methods(self):
upgrader = UpgradeManager(**self.default_args())
upgrader._on_success = mock.Mock()
upgrader.run()
self.called_once(upgrader._on_success)
