From 84fc954e46aef01c1cd15236edc70d5d3acc42f2 Mon Sep 17 00:00:00 2001
From: Ernesto Rico Schmidt <ernesto@helsinki.at>
Date: Wed, 17 Jul 2024 17:55:48 -0400
Subject: [PATCH] test: add one week calendar program tests
---
program/tests/test_calendar_program.py | 86 +++++++++++++++++++++-----
1 file changed, 71 insertions(+), 15 deletions(-)
diff --git a/program/tests/test_calendar_program.py b/program/tests/test_calendar_program.py
index 4b265472..417f840f 100644
--- a/program/tests/test_calendar_program.py
+++ b/program/tests/test_calendar_program.py
@@ -1,3 +1,5 @@
+from datetime import datetime, timedelta
+
import pytest
from conftest import create_daily_schedule
@@ -16,19 +18,20 @@ def url(include_virtual=False, start=None, end=None):
return "/api/v1/program/calendar/"
-def assert_episodes(episodes) -> None:
+def assert_episodes(episodes, one_week=False) -> None:
"""asserts the episodes are valid."""
- assert len(episodes) == 1
+ assert len(episodes) == 1 if not one_week else 7
- assert episodes[0]["id"]
- assert episodes[0]["timeslotId"]
+ for episode in episodes:
+ assert episode["id"]
+ assert episode["timeslotId"]
-def assert_program(program, show, fallback_show=None) -> 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 else 3
+ assert len(program) == 1 if fallback_show is None and not one_week else (7 if one_week else 3)
assert program[0]["end"]
assert program[0]["id"]
@@ -60,16 +63,17 @@ def assert_shows(shows, show, fallback_show=None) -> None:
assert shows[1]["name"] == fallback_show.name
-def assert_timeslots(timeslots, show) -> None:
- """asserts the timeslots are valid correspond to the given show."""
+def assert_timeslots(timeslots, show, one_week=False) -> None:
+ """asserts the timeslots are valid and correspond to the given show."""
- assert len(timeslots) == 1
+ assert len(timeslots) == 1 if not one_week else 7
- assert timeslots[0]["id"]
- assert timeslots[0]["noteId"]
- assert timeslots[0]["end"]
- assert timeslots[0]["start"]
- assert timeslots[0]["showId"] == show.id
+ for timeslot in timeslots:
+ assert timeslot["id"]
+ assert timeslot["noteId"]
+ assert timeslot["end"]
+ assert timeslot["start"]
+ assert timeslot["showId"] == show.id
def test_calendar(admin_api_client, api_client, daily_rrule, show):
@@ -91,6 +95,28 @@ def test_calendar(admin_api_client, api_client, daily_rrule, show):
assert_timeslots(value, show)
+def test_calendar_one_week(admin_api_client, api_client, daily_rrule, show):
+ create_daily_schedule(admin_api_client, daily_rrule, show)
+
+ now = datetime.now()
+ in_one_week = now + timedelta(days=7)
+
+ response = api_client.get(url(start=now.isoformat(), end=in_one_week.isoformat()))
+
+ assert response.status_code == 200
+
+ for key, value in response.json().items():
+ match key:
+ case "episodes":
+ assert_episodes(value, one_week=True)
+ case "program":
+ assert_program(value, show, one_week=True)
+ case "shows":
+ assert_shows(value, show)
+ case "timeslots":
+ assert_timeslots(value, show, one_week=True)
+
+
def test_calendar_include_virtual(
admin_api_client,
api_client,
@@ -110,9 +136,39 @@ def test_calendar_include_virtual(
case "episodes":
assert_episodes(value)
case "program":
- print(value)
assert_program(value, show, fallback_show)
case "shows":
assert_shows(value, show, fallback_show)
case "timeslots":
assert_timeslots(value, show)
+
+
+def test_calendar_one_week_include_virtual(
+ admin_api_client,
+ api_client,
+ daily_rrule,
+ show,
+ fallback_show,
+ radio_settings,
+):
+ create_daily_schedule(admin_api_client, daily_rrule, show)
+
+ now = datetime.now()
+ in_one_week = now + timedelta(days=7)
+
+ response = api_client.get(
+ url(include_virtual=True, start=now.isoformat(), end=in_one_week.isoformat())
+ )
+
+ assert response.status_code == 200
+
+ for key, value in response.json().items():
+ match key:
+ case "episodes":
+ assert_episodes(value, one_week=True)
+ case "program":
+ assert_program(value, show=show, fallback_show=fallback_show, one_week=True)
+ case "shows":
+ assert_shows(value, show=show, fallback_show=fallback_show)
+ case "timeslots":
+ assert_timeslots(value, show=show, one_week=True)
--
GitLab