From 6b2421e55ce5d456f5b99f79f03097e57677c08c Mon Sep 17 00:00:00 2001 From: Gabriella Date: Fri, 17 Oct 2025 10:58:06 -0600 Subject: [PATCH] Add break-related tests --- src/py/db_helper_tests.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/src/py/db_helper_tests.py b/src/py/db_helper_tests.py index d3ce16f..5ab959e 100644 --- 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()