Add single-day view / retrieval.

[plomtask] / tests / test_days.py

diff --git a/tests/test_days.py b/tests/test_days.py
index 8f5f2678c002034f02a1f036c28b976dd1b589b6..5e3713882ca622d207026792f2f47df23d807283 100644 (file)
--- a/tests/test_days.py
+++ b/tests/test_days.py
@@ -36,6 +36,18 @@ class DayTests(TestCase):
         """Test Day.weekday."""
         self.assertEqual(Day('2024-03-17').weekday, 'Sunday')
 
+    def test_Day_by_date(self):
+        """Test Day.by_date()."""
+        timestamp = datetime.now().timestamp()
+        db_file = DatabaseFile(f'test_db:{timestamp}')
+        db_file.remake()
+        conn = DatabaseConnection(db_file)
+        self.assertEqual(None, Day.by_date(conn, '2024-01-01'))
+        Day('2024-01-01').save(conn)
+        self.assertEqual(Day('2024-01-01'), Day.by_date(conn, '2024-01-01'))
+        conn.close()
+        remove_file(db_file.path)
+
     def test_Day_all(self):
         """Test Day.all(), especially in regards to date range filtering."""
         timestamp = datetime.now().timestamp()
@@ -49,6 +61,7 @@ class DayTests(TestCase):
         day2.save(conn)
         day3.save(conn)
         self.assertEqual(Day.all(conn), [day1, day2, day3])
+        self.assertEqual(Day.all(conn, ('', '')), [day1, day2, day3])
         self.assertEqual(Day.all(conn, ('2024-01-01', '2024-01-03')),
                          [day1, day2, day3])
         self.assertEqual(Day.all(conn, ('2024-01-02', '2024-01-03')),