def setUp(self):
super(TestMoveKeysHook, self).setUp()
conf = self.fake_config
conf.from_version = '6.0'
self.hook = MoveKeysHook(self.upgraders, conf)
class TestMoveKeysHook(TestPreUpgradeHooksBase):
def setUp(self):
super(TestMoveKeysHook, self).setUp()
conf = self.fake_config
conf.from_version = '6.0'
self.hook = MoveKeysHook(self.upgraders, conf)
def test_is_required_returns_true(self):
self.hook.config.from_version = '6.0'
self.assertTrue(self.hook.check_if_required())
self.hook.config.from_version = '6.0.1'
self.assertTrue(self.hook.check_if_required())
def test_is_required_returns_false(self):
self.hook.config.from_version = '6.1'
self.assertFalse(self.hook.check_if_required())
self.hook.config.from_version = '6.2'
self.assertFalse(self.hook.check_if_required())
@mock.patch(
'fuel_upgrade.pre_upgrade_hooks.from_6_0_to_any_copy_keys.utils.'
'file_exists', return_value=True)
@mock.patch(
'fuel_upgrade.pre_upgrade_hooks.from_6_0_to_any_copy_keys.utils.'
'exec_cmd')
def test_run(self, cmd_exec, f_exist):
self.hook.run()
f_exist.assert_called_once_with(self.hook.dst_path)
cmd_exec.assert_has_calls(
[mock.call('docker cp fuel-core-6.0-astute:/var/lib/astute/ '
'/var/lib/fuel/keys/'),
mock.call('mv /var/lib/fuel/keys/astute/* /var/lib/fuel/keys/'),
mock.call('rm -r /var/lib/fuel/keys/astute/')])
