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)