Dmitry Malinovsky 11 anni fa
parent
commit
1fe8e281fe
1 ha cambiato i file con 5 aggiunte e 4 eliminazioni
  1. 5 4
      celery/tests/tasks/test_tasks.py

+ 5 - 4
celery/tests/tasks/test_tasks.py

@@ -100,15 +100,16 @@ class TasksCase(AppCase):
                     raise self.retry(countdown=0, exc=exc)
         self.retry_task_customexc = retry_task_customexc
 
-        @self.app.task(autoretry_for=(ZeroDivisionError,), shared=False)
-        def autoretry_task_no_kwargs(a, b):
+        @self.app.task(bind=True, autoretry_for=(ZeroDivisionError,),
+                       shared=False)
+        def autoretry_task_no_kwargs(self, a, b):
             self.iterations += 1
             return a/b
         self.autoretry_task_no_kwargs = autoretry_task_no_kwargs
 
-        @self.app.task(autoretry_for=(ZeroDivisionError,),
+        @self.app.task(bind=True, autoretry_for=(ZeroDivisionError,),
                        retry_kwargs={'max_retries': 5}, shared=False)
-        def autoretry_task(a, b):
+        def autoretry_task(self, a, b):
             self.iterations += 1
             return a/b
         self.autoretry_task = autoretry_task