Merge pull request #42 from robgolding63/issue-41-fix

Override __copy__ and __deepcopy__ on LazyUUIDTask

diff --git a/tasklib/lazy.py b/tasklib/lazy.py
index aa2c0654c23f93e2b361bf286c27d677e05b52a2..a1b63ef4fcf48ce1dfb881564b408f77a0374207 100644 (file)
--- a/tasklib/lazy.py
+++ b/tasklib/lazy.py
@@ -42,6 +42,12 @@ class LazyUUIDTask(object):
     def __repr__(self):
         return "LazyUUIDTask: {0}".format(self._uuid)
 
+    def __copy__(self):
+        return LazyUUIDTask(self._tw, self._uuid)
+
+    def __deepcopy__(self, memo):
+        return LazyUUIDTask(self._tw, self._uuid)
+
     @property
     def saved(self):
         """

diff --git a/tasklib/tests.py b/tasklib/tests.py
index b0e342d8348ddd9e451d78d9ecf7f16887e9b5af..be54504a5b69b2368bdabb116dfa215839ab2565 100644 (file)
--- a/tasklib/tests.py
+++ b/tasklib/tests.py
@@ -781,6 +781,14 @@ class TaskFromHookTest(TasklibTest):
         '"start":"20141119T152233Z",'
         '"uuid":"a360fc44-315c-4366-b70c-ea7e7520b749"}')
 
+    input_add_data_recurring = six.StringIO(
+        '{"description":"Mow the lawn",'
+        '"entry":"20160210T224304Z",'
+        '"parent":"62da6227-519c-42c2-915d-dccada926ad7",'
+        '"recur":"weekly",'
+        '"status":"pending",'
+        '"uuid":"81305335-0237-49ff-8e87-b3cdc2369cec"}')
+
     input_modify_data = six.StringIO(input_add_data.getvalue() + '\n' +
         '{"description":"Buy some milk finally",'
         '"entry":"20141118T050231Z",'
@@ -797,6 +805,12 @@ class TaskFromHookTest(TasklibTest):
         self.assertEqual(t['description'], "Buy some milk")
         self.assertEqual(t.pending, True)
 
+    def test_setting_up_from_add_hook_input_recurring(self):
+        t = Task.from_input(input_file=self.input_add_data_recurring,
+                            backend=self.tw)
+        self.assertEqual(t['description'], "Mow the lawn")
+        self.assertEqual(t.pending, True)
+
     def test_setting_up_from_modified_hook_input(self):
         t = Task.from_input(input_file=self.input_modify_data, modify=True,
                             backend=self.tw)