diff --git a/program/tests/test_playout.py b/program/tests/test_playout.py index acf2c217a01dbee02c8751c5e15a0f88ffe4b2de..dd6e29063c0441353c9ba3aea1c4365e8a5317a7 100644 --- a/program/tests/test_playout.py +++ b/program/tests/test_playout.py @@ -51,9 +51,10 @@ def test_playout(admin_api_client, api_client, daily_rrule, show): response = api_client.get(url()) assert response.status_code == 200 - assert len(response.json()) == 1 + assert len(response.json()) == 2 - assert_entry(response.json()[0], show) + for entry in response.json(): + assert_entry(entry, show) def test_playout_one_week(admin_api_client, api_client, daily_rrule, show): @@ -86,14 +87,14 @@ def test_playout_include_virtual( assert response.status_code == 200 assert len(response.json()) == 3 - virtual_entry1, entry, virtual_entry2 = response.json() + entry1, virtual_entry, entry2 = response.json() - assert_virtual_entry(virtual_entry1, fallback_show) - assert_entry(entry, show) - assert_virtual_entry(virtual_entry2, fallback_show) + assert_entry(entry1, show) + assert_virtual_entry(virtual_entry, fallback_show) + assert_entry(entry1, show) - assert virtual_entry1["end"] == entry["start"] - assert entry["end"] == virtual_entry2["start"] + assert entry1["end"] == virtual_entry["start"] + assert virtual_entry["end"] == entry2["start"] def test_playout_one_week_include_virtual( @@ -118,11 +119,11 @@ def test_playout_one_week_include_virtual( entries = response.json() - for virtual_entry in entries[0::2]: - assert_virtual_entry(virtual_entry, fallback_show) - - for entry in entries[1::2]: + for entry in entries[0::2]: assert_entry(entry, show) + for virtual_entry in entries[1::2]: + assert_virtual_entry(virtual_entry, fallback_show) + for entry1, entry2 in pairwise(entries): assert entry1["end"] == entry2["start"]