def test_decorated_method_use_arguments(self, _):
method = mock.Mock(__name__='fn')
wrapped_method = http_retry(status_codes=[500])(method)
wrapped_method(42, 'test')
method.assert_called_once_with(42, 'test')
def test_decorated_method_use_arguments(self, _):
method = mock.Mock(__name__='fn')
wrapped_method = http_retry(status_codes=[500])(method)
wrapped_method(42, 'test')
method.assert_called_once_with(42, 'test')
def test_do_retry_on_interesting_errors(self, msleep):
method = mock.Mock(side_effect=self._get_http_error(500),
__name__='fn')
wrapped_method = http_retry(status_codes=[500], attempts=13)(method)
self.assertRaises(requests.HTTPError, wrapped_method)
self.called_times(method, 13)
self.called_times(msleep, 12)
def test_do_not_retry_on_not_interesting_errors(self, msleep):
method = mock.Mock(side_effect=self._get_http_error(404),
__name__='fn')
wrapped_method = http_retry(status_codes=[500])(method)
self.assertRaises(requests.HTTPError, wrapped_method)
self.called_once(method)
self.method_was_not_called(msleep)
def test_do_sleep_on_attempts(self, msleep):
method = mock.Mock(
side_effect=self._get_http_error(500),
__name__='fn')
wrapped_method = http_retry(
status_codes=[500], attempts=2, interval=42)(method)
self.assertRaises(requests.HTTPError, wrapped_method)
msleep.assert_called_once_with(42)
def test_do_retry_on_interesting_errors(self, msleep):
method = mock.Mock(
side_effect=self._get_http_error(500),
__name__='fn')
wrapped_method = http_retry(status_codes=[500], attempts=13)(method)
self.assertRaises(requests.HTTPError, wrapped_method)
self.called_times(method, 13)
self.called_times(msleep, 12)
def test_do_not_retry_on_not_interesting_errors(self, msleep):
method = mock.Mock(
side_effect=self._get_http_error(404),
__name__='fn')
wrapped_method = http_retry(status_codes=[500])(method)
self.assertRaises(requests.HTTPError, wrapped_method)
self.called_once(method)
self.method_was_not_called(msleep)
def test_do_sleep_on_attempts(self, msleep):
method = mock.Mock(side_effect=self._get_http_error(500),
__name__='fn')
wrapped_method = http_retry(status_codes=[500],
attempts=2,
interval=42)(method)
self.assertRaises(requests.HTTPError, wrapped_method)
msleep.assert_called_once_with(42)
def test_stop_retrying_if_success(self, msleep):
method = mock.Mock(
side_effect=[self._get_http_error(500), 'return value'],
__name__='fn')
wrapped_method = http_retry(status_codes=[500], attempts=13)(method)
result = wrapped_method()
self.assertEqual(result, 'return value')
self.called_times(method, 2)
self.called_once(msleep)
def test_stop_retrying_if_success(self, msleep):
method = mock.Mock(
side_effect=[self._get_http_error(500), 'return value'],
__name__='fn')
wrapped_method = http_retry(status_codes=[500], attempts=13)(method)
result = wrapped_method()
self.assertEqual(result, 'return value')
self.called_times(method, 2)
self.called_once(msleep)
