Merge "common/test_sender: Update test for perform_request"

This commit is contained in:
Jenkins 2017-09-07 11:12:41 +00:00 committed by Gerrit Code Review
commit 46ea48fb44

View File

@ -49,19 +49,37 @@ class TestSender(unittest.TestCase):
get.assert_called_with("my-url", params="some payload", get.assert_called_with("my-url", params="some payload",
headers=mock.ANY, timeout=mock.ANY) headers=mock.ANY, timeout=mock.ANY)
@mock.patch.object(requests, 'post')
@mock.patch.object(requests, 'put')
def test_perform_request_req_type_put(self, put, post):
"""Test the behaviour when performing a post request
Set-up: None
Test: call _perform_request with req_type="put"
Expected behaviour:
* requests.put is called
* requests.post is not called (i.e. no fall back to default)
"""
self.sender._perform_request("my-url", "some payload",
"some headers", req_type="put")
post.assert_not_called()
put.assert_called_with("my-url", data="some payload",
headers=mock.ANY, timeout=mock.ANY)
@mock.patch.object(requests, 'post') @mock.patch.object(requests, 'post')
@mock.patch.object(requests, 'get') @mock.patch.object(requests, 'get')
def test_perform_request_req_type_post(self, get, post): def test_perform_request_req_type_post(self, get, post):
"""Test the behaviour when performing a post request """Test the behaviour when performing a post request
Set-up: None Set-up: None
Test: call _perform_request with req_type="post" Test: call _perform_request with no req_type
Expected behaviour: Expected behaviour:
* requests.get is not called * requests.get is not called
* requests.post is called with appropriate params * requests.post is called with appropriate params
""" """
self.sender._perform_request("my-url", "some payload", self.sender._perform_request("my-url", "some payload",
"some headers", req_type="post") "some headers")
get.assert_not_called() get.assert_not_called()
post.assert_called_with("my-url", data="some payload", post.assert_called_with("my-url", data="some payload",