diff options
Diffstat (limited to 'tests/jenkins_manager/test_manager.py')
-rw-r--r-- | tests/jenkins_manager/test_manager.py | 55 |
1 files changed, 27 insertions, 28 deletions
diff --git a/tests/jenkins_manager/test_manager.py b/tests/jenkins_manager/test_manager.py index ba579d15..530a8eaf 100644 --- a/tests/jenkins_manager/test_manager.py +++ b/tests/jenkins_manager/test_manager.py @@ -21,66 +21,65 @@ from tests.base import mock _plugins_info = {} -_plugins_info['plugin1'] = {'longName': '', - 'shortName': '', - 'version': ''} +_plugins_info["plugin1"] = {"longName": "", "shortName": "", "version": ""} -@mock.patch('jenkins_jobs.builder.JobCache', mock.MagicMock) +@mock.patch("jenkins_jobs.builder.JobCache", mock.MagicMock) class TestCaseTestJenkinsManager(base.BaseTestCase): - def setUp(self): super(TestCaseTestJenkinsManager, self).setUp() self.jjb_config = JJBConfig() self.jjb_config.validate() def test_plugins_list(self): - self.jjb_config.builder['plugins_info'] = _plugins_info + self.jjb_config.builder["plugins_info"] = _plugins_info self.builder = jenkins_jobs.builder.JenkinsManager(self.jjb_config) self.assertEqual(self.builder.plugins_list, _plugins_info) - @mock.patch.object(jenkins_jobs.builder.jenkins.Jenkins, - 'get_plugins', - return_value=_plugins_info) + @mock.patch.object( + jenkins_jobs.builder.jenkins.Jenkins, "get_plugins", return_value=_plugins_info + ) def test_plugins_list_from_jenkins(self, jenkins_mock): # Trigger fetching the plugins from jenkins when accessing the property - self.jjb_config.builder['plugins_info'] = {} + self.jjb_config.builder["plugins_info"] = {} self.builder = jenkins_jobs.builder.JenkinsManager(self.jjb_config) # See https://github.com/formiaczek/multi_key_dict/issues/17 # self.assertEqual(self.builder.plugins_list, k) for key_tuple in self.builder.plugins_list.keys(): for key in key_tuple: - self.assertEqual(self.builder.plugins_list[key], - _plugins_info[key]) + self.assertEqual(self.builder.plugins_list[key], _plugins_info[key]) def test_delete_managed(self): - self.jjb_config.builder['plugins_info'] = {} + self.jjb_config.builder["plugins_info"] = {} self.builder = jenkins_jobs.builder.JenkinsManager(self.jjb_config) - with mock.patch.multiple('jenkins_jobs.builder.JenkinsManager', - get_jobs=mock.DEFAULT, - is_job=mock.DEFAULT, - is_managed=mock.DEFAULT, - delete_job=mock.DEFAULT) as patches: - patches['get_jobs'].return_value = [{'fullname': 'job1'}, - {'fullname': 'job2'}] - patches['is_managed'].side_effect = [True, True] - patches['is_job'].side_effect = [True, True] + with mock.patch.multiple( + "jenkins_jobs.builder.JenkinsManager", + get_jobs=mock.DEFAULT, + is_job=mock.DEFAULT, + is_managed=mock.DEFAULT, + delete_job=mock.DEFAULT, + ) as patches: + patches["get_jobs"].return_value = [ + {"fullname": "job1"}, + {"fullname": "job2"}, + ] + patches["is_managed"].side_effect = [True, True] + patches["is_job"].side_effect = [True, True] self.builder.delete_old_managed() - self.assertEqual(patches['delete_job'].call_count, 2) + self.assertEqual(patches["delete_job"].call_count, 2) def _get_plugins_info_error_test(self, error_string): builder = jenkins_jobs.builder.JenkinsManager(self.jjb_config) exception = jenkins_jobs.builder.jenkins.JenkinsException(error_string) - with mock.patch.object(builder.jenkins, 'get_plugins', - side_effect=exception): + with mock.patch.object(builder.jenkins, "get_plugins", side_effect=exception): plugins_info = builder.get_plugins_info() - self.assertEqual([_plugins_info['plugin1']], plugins_info) + self.assertEqual([_plugins_info["plugin1"]], plugins_info) def test_get_plugins_info_handles_connectionrefused_errors(self): - self._get_plugins_info_error_test('Connection refused') + self._get_plugins_info_error_test("Connection refused") def test_get_plugins_info_handles_forbidden_errors(self): - self._get_plugins_info_error_test('Forbidden') + self._get_plugins_info_error_test("Forbidden") |