Skip to content

Commit

Permalink
Added additional unit test for not including today
Browse files Browse the repository at this point in the history
  • Loading branch information
BenMillar-CGI committed Mar 25, 2024
1 parent 3af6419 commit 66807ca
Showing 1 changed file with 9 additions and 2 deletions.
11 changes: 9 additions & 2 deletions cla_public/apps/contact/tests/test_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,9 +6,10 @@

class TestGetValidCallbackDays(TestCase):
def test_full_week(self):
full_week_of_datetimes = [datetime.datetime(2024, 1, day, 9, 0) for day in range(1, 7)]
full_week_of_datetimes = [datetime.datetime(2024, 1, day, 9, 0) for day in range(1, 8)]
with patch("cla_public.apps.contact.api.get_valid_callback_slots", return_value=full_week_of_datetimes):
valid_days = get_valid_callback_days()
valid_days = get_valid_callback_days(include_today=True)
assert len(valid_days) == 7
for index, day in enumerate(valid_days):
assert day.date() == full_week_of_datetimes[index].date()

Expand All @@ -25,6 +26,12 @@ def test_single_day(self):
for index, day in enumerate(valid_days):
assert day.date() == valid_slots[index].date()

def test_not_include_today(self):
full_week_of_datetimes = [datetime.datetime(2024, 1, day, 9, 0) for day in range(1, 7)]
with patch("cla_public.apps.contact.api.get_valid_callback_slots", return_value=full_week_of_datetimes):
valid_days = get_valid_callback_days(include_today=False)
assert len(valid_days) == 6


class TestGetValidCallbackTimeslotsOnDate(TestCase):
def test_single_day(self):
Expand Down

0 comments on commit 66807ca

Please sign in to comment.