From 7a356089d89bf0706cc6502897877cf4342c6d03 Mon Sep 17 00:00:00 2001
From: Ernesto Rico Schmidt <ernesto@helsinki.at>
Date: Tue, 23 Jul 2024 12:31:15 -0400
Subject: [PATCH] test: fix calendar program tests
---
program/tests/test_calendar_program.py | 28 +++++++++++++++-----------
1 file changed, 16 insertions(+), 12 deletions(-)
diff --git a/program/tests/test_calendar_program.py b/program/tests/test_calendar_program.py
index 417f840f..58379cd0 100644
--- a/program/tests/test_calendar_program.py
+++ b/program/tests/test_calendar_program.py
@@ -21,7 +21,7 @@ def url(include_virtual=False, start=None, end=None):
def assert_episodes(episodes, one_week=False) -> None:
"""asserts the episodes are valid."""
- assert len(episodes) == 1 if not one_week else 7
+ assert len(episodes) == 2 if not one_week else 7
for episode in episodes:
assert episode["id"]
@@ -31,20 +31,24 @@ def assert_episodes(episodes, one_week=False) -> None:
def assert_program(program, show, fallback_show=None, one_week=False) -> None:
"""asserts the program are valid and correspond to the given show and fallback show."""
- assert len(program) == 1 if fallback_show is None and not one_week else (7 if one_week else 3)
+ assert len(program) == 2 if fallback_show is None and not one_week else (7 if one_week else 3)
- assert program[0]["end"]
- assert program[0]["id"]
- assert program[0]["start"]
+ for entry in program:
+ assert entry["end"]
+ assert entry["id"]
+ assert entry["start"]
if fallback_show is None:
- assert program[0]["showId"] == show.id
- assert program[0]["timeslotId"]
+ for entry in program:
+ assert entry["showId"] == show.id
+ assert entry["timeslotId"]
else:
- assert program[0]["showId"] == fallback_show.id
-
- assert program[1]["showId"] == show.id
- assert program[1]["timeslotId"]
+ for entry in program:
+ if entry["timeslotId"]:
+ assert entry["showId"] == show.id
+ assert entry["timeslotId"]
+ else:
+ assert entry["showId"] == fallback_show.id
def assert_shows(shows, show, fallback_show=None) -> None:
@@ -66,7 +70,7 @@ def assert_shows(shows, show, fallback_show=None) -> None:
def assert_timeslots(timeslots, show, one_week=False) -> None:
"""asserts the timeslots are valid and correspond to the given show."""
- assert len(timeslots) == 1 if not one_week else 7
+ assert len(timeslots) == 2 if not one_week else 7
for timeslot in timeslots:
assert timeslot["id"]
--
GitLab