Ask Solem 8 年 前
コミット
c075caebd0
2 ファイル変更5 行追加2 行削除
  1. 3 1
      celery/app/registry.py
  2. 2 1
      t/unit/app/test_registry.py

+ 3 - 1
celery/app/registry.py

@@ -25,7 +25,9 @@ class TaskRegistry(dict):
         instance. Name must be configured prior to registration.
         """
         if task.name is None:
-            raise InvalidTaskError('Task "class {0}" must specify name'.format(task.__class__.__name__))
+            raise InvalidTaskError(
+                'Task class {0!r} must specify .name attribute'.format(
+                    type(task).__name__))
         self[task.name] = inspect.isclass(task) and task() or task
 
     def unregister(self, name):

+ 2 - 1
t/unit/app/test_registry.py

@@ -25,7 +25,8 @@ class test_TaskRegistry:
 
     def setup(self):
         self.mytask = self.app.task(name='A', shared=False)(returns)
-        self.missing_name_task = self.app.task(name=None, shared=False)(returns)
+        self.missing_name_task = self.app.task(
+            name=None, shared=False)(returns)
         self.missing_name_task.name = None  # name is overridden with path
         self.myperiodic = self.app.task(
             name='B', shared=False, type='periodic',