def test_store_result_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.return_value = {
'status': states.RETRY,
'result': {
"exc_type": "Exception",
"exc_message": ["failed"],
"exc_module": "builtins",
},
}
b._store_result = Mock()
b._store_result.side_effect = [Exception("failed"), 42]
try:
b.store_result(sentinel.task_id, 42, states.SUCCESS)
assert False
except BackendStoreError:
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_store_result_never_retries(self):
self.app.conf.result_backend_always_retry, prev = False, self.app.conf.result_backend_always_retry
expected_exc = Exception("failed")
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.return_value = {
'status': states.RETRY,
'result': {
"exc_type": "Exception",
"exc_message": ["failed"],
"exc_module": "builtins",
},
}
b._store_result = Mock()
b._store_result.side_effect = [expected_exc, 42]
try:
b.store_result(sentinel.task_id, 42, states.SUCCESS)
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_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
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