def test_success_on_second_attempt(self, get_mock, _): response = http_get('url1', retries=2) self.assertIs(response, self._response_ok) self.assertEqual(get_mock.call_args_list, [ mock.call('url1'), mock.call('url1'), ])
def test_last_response_is_returned(self, get_mock, _): response_500 = mock.Mock(status_code=500, url='url1') get_mock.side_effect = [ self._response_error, self._response_error, response_500, ] response = http_get('url1', retries=3) self.assertIsNot(response, self._response_error) self.assertIs(response, response_500) self.assertEqual(get_mock.call_args_list, [ mock.call('url1'), mock.call('url1'), mock.call('url1'), ])
def test_success_on_first_attempt(self, get_mock, _): response = http_get('url1', retries=2) self.assertIs(response, self._response_ok) get_mock.assert_called_once_with('url1')