def test_get_unsafe_exception(self):
self.app.conf.result_backend_always_retry, prev = True, self.app.conf.result_backend_always_retry
expected_exc = Exception("failed")
try:
b = BaseBackend(app=self.app)
b._sleep = Mock()
b._get_task_meta_for = Mock()
b._get_task_meta_for.side_effect = [
expected_exc,
{'status': states.SUCCESS, 'result': 42}
]
try:
b.get_task_meta(sentinel.task_id)
assert False
except Exception as exc:
assert b._sleep.call_count == 0
assert exc == expected_exc
finally:
self.app.conf.result_backend_always_retry = prev
def test_get_reaching_max_retries(self):
self.app.conf.result_backend_always_retry, prev = True, self.app.conf.result_backend_always_retry
self.app.conf.result_backend_max_retries, prev_max_retries = 0, self.app.conf.result_backend_max_retries
try:
b = BaseBackend(app=self.app)
b.exception_safe_to_retry = lambda exc: True
b._sleep = Mock()
b._get_task_meta_for = Mock()
b._get_task_meta_for.side_effect = [
Exception("failed"),
{'status': states.SUCCESS, 'result': 42}
]
try:
b.get_task_meta(sentinel.task_id)
assert False
except BackendGetMetaError:
assert b._sleep.call_count == 0
finally:
self.app.conf.result_backend_always_retry = prev
self.app.conf.result_backend_max_retries = prev_max_retries
def test_get_with_retries(self):
self.app.conf.result_backend_always_retry, prev = True, self.app.conf.result_backend_always_retry
try:
b = BaseBackend(app=self.app)
b.exception_safe_to_retry = lambda exc: True
b._sleep = Mock()
b._get_task_meta_for = Mock()
b._get_task_meta_for.side_effect = [
Exception("failed"),
{'status': states.SUCCESS, 'result': 42}
]
res = b.get_task_meta(sentinel.task_id)
assert res == {'status': states.SUCCESS, 'result': 42}
assert b._sleep.call_count == 1
finally:
self.app.conf.result_backend_always_retry = prev
