Add Day.comment field.

[plomtask] / tests / test_days.py

diff --git a/tests/test_days.py b/tests/test_days.py
index d3a7bba656d19d39f35b93535564e4e3d17b0a2c..be2719318140fa52e5c53613e854a727b76d1887 100644 (file)
--- a/tests/test_days.py
+++ b/tests/test_days.py
@@ -81,3 +81,15 @@ class TestsWithDB(TestCase):
                          [day1, day2])
         self.assertEqual(Day.all(self.db_conn, ('2024-01-03, 2024-01-01')),
                          [])
+
+    def test_Day_comment(self):
+        """Test Day.by_date()."""
+        self.assertEqual(None, Day.by_date(self.db_conn, '2024-01-01'))
+        d1 = Day('2024-01-01')
+        d1.save(self.db_conn)
+        day_retrieved = Day.by_date(self.db_conn, d1.date)
+        self.assertEqual(day_retrieved.comment, '')
+        d2 = Day('2024-01-02', 'foo')
+        d2.save(self.db_conn)
+        day_retrieved = Day.by_date(self.db_conn, d2.date)
+        self.assertEqual(day_retrieved.comment, 'foo')