浏览代码

Tests passed

Dmitry Malinovsky 11 年之前
父节点
当前提交
1fe8e281fe
共有 1 个文件被更改,包括 5 次插入4 次删除
  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