浏览代码

100% coverage for celery.exceptions

Ask Solem 12 年之前
父节点
当前提交
1836c70c73
共有 3 个文件被更改,包括 27 次插入2 次删除
  1. 2 2
      celery/five.py
  2. 20 0
      celery/tests/app/test_exceptions.py
  3. 5 0
      celery/tests/compat_modules/test_decorators.py

+ 2 - 2
celery/five.py

@@ -12,7 +12,7 @@ from __future__ import absolute_import
 
 
 try:
 try:
     from collections import Counter
     from collections import Counter
-except ImportError:
+except ImportError:  # pragma: no cover
     from collections import defaultdict
     from collections import defaultdict
 
 
     def Counter():
     def Counter():
@@ -38,7 +38,7 @@ except ImportError:                         # pragma: no cover
     from collections import UserDict        # noqa
     from collections import UserDict        # noqa
 
 
 
 
-if PY3:
+if PY3:  # pragma: no cover
     import builtins
     import builtins
 
 
     from queue import Queue, Empty
     from queue import Queue, Empty

+ 20 - 0
celery/tests/app/test_exceptions.py

@@ -0,0 +1,20 @@
+from __future__ import absolute_import
+
+import pickle
+
+from datetime import datetime
+
+from celery.exceptions import RetryTaskError
+
+from celery.tests.case import Case
+
+
+class test_RetryTaskError(Case):
+
+    def test_when_datetime(self):
+        x = RetryTaskError('foo', KeyError(), when=datetime.utcnow())
+        self.assertTrue(x.humanize())
+
+    def test_pickleable(self):
+        x = RetryTaskError('foo', KeyError(), when=datetime.utcnow())
+        self.assertTrue(pickle.loads(pickle.dumps(x)))

+ 5 - 0
celery/tests/compat_modules/test_decorators.py

@@ -13,6 +13,11 @@ def add(x, y):
 
 
 class test_decorators(Case):
 class test_decorators(Case):
 
 
+    def test_task_alias(self):
+        from celery import task
+        self.assertTrue(task.__file__)
+        self.assertTrue(task(add))
+
     def setUp(self):
     def setUp(self):
         with warnings.catch_warnings(record=True):
         with warnings.catch_warnings(record=True):
             from celery import decorators
             from celery import decorators