Tests: Add tests for setting and removing due date

diff --git a/tasklib/tests.py b/tasklib/tests.py
index fe37e01d55d0d3efc3c0171d2aec97fc653135cd..b87868f4125cde5e0eb7a96aaa7b68fb9a30ba6a 100644 (file)
--- a/tasklib/tests.py
+++ b/tasklib/tests.py
@@ -1,5 +1,6 @@
 # coding=utf-8
 
 # coding=utf-8
 

+import datetime

 import shutil
 import tempfile
 import unittest
 import shutil
 import tempfile
 import unittest


@@ -255,6 +256,21 @@ class TaskTest(TasklibTest):
         # Assert that priority is not there after saving
         self.assertEqual(t['priority'], None)
 
         # Assert that priority is not there after saving
         self.assertEqual(t['priority'], None)
 

+    def test_adding_task_with_due_time(self):
+        t = Task(self.tw, description="test task", due=datetime.datetime.now())
+        t.save()
+
+    def test_removing_due_time_with_none(self):
+        t = Task(self.tw, description="test task", due=datetime.datetime.now())
+        t.save()
+
+        # Remove the due timestamp
+        t['due'] = None
+        t.save()
+
+        # Assert that due timestamp is no longer there
+        self.assertEqual(t['due'], None)
+

 
 class AnnotationTest(TasklibTest):
 
 
 class AnnotationTest(TasklibTest):