Merge pull request #71 from glennyonemitsu/wrapper-update

unit test added for task decorator attributes preservation

tests/test_task.py

@@ -1,4 +1,5 @@
 from __future__ import absolute_import
+import functools
 try:
     from unittest.mock import MagicMock
 except:
@@ -132,6 +133,21 @@ def test(user):
     call(dict(user='something'))
 
 
+def test_task_decorator_attributes():
+    """Test that the decorated task has attributes similar to the original
+    function.
+    """
+
+    decorator = task.decorate()
+
+    def test():
+        pass
+
+    test_task = decorator(test)
+    for attr_name in functools.WRAPPER_ASSIGNMENTS:
+        assert getattr(test_task, attr_name) == getattr(test, attr_name)
+
+
 def test_task_decorator_with_heartbeat():
     """Test the task decorator with heartbeat.
     """