TaskWarrior: Actually return converted datetime object

[etc/taskwarrior.git] / tasklib / tests.py

diff --git a/tasklib/tests.py b/tasklib/tests.py
index f63fbb4935ba5f30c82de7ea66c7e47ed07fc693..0dee2d2f4489857e9ba87ed13812001578c86c7f 100644 (file)
--- a/tasklib/tests.py
+++ b/tasklib/tests.py
@@ -11,7 +11,9 @@ import sys
 import tempfile
 import unittest
 
 import tempfile
 import unittest
 

-from .task import TaskWarrior, Task, ReadOnlyDictView, local_zone, DATE_FORMAT
+from .backends import TaskWarrior
+from .task import Task, ReadOnlyDictView
+from .serializing import DATE_FORMAT, local_zone

 
 # http://taskwarrior.org/docs/design/task.html , Section: The Attributes
 TASK_STANDARD_ATTRS = (
 
 # http://taskwarrior.org/docs/design/task.html , Section: The Attributes
 TASK_STANDARD_ATTRS = (


@@ -285,9 +287,18 @@ class TaskTest(TasklibTest):
         self.assertRaises(Task.CompletedTask, t.start)
 
     def test_add_completed_task(self):
         self.assertRaises(Task.CompletedTask, t.start)
 
     def test_add_completed_task(self):

-        t = Task(self.tw, description="test", status="completed", end="now")
+        t = Task(self.tw, description="test", status="completed",
+                 end=datetime.datetime.now())

         t.save()
 
         t.save()
 

+    def test_add_multiple_completed_tasks(self):
+        t1 = Task(self.tw, description="test1", status="completed",
+                 end=datetime.datetime.now())
+        t2 = Task(self.tw, description="test2", status="completed",
+                 end=datetime.datetime.now())
+        t1.save()
+        t2.save()
+

     def test_complete_deleted_task(self):
         t = Task(self.tw, description="test task")
         t.save()
     def test_complete_deleted_task(self):
         t = Task(self.tw, description="test task")
         t.save()