From: Tomas Babej <tomasbabej@gmail.com>
Date: Fri, 19 Dec 2014 07:37:43 +0000 (+0100)
Subject: tests: Add tests to filter by attributes and their empty values
X-Git-Url: https://git.madduck.net/etc/taskwarrior.git/commitdiff_plain/98a9ea60e0c22e140650ff5a89dfd544d9c05cb0

tests: Add tests to filter by attributes and their empty values
---

diff --git a/tasklib/tests.py b/tasklib/tests.py
index e8bc0c3..2fa1371 100644
--- a/tasklib/tests.py
+++ b/tasklib/tests.py
@@ -45,6 +45,30 @@ class TaskFilterTest(TasklibTest):
         self.tw.tasks.all()[0].done()
         self.assertEqual(len(self.tw.tasks.completed()), 1)
 
+    def test_filtering_by_attribute(self):
+        Task(self.tw, description="no priority task").save()
+        Task(self.tw, priority="H", description="high priority task").save()
+        self.assertEqual(len(self.tw.tasks.all()), 2)
+
+        # Assert that the correct number of tasks is returned
+        self.assertEqual(len(self.tw.tasks.filter(priority="H")), 1)
+
+        # Assert that the correct tasks are returned
+        high_priority_task = self.tw.tasks.get(priority="H")
+        self.assertEqual(high_priority_task['description'], "high priority task")
+
+    def test_filtering_by_empty_attribute(self):
+        Task(self.tw, description="no priority task").save()
+        Task(self.tw, priority="H", description="high priority task").save()
+        self.assertEqual(len(self.tw.tasks.all()), 2)
+
+        # Assert that the correct number of tasks is returned
+        self.assertEqual(len(self.tw.tasks.filter(priority=None)), 1)
+
+        # Assert that the correct tasks are returned
+        no_priority_task = self.tw.tasks.get(priority=None)
+        self.assertEqual(no_priority_task['description'], "no priority task")
+
 
 class AnnotationTest(TasklibTest):