home · contact · privacy
Add Day.prev_date/Day.next_date and use them for navigation in Day page.
[plomtask] / tests / test_days.py
index 2adf7657a46f2f7d49102c7aa3db06bbcc117302..ba91c057a1d38bbe1aaef197e2adeb320fcc2f94 100644 (file)
@@ -94,7 +94,7 @@ class TestsWithDB(TestCase):
         self.assertEqual(Day.all(self.db_conn, ('today', 'today')), [today])
 
     def test_Day_comment(self):
-        """Test Day.by_date()."""
+        """Test Day.comment."""
         self.assertEqual(None, Day.by_date(self.db_conn, '2024-01-01'))
         d1 = Day('2024-01-01')
         d1.save(self.db_conn)
@@ -105,6 +105,11 @@ class TestsWithDB(TestCase):
         day_retrieved = Day.by_date(self.db_conn, d2.date)
         self.assertEqual(day_retrieved.comment, 'foo')
 
+    def test_Day_neighbor_dates(self):
+        """Test Day.prev_date and Day.next_date."""
+        self.assertEqual(Day('2024-01-01').prev_date, '2023-12-31')
+        self.assertEqual(Day('2023-02-28').next_date, '2023-03-01')
+
 
 class TestsWithServer(TestsWithDB):
     """Tests against our HTTP server/handler (and database)."""