Merge branch 'master' of github.com:robgolding63/tasklib

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

diff --git a/tasklib/tests.py b/tasklib/tests.py
index 66f04c7b9394b83c5bfd3912fd4d108d7a9788c2..4f127f01c2475b4f92fd5578782996c0c6b62243 100644 (file)
--- a/tasklib/tests.py
+++ b/tasklib/tests.py
@@ -1,7 +1,6 @@
 import shutil
 import tempfile
 import unittest
-import uuid
 
 from .task import TaskWarrior
 
@@ -26,3 +25,20 @@ class TaskFilterTest(TasklibTest):
         self.assertEqual(len(self.tw.tasks.all()), 1)
         self.assertEqual(self.tw.tasks.all()[0]['description'], 'test task')
         self.assertEqual(self.tw.tasks.all()[0]['status'], 'pending')
+
+    def test_pending_non_empty(self):
+        self.tw.execute_command(['add', 'test task'])
+        self.assertEqual(len(self.tw.tasks.pending()), 1)
+        self.assertEqual(self.tw.tasks.pending()[0]['description'],
+                         'test task')
+        self.assertEqual(self.tw.tasks.pending()[0]['status'], 'pending')
+
+    def test_completed_empty(self):
+        self.tw.execute_command(['add', 'test task'])
+        self.assertEqual(len(self.tw.tasks.completed()), 0)
+
+    def test_completed_non_empty(self):
+        self.tw.execute_command(['add', 'test task'])
+        self.assertEqual(len(self.tw.tasks.completed()), 0)
+        self.tw.tasks.all()[0].done()
+        self.assertEqual(len(self.tw.tasks.completed()), 1)