common/test_sender: Update test for perform_request

* Add test_perform_request_req_type_put
* Updated test_perform_request_req_type_post
  * Now tests the default case

Change-Id: Iff7bc4d4370c5517e481db15ae1318f2e4fd34a5
This commit is contained in:
Emma Foley 2017-09-05 12:48:45 +00:00
parent d1ce958bee
commit a93c57e61f
1 changed files with 20 additions and 2 deletions

View File

@ -49,19 +49,37 @@ class TestSender(unittest.TestCase):
get.assert_called_with("my-url", params="some payload",
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, 'get')
def test_perform_request_req_type_post(self, get, post):
"""Test the behaviour when performing a post request
Set-up: None
Test: call _perform_request with req_type="post"
Test: call _perform_request with no req_type
Expected behaviour:
* requests.get is not called
* requests.post is called with appropriate params
"""
self.sender._perform_request("my-url", "some payload",
"some headers", req_type="post")
"some headers")
get.assert_not_called()
post.assert_called_with("my-url", data="some payload",