def test_200_response(self): response = CoroutineMock() response.status = 200 response.text.return_value = "this is my return value" self.mock_aiohttp_get.side_effect = [response] retval = yield from self.http_get_request( "https://www.example.com") # NOQA expected = call('https://www.example.com', headers={'Content-type': 'application/json'}) self.assertEqual(self.mock_aiohttp_get.mock_calls, [expected]) self.assertEqual(response.close.mock_calls, []) self.assertEqual(retval, "this is my return value")
def test_200_response(self): response = CoroutineMock() response.status = 200 response.text.return_value = "this is my return value" self.mock_aiohttp_get.side_effect = [response] retval = yield from self.http_get_request("https://www.example.com") # NOQA expected = call('https://www.example.com', headers={ 'Content-type': 'application/json' }) self.assertEqual(self.mock_aiohttp_get.mock_calls, [expected]) self.assertEqual(response.close.mock_calls, []) self.assertEqual(retval, "this is my return value")
def test_non_200_response(self): response = CoroutineMock() response.status = 201 response.text.return_value = "this is my return value" self.mock_aiohttp_get.side_effect = [response] retval = yield from self.http_get_auth_request("sekrit", "https://www.example.com") # NOQA expected = call('https://www.example.com', headers={ 'Authorization': 'Token sekrit', 'Content-type': 'application/json' }, params={}) self.assertEqual(self.mock_aiohttp_get.mock_calls, [expected]) self.assertEqual(response.close.mock_calls, call()) self.assertEqual(retval, "")