def test_worker_task_trace_handle_failure(self):
tid = uuid()
self.mytask.push_request()
try:
self.mytask.request.id = tid
try:
raise ValueError("foo")
except Exception as exc:
w = TraceInfo(states.FAILURE, exc)
w.handle_failure(self.mytask, store_errors=False)
self.assertEqual(self.mytask.backend.get_status(tid), states.PENDING)
w.handle_failure(self.mytask, store_errors=True)
self.assertEqual(self.mytask.backend.get_status(tid), states.FAILURE)
finally:
self.mytask.pop_request()
def test_worker_task_trace_handle_retry(self):
tid = uuid()
self.mytask.push_request(id=tid)
try:
raise ValueError("foo")
except Exception as exc:
try:
raise Retry(str(exc), exc=exc)
except Retry as exc:
w = TraceInfo(states.RETRY, exc)
w.handle_retry(self.mytask, store_errors=False)
self.assertEqual(self.mytask.backend.get_status(tid), states.PENDING)
w.handle_retry(self.mytask, store_errors=True)
self.assertEqual(self.mytask.backend.get_status(tid), states.RETRY)
finally:
self.mytask.pop_request()
def test_worker_task_trace_handle_retry(self):
from celery.exceptions import RetryTaskError
tid = uuid()
mytask.push_request(id=tid)
try:
raise ValueError('foo')
except Exception as exc:
try:
raise RetryTaskError(str(exc), exc=exc)
except RetryTaskError as exc:
w = TraceInfo(states.RETRY, exc)
w.handle_retry(mytask, store_errors=False)
self.assertEqual(mytask.backend.get_status(tid),
states.PENDING)
w.handle_retry(mytask, store_errors=True)
self.assertEqual(mytask.backend.get_status(tid),
states.RETRY)
finally:
mytask.pop_request()
def test_worker_task_trace_handle_failure(self):
tid = uuid()
self.mytask.push_request()
try:
self.mytask.request.id = tid
try:
raise ValueError('foo')
except Exception as exc:
w = TraceInfo(states.FAILURE, exc)
w.handle_failure(
self.mytask,
self.mytask.request,
store_errors=False,
)
self.assertEqual(
self.mytask.backend.get_status(tid),
states.PENDING,
)
w.handle_failure(
self.mytask,
self.mytask.request,
store_errors=True,
)
self.assertEqual(
self.mytask.backend.get_status(tid),
states.FAILURE,
)
finally:
self.mytask.pop_request()
def test_worker_task_trace_handle_retry(self):
tid = uuid()
self.mytask.push_request(id=tid)
try:
raise ValueError('foo')
except Exception as exc:
try:
raise Retry(str(exc), exc=exc)
except Retry as exc:
w = TraceInfo(states.RETRY, exc)
w.handle_retry(
self.mytask,
self.mytask.request,
store_errors=False,
)
self.assertEqual(
self.mytask.backend.get_status(tid),
states.PENDING,
)
w.handle_retry(
self.mytask,
self.mytask.request,
store_errors=True,
)
self.assertEqual(
self.mytask.backend.get_status(tid),
states.RETRY,
)
finally:
self.mytask.pop_request()
def test_worker_task_trace_handle_retry(self):
from celery.exceptions import RetryTaskError
tid = uuid()
mytask.push_request(id=tid)
try:
raise ValueError('foo')
except Exception as exc:
try:
raise RetryTaskError(str(exc), exc=exc)
except RetryTaskError as exc:
w = TraceInfo(states.RETRY, exc)
w.handle_retry(mytask, store_errors=False)
self.assertEqual(mytask.backend.get_status(tid),
states.PENDING)
w.handle_retry(mytask, store_errors=True)
self.assertEqual(mytask.backend.get_status(tid), states.RETRY)
finally:
mytask.pop_request()
