2025-10-17 17:53:55 +00:00
1 changed files with 8 additions and 0 deletions
--- a/src/py/db_helper_tests.py
+++ b/src/py/db_helper_tests.py
@@ -59,3 +59,11 @@ class TestClass():
        assert formatted_callouts.__class__ is str

    # TODO write tests for the break addition/removal
+
+    def test_add_break(self) -> None:
+        today: datetime.date = datetime.date.today()
+        new_break = self.DATABASE_CONN.add_break(user_id=1, break_start=today, break_end=today)
+        assert new_break is None
+
+    def test_remove_break(self) -> None:
+        today: datetime.date = datetime.date.today()