def test_worker_task_trace_handle_failure(self): tid = uuid() mytask.request.update({"id": tid}) try: einfo = self.create_exception(ValueError("foo")) w = TraceInfo(states.FAILURE, einfo[1], einfo) w.handle_failure(mytask, store_errors=False) self.assertEqual(mytask.backend.get_status(tid), states.PENDING) w.handle_failure(mytask, store_errors=True) self.assertEqual(mytask.backend.get_status(tid), states.FAILURE) finally: mytask.request.clear()
def test_worker_task_trace_handle_failure(self): tid = uuid() mytask.request.update({"id": tid}) einfo = None try: try: raise ValueError("foo") except Exception, exc: w = TraceInfo(states.FAILURE, exc) w.handle_failure(mytask, store_errors=False) self.assertEqual(mytask.backend.get_status(tid), states.PENDING) w.handle_failure(mytask, store_errors=True) self.assertEqual(mytask.backend.get_status(tid), states.FAILURE) finally: mytask.request.clear()
def test_worker_task_trace_handle_retry(self): from celery.exceptions import RetryTaskError tid = uuid() mytask.request.update({"id": tid}) try: _, value_, _ = self.create_exception(ValueError("foo")) einfo = self.create_exception(RetryTaskError(str(value_), exc=value_)) w = TraceInfo(states.RETRY, einfo[1], einfo) 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.request.clear()
def test_worker_task_trace_handle_failure(self): tid = uuid() mytask.push_request() try: mytask.request.id = tid try: raise ValueError("foo") except Exception, exc: w = TraceInfo(states.FAILURE, exc) w.handle_failure(mytask, store_errors=False) self.assertEqual(mytask.backend.get_status(tid), states.PENDING) w.handle_failure(mytask, store_errors=True) self.assertEqual(mytask.backend.get_status(tid), states.FAILURE) finally: mytask.pop_request()
def test_worker_task_trace_handle_retry(self): from celery.exceptions import RetryTaskError tid = uuid() mytask.request.update({"id": tid}) einfo = tb = None try: raise ValueError("foo") except Exception, exc: try: raise RetryTaskError(str(exc), exc=exc) except RetryTaskError, 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)
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_retry(self): from celery.exceptions import RetryTaskError tid = uuid() mytask.request.update({"id": tid}) try: _, value_, _ = self.create_exception(ValueError("foo")) einfo = self.create_exception( RetryTaskError(str(value_), exc=value_)) w = TraceInfo(states.RETRY, einfo[1], einfo) 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.request.clear()
def test_worker_task_trace_handle_retry(self): from celery.exceptions import RetryTaskError tid = uuid() mytask.request.update({"id": tid}) try: raise ValueError("foo") except Exception, exc: try: raise RetryTaskError(str(exc), exc=exc) except RetryTaskError, 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)
def test_worker_task_trace_handle_failure(self): tid = uuid() mytask.request.update({'id': tid}) try: try: raise ValueError('foo') except Exception as exc: w = TraceInfo(states.FAILURE, exc) w.handle_failure(mytask, store_errors=False) self.assertEqual(mytask.backend.get_status(tid), states.PENDING) w.handle_failure(mytask, store_errors=True) self.assertEqual(mytask.backend.get_status(tid), states.FAILURE) finally: mytask.request.clear()
def test_without_exc_info(self): x = TraceInfo(states.SUCCESS) self.assertIsNone(x.exc_type) self.assertIsNone(x.exc_value) self.assertIsNone(x.tb) self.assertEqual(x.strtb, '')