X-Git-Url: https://plomlompom.com/repos/foo.html?a=blobdiff_plain;f=tests%2Ftest_days.py;h=be2719318140fa52e5c53613e854a727b76d1887;hb=abef15c28141490af71a1b8a11882aaabc0d3775;hp=d3a7bba656d19d39f35b93535564e4e3d17b0a2c;hpb=a8ee9d57c7ec297ef93d616141ac8970869fb6ef;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')