X-Git-Url: https://plomlompom.com/repos/bar%20baz.html?a=blobdiff_plain;f=tests%2Ftest_days.py;h=be2719318140fa52e5c53613e854a727b76d1887;hb=bf8b491d50379772879f1cc9cbe6846fe50ce63b;hp=d3a7bba656d19d39f35b93535564e4e3d17b0a2c;hpb=1cddadfe3426a5845008e38f430b4bd371986b51;p=plomtask

diff --git a/tests/test_days.py b/tests/test_days.py
index d3a7bba..be27193 100644
--- 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')