Ejemplo n.º 1
0
 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()
Ejemplo n.º 2
0
 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()
Ejemplo n.º 3
0
    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()
Ejemplo n.º 4
0
 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()
Ejemplo n.º 5
0
 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)
Ejemplo n.º 6
0
 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()
Ejemplo n.º 7
0
 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()
Ejemplo n.º 8
0
 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()
Ejemplo n.º 9
0
 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)
Ejemplo n.º 10
0
 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()
Ejemplo n.º 11
0
 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()
Ejemplo n.º 12
0
 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, '')