tests: Check that the priority is removed when assigned None

diff --git a/tasklib/tests.py b/tasklib/tests.py
index 76a951430a9fc48d9e26cd07fad2f81dfd4995b2..fe37e01d55d0d3efc3c0171d2aec97fc653135cd 100644 (file)
--- a/tasklib/tests.py
+++ b/tasklib/tests.py
@@ -240,6 +240,22 @@ class TaskTest(TasklibTest):
         t2 = self.tw.tasks.get(uuid=t1['uuid'])
         self.assertEqual(t1.__hash__(), t2.__hash__())
 
+    def test_adding_task_with_priority(self):
+        t = Task(self.tw, description="test task", priority="M")
+        t.save()
+
+    def test_removing_priority_with_none(self):
+        t = Task(self.tw, description="test task", priority="L")
+        t.save()
+
+        # Remove the priority mark
+        t['priority'] = None
+        t.save()
+
+        # Assert that priority is not there after saving
+        self.assertEqual(t['priority'], None)
+
+
 class AnnotationTest(TasklibTest):
 
     def setUp(self):