34cca0c4d9
Convert to the requests library to allow for more sophisticated response handling to be added. Want to allow for applications to override the response handling in certain cases where the default is incorrect. Handling of urlopen responses results in version specific handling to ensure correct behaviour across multiple versions of python. Changing to use the requests package, provides a higher level interface and removes some of the version specific handling for exceptions. Change-Id: I5369a0d35be4bf8b3b197a51e60aba21b5742cc7 Depends-On: Iabd70aa457ceb4dbc147d7cbaeec913148cb3b56
60 lines
1.9 KiB
Python
60 lines
1.9 KiB
Python
import json
|
|
from mock import patch
|
|
|
|
import jenkins
|
|
from tests.base import JenkinsTestBase
|
|
|
|
|
|
class JenkinsQuietDownTest(JenkinsTestBase):
|
|
@patch.object(jenkins.Jenkins, 'jenkins_open')
|
|
def test_success(self, jenkins_mock):
|
|
job_info_to_return = {
|
|
"quietingDown": True,
|
|
}
|
|
jenkins_mock.return_value = json.dumps(job_info_to_return)
|
|
|
|
self.j.quiet_down()
|
|
|
|
self.assertEqual(
|
|
jenkins_mock.call_args_list[0][0][0].url,
|
|
self.make_url('quietDown'))
|
|
self.assertEqual(
|
|
jenkins_mock.call_args_list[1][0][0].url,
|
|
self.make_url('api/json'))
|
|
self._check_requests(jenkins_mock.call_args_list)
|
|
|
|
@patch.object(jenkins.Jenkins, 'jenkins_open')
|
|
def test_fail(self, jenkins_mock):
|
|
job_info_to_return = {
|
|
"quietingDown": False,
|
|
}
|
|
jenkins_mock.return_value = json.dumps(job_info_to_return)
|
|
|
|
with self.assertRaises(jenkins.JenkinsException) as context_manager:
|
|
self.j.quiet_down()
|
|
|
|
self.assertEqual(
|
|
jenkins_mock.call_args_list[0][0][0].url,
|
|
self.make_url('quietDown'))
|
|
self.assertEqual(
|
|
jenkins_mock.call_args_list[1][0][0].url,
|
|
self.make_url('api/json'))
|
|
self.assertEqual(
|
|
str(context_manager.exception),
|
|
'quiet down failed')
|
|
self._check_requests(jenkins_mock.call_args_list)
|
|
|
|
@patch.object(jenkins.Jenkins, 'jenkins_open')
|
|
def test_http_fail(self, jenkins_mock):
|
|
jenkins_mock.side_effect = jenkins.JenkinsException(
|
|
'Error in request. Possibly authentication failed [401]: '
|
|
'basic auth failed')
|
|
|
|
with self.assertRaises(jenkins.JenkinsException):
|
|
self.j.quiet_down()
|
|
|
|
self.assertEqual(
|
|
jenkins_mock.call_args[0][0].url,
|
|
self.make_url('quietDown'))
|
|
self._check_requests(jenkins_mock.call_args_list)
|