def test_on_success(self, glob_mock, utils_mock):
upgrader = UpgradeManager(**self.default_args())
upgrader._on_success()
glob_mock.assert_called_once_with(self.fake_config.version_files_mask)
self.assertEqual(
utils_mock.remove.call_args_list,
[mock.call('file1'), mock.call('file2')])
templates_path = os.path.abspath(
os.path.join(os.path.dirname(__file__), '../templates'))
utils_mock.render_template_to_file.assert_has_calls([
mock.call(
'{0}/nailgun.repo'.format(templates_path),
'/etc/yum.repos.d/mos9999-updates.repo',
{
'name': 'mos9999-updates',
'baseurl': 'http://mirror.fuel-infra.org/mos-repos/'
'centos/mos9999-centos6-fuel/updates/x86_64/',
'gpgcheck': 0,
'skip_if_unavailable': 1,
}),
mock.call(
'{0}/nailgun.repo'.format(templates_path),
'/etc/yum.repos.d/mos9999-security.repo',
{
'name': 'mos9999-security',
'baseurl': 'http://mirror.fuel-infra.org/mos-repos/'
'centos/mos9999-centos6-fuel/security/x86_64/',
'gpgcheck': 0,
'skip_if_unavailable': 1,
}),
])
def test_on_success(self, glob_mock, utils_mock):
upgrader = UpgradeManager(**self.default_args())
upgrader._on_success()
glob_mock.assert_called_once_with(self.fake_config.version_files_mask)
self.assertEqual(
utils_mock.remove.call_args_list,
[mock.call('file1'), mock.call('file2')])
templates_path = os.path.abspath(
os.path.join(os.path.dirname(__file__), '../templates'))
utils_mock.render_template_to_file.assert_has_calls([
mock.call(
'{0}/nailgun.repo'.format(templates_path),
'/etc/yum.repos.d/mos9999-updates.repo', {
'name': 'mos9999-updates',
'baseurl': 'http://mirror.fuel-infra.org/mos-repos/'
'centos/mos9999-centos6-fuel/updates/x86_64/',
'gpgcheck': 0,
'skip_if_unavailable': 1,
}),
mock.call(
'{0}/nailgun.repo'.format(templates_path),
'/etc/yum.repos.d/mos9999-security.repo', {
'name': 'mos9999-security',
'baseurl': 'http://mirror.fuel-infra.org/mos-repos/'
'centos/mos9999-centos6-fuel/security/x86_64/',
'gpgcheck': 0,
'skip_if_unavailable': 1,
}),
])
def test_on_success(self, glob_mock, utils_mock):
upgrader = UpgradeManager(**self.default_args())
upgrader._on_success()
glob_mock.assert_called_once_with(self.fake_config.version_files_mask)
self.assertEqual(
utils_mock.remove.call_args_list,
[mock.call('file1'), mock.call('file2')])
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)
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)
