X-Git-Url: https://git.madduck.net/etc/taskwarrior.git/blobdiff_plain/b0013c774cf1667bbde77a5515984e27635dfc1f..73b53d933f2285aaa2ecf1672be44eabf2979316:/tasklib/tests.py?ds=sidebyside

diff --git a/tasklib/tests.py b/tasklib/tests.py
index f63fbb4..0dee2d2 100644
--- a/tasklib/tests.py
+++ b/tasklib/tests.py
@@ -11,7 +11,9 @@ import sys
 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 = (
@@ -285,9 +287,18 @@ class TaskTest(TasklibTest):
         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()
 
+    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()