Implement tasks.completed() and add more tests

diff --git a/tasklib/task.py b/tasklib/task.py
index 182fead8b940a3cd4a4532cb62d57345504f6c57..9a02f35e6811c22d17f43e5796c678346ce2295d 100644 (file)
--- a/tasklib/task.py
+++ b/tasklib/task.py
@@ -8,6 +8,7 @@ import subprocess
 DATE_FORMAT = '%Y%m%dT%H%M%SZ'
 REPR_OUTPUT_SIZE = 10
 PENDING = 'pending'
 DATE_FORMAT = '%Y%m%dT%H%M%SZ'
 REPR_OUTPUT_SIZE = 10
 PENDING = 'pending'

+COMPLETED = 'completed'

 
 logger = logging.getLogger(__name__)
 
 
 logger = logging.getLogger(__name__)
 


@@ -195,6 +196,9 @@ class TaskQuerySet(object):
     def pending(self):
         return self.filter(status=PENDING)
 
     def pending(self):
         return self.filter(status=PENDING)
 

+    def completed(self):
+        return self.filter(status=COMPLETED)
+

     def filter(self, *args, **kwargs):
         """
         Returns a new TaskQuerySet with the given filters added.
     def filter(self, *args, **kwargs):
         """
         Returns a new TaskQuerySet with the given filters added.

diff --git a/tasklib/tests.py b/tasklib/tests.py
index 5344d7115faa237b4e13ba80fd8b2bb55b6095fb..94294dc14d89e8331f241de24e0dca2ea49b4868 100644 (file)
--- a/tasklib/tests.py
+++ b/tasklib/tests.py
@@ -25,3 +25,14 @@ 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')
         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)