def test_async_retrier_run_valid_retries(MagicMock, coro):
task = coro(4)
on_retry = MagicMock()
retrier = AsyncRetrier(on_retry=on_retry)
res = run_coro(retrier.run(task, 2, 4, foo=6))
assert res == 12
assert retrier.attempts == 3
assert retrier.error is None
def test_async_retrier_run_success(MagicMock, coro):
on_retry = MagicMock()
retrier = AsyncRetrier(on_retry=on_retry)
task = coro(0)
res = run_coro(retrier.run(task, 2, 4, foo=6))
assert res == 12
assert retrier.attempts == 0
assert retrier.error is None
def test_async_retrier_run_max_retries_error(MagicMock, coro):
on_retry = MagicMock()
task = coro(10)
retrier = AsyncRetrier(on_retry=on_retry,
backoff=ConstantBackoff(interval=0, retries=2))
with pytest.raises(MaxRetriesExceeded):
run_coro(retrier.run(task, 2, 4, foo=6))
assert on_retry.called
assert on_retry.call_count == 2
assert retrier.attempts == 2
assert isinstance(retrier.error, RuntimeError)
def test_async_retrier_run_max_timeout(MagicMock, coro):
on_retry = MagicMock()
task = coro(10)
retrier = AsyncRetrier(timeout=.25, on_retry=on_retry,
backoff=ConstantBackoff(interval=.1))
with pytest.raises(asyncio.TimeoutError):
run_coro(retrier.run(task, 2, 4, foo=6))
assert on_retry.called
assert on_retry.call_count == 3
assert retrier.attempts >= 1
assert retrier.attempts < 4
assert isinstance(retrier.error, RuntimeError)
def test_async_retrier_cancelled_error(MagicMock, coro):
on_retry = MagicMock()
@asyncio.coroutine
def coro(x):
if on_retry.call_count < x:
raise ValueError('small number')
raise asyncio.CancelledError('oops')
retrier = AsyncRetrier(on_retry=on_retry,
backoff=ConstantBackoff(interval=0, retries=10))
with pytest.raises(asyncio.CancelledError):
run_coro(retrier.run(coro, 4))
assert on_retry.called
assert on_retry.call_count == 4
assert retrier.attempts == 5
assert retrier.error is not None
def test_async_retrier_evaluator_error_default(MagicMock, coro):
on_retry = MagicMock()
task = coro(4)
def evaluator(x):
if x < 4:
raise ValueError('small number')
raise ImportError('pass error')
retrier = AsyncRetrier(evaluator=evaluator, on_retry=on_retry,
backoff=ConstantBackoff(interval=0, retries=10))
with pytest.raises(ImportError):
run_coro(retrier.run(task, 2, 4, foo=6))
assert on_retry.called
assert on_retry.call_count == 3
assert retrier.attempts == 3
assert isinstance(retrier.error, ImportError)
def test_async_retrier_evaluator(MagicMock, coro):
on_retry = MagicMock()
task = coro(4)
def evaluator(x):
if x < 4:
raise ValueError('small number')
return False
retrier = AsyncRetrier(evaluator=evaluator, on_retry=on_retry,
backoff=ConstantBackoff(interval=0, retries=10))
res = run_coro(retrier.run(task, 2, 4, foo=6))
assert res == 12
assert on_retry.called
assert on_retry.call_count == 3
assert retrier.attempts == 3
assert retrier.error is None