sintonia/playout/tests/player/schedule_test.py

643 lines
19 KiB
Python
Raw Normal View History

import random
from datetime import timedelta
import pytest
from dateutil.parser import isoparse
from libretime_api_client.v2 import ApiClient
from libretime_playout.player.events import EventKind
from libretime_playout.player.schedule import (
generate_file_events,
generate_live_events,
generate_webstream_events,
get_schedule,
)
@pytest.fixture(name="api_client")
def _api_client_fixture():
base_url = "http://localhost"
return ApiClient(base_url=base_url, api_key="test_key")
SHOW_1 = {"id": 1, "name": "Show 1", "live_enabled": False}
SHOW_2 = {"id": 2, "name": "Show 2", "live_enabled": False}
SHOW_3 = {"id": 3, "name": "Show 3", "live_enabled": True}
SHOW_4 = {"id": 4, "name": "Show 4", "live_enabled": False}
SHOW_INSTANCE_1 = {
"id": 1,
"starts_at": "2022-09-05T11:00:00Z",
"ends_at": "2022-09-05T11:10:00Z",
"show": 1,
}
SHOW_INSTANCE_2 = {
"id": 2,
"starts_at": "2022-09-05T11:10:00Z",
"ends_at": "2022-09-05T12:10:00Z",
"show": 2,
}
SHOW_INSTANCE_3 = {
"id": 3,
"starts_at": "2022-09-05T12:10:00Z",
"ends_at": "2022-09-05T13:00:00Z",
"show": 3,
}
SHOW_INSTANCE_4 = {
"id": 4,
"starts_at": "2022-09-05T13:00:00Z",
"ends_at": "2022-09-05T14:10:00Z",
"show": 4,
}
FILE_1 = {
"id": 1,
"mime": "audio/flac",
"length": "00:03:41.041723",
"replay_gain": "4.52",
"cue_in": "00:00:08.252450",
"cue_out": "00:03:27.208000",
"artist_name": "Nils Frahm",
"album_title": "Tripping with Nils Frahm",
"track_title": "The Dane",
"url": None,
"size": 10000,
}
FILE_2 = {
"id": 2,
"mime": "audio/flac",
"length": "00:06:08.668798",
"replay_gain": "11.46",
"cue_in": "00:00:13.700800",
"cue_out": "00:05:15.845000",
"artist_name": "Nils Frahm",
"album_title": "Tripping with Nils Frahm",
"track_title": "My Friend the Forest",
"url": None,
"size": 10000,
}
FILE_3 = {
"id": 3,
"mime": "audio/flac",
"length": "00:14:18.400000",
"replay_gain": "-2.13",
"cue_in": "00:00:55.121100",
"cue_out": "00:14:18.400000",
"artist_name": "Nils Frahm",
"album_title": "Tripping with Nils Frahm",
"track_title": "All Melody",
"url": None,
"size": 10000,
}
FILE_4 = {
"id": 4,
"mime": "audio/flac",
"length": "00:10:45.472200",
"replay_gain": "-1.65",
"cue_in": "00:00:00",
"cue_out": "00:10:26.891000",
"artist_name": "Nils Frahm",
"album_title": "Tripping with Nils Frahm",
"track_title": "#2",
"url": None,
"size": 10000,
}
FILE_5 = {
"id": 5,
"mime": "audio/mp3",
"length": "00:59:04.989000",
"replay_gain": "-1.39",
"cue_in": "00:00:00",
"cue_out": "00:58:59.130000",
"artist_name": "Democracy Now! Audio",
"album_title": "Democracy Now! Audio",
"track_title": "Democracy Now! 2022-09-05 Monday",
"url": None,
"size": 10000,
}
WEBSTREAM_1 = {
"id": 1,
"name": "External radio",
"url": "http://stream.radio.org/main.ogg",
}
SCHEDULE_1 = {
"id": 1,
"starts_at": "2022-09-05T11:00:00Z",
"ends_at": "2022-09-05T11:05:02.144200Z",
"cue_in": "00:00:13.700800",
"cue_out": "00:05:15.845000",
"fade_in": "00:00:00.500000",
"fade_out": "00:00:00.500000",
"file": 2,
"instance": 1,
"length": "00:05:02.144200",
"stream": None,
}
SCHEDULE_2 = {
"id": 2,
"starts_at": "2022-09-05T11:05:02.144200Z",
"ends_at": "2022-09-05T11:10:00Z",
"cue_in": "00:00:00",
"cue_out": "00:04:57.855800",
"fade_in": "00:00:00.500000",
"fade_out": "00:00:00.500000",
"file": 4,
"instance": 1,
"length": "00:10:26.891000",
"stream": None,
}
SCHEDULE_3 = {
"id": 3,
"starts_at": "2022-09-05T11:10:00Z",
"ends_at": "2022-09-05T12:08:59Z",
"cue_in": "00:00:00",
"cue_out": "00:58:59.130000",
"fade_in": "00:00:00.500000",
"fade_out": "00:00:00.500000",
"file": 5,
"instance": 2,
"length": "00:58:59.130000",
"stream": None,
}
SCHEDULE_4 = {
"id": 4,
"starts_at": "2022-09-05T12:08:59Z",
"ends_at": "2022-09-05T12:10:00Z",
"cue_in": "00:00:00",
"cue_out": "00:01:01",
"fade_in": "00:00:00.500000",
"fade_out": "00:00:00.500000",
"file": 4,
"instance": 2,
"length": "00:10:26.891000",
"stream": None,
}
SCHEDULE_5 = {
"id": 5,
"starts_at": "2022-09-05T12:10:00Z",
"ends_at": "2022-09-05T12:40:00Z",
"cue_in": "00:00:00",
"cue_out": "00:30:00",
"fade_in": "00:00:00.500000",
"fade_out": "00:00:00.500000",
"file": None,
"instance": 3,
"length": "00:30:00",
"stream": 1,
}
SCHEDULE_6 = {
"id": 6,
"starts_at": "2022-09-05T12:40:00Z",
"ends_at": "2022-09-05T12:53:23Z",
"cue_in": "00:00:55.121100",
"cue_out": "00:14:18.400000",
"fade_in": "00:00:00.500000",
"fade_out": "00:00:00.500000",
"file": 3,
"instance": 3,
"length": "00:13:23.278900",
"stream": None,
}
SCHEDULE_7 = {
"id": 7,
"starts_at": "2022-09-05T12:53:23Z",
"ends_at": "2022-09-05T12:58:25Z",
"cue_in": "00:00:13.700800",
"cue_out": "00:05:15.845000",
"fade_in": "00:00:00.500000",
"fade_out": "00:00:00.500000",
"file": 2,
"instance": 3,
"length": "00:05:02.144200",
"stream": None,
}
SCHEDULE_8 = {
"id": 8,
"starts_at": "2022-09-05T12:58:25Z",
"ends_at": "2022-09-05T13:00:00Z",
"cue_in": "00:00:08.252450",
"cue_out": "00:01:35",
"fade_in": "00:00:00.500000",
"fade_out": "00:00:00.500000",
"file": 1,
"instance": 3,
"length": "00:03:18.955550",
"stream": None,
}
SCHEDULE_9 = {
"id": 9,
"starts_at": "2022-09-05T13:00:00Z",
"ends_at": "2022-09-05T13:05:02.144200Z",
"cue_in": "00:00:13.700800",
"cue_out": "00:05:15.845000",
"fade_in": "00:00:00.500000",
"fade_out": "00:00:00.500000",
"file": 2,
"instance": 4,
"length": "00:05:02.144200",
"stream": None,
}
SCHEDULE_10 = {
"id": 10,
"starts_at": "2022-09-05T13:05:02.144200Z",
"ends_at": "2022-09-05T13:10:00Z",
"cue_in": "00:00:00",
"cue_out": "00:04:57.855800",
"fade_in": "00:00:00.500000",
"fade_out": "00:00:00.500000",
"file": 4,
"instance": 4,
"length": "00:10:26.891000",
"stream": None,
}
SCHEDULE = [
SCHEDULE_1,
SCHEDULE_2,
SCHEDULE_3,
SCHEDULE_4,
SCHEDULE_5,
SCHEDULE_6,
SCHEDULE_7,
SCHEDULE_8,
SCHEDULE_9,
SCHEDULE_10,
]
def test_generate_live_events():
show_instance_3 = SHOW_INSTANCE_3.copy()
show_instance_3["starts_at"] = isoparse(show_instance_3["starts_at"])
show_instance_3["ends_at"] = isoparse(show_instance_3["ends_at"])
result = {}
generate_live_events(result, show_instance_3, 0.0)
assert result == {
"2022-09-05-13-00-00": {
"type": EventKind.EVENT,
"event_type": "kick_out",
"start": "2022-09-05-13-00-00",
"end": "2022-09-05-13-00-00",
}
}
result = {}
generate_live_events(result, show_instance_3, 2.0)
assert result == {
"2022-09-05-12-59-58": {
"type": EventKind.EVENT,
"event_type": "switch_off",
"start": "2022-09-05-12-59-58",
"end": "2022-09-05-12-59-58",
},
"2022-09-05-13-00-00": {
"type": EventKind.EVENT,
"event_type": "kick_out",
"start": "2022-09-05-13-00-00",
"end": "2022-09-05-13-00-00",
},
}
def test_generate_file_events():
schedule_1 = SCHEDULE_1.copy()
schedule_1["starts_at"] = isoparse(schedule_1["starts_at"])
schedule_1["ends_at"] = isoparse(schedule_1["ends_at"])
result = {}
generate_file_events(result, schedule_1, FILE_2, SHOW_1)
assert result == {
"2022-09-05-11-00-00": {
"type": EventKind.FILE,
"row_id": 1,
"start": "2022-09-05-11-00-00",
"end": "2022-09-05-11-05-02",
"uri": None,
"id": 2,
"show_name": "Show 1",
"fade_in": 500.0,
"fade_out": 500.0,
"cue_in": 13.7008,
"cue_out": 315.845,
"metadata": {
"track_title": "My Friend the Forest",
"artist_name": "Nils Frahm",
"mime": "audio/flac",
},
"replay_gain": "11.46",
"filesize": 10000,
}
}
def test_generate_webstream_events():
schedule_5 = SCHEDULE_5.copy()
schedule_5["starts_at"] = isoparse(schedule_5["starts_at"])
schedule_5["ends_at"] = isoparse(schedule_5["ends_at"])
result = {}
generate_webstream_events(result, schedule_5, WEBSTREAM_1, SHOW_3)
assert result == {
"2022-09-05-12-10-00": {
"type": EventKind.STREAM_BUFFER_START,
"row_id": 5,
"start": "2022-09-05-12-09-55",
"end": "2022-09-05-12-09-55",
"uri": "http://stream.radio.org/main.ogg",
"id": 1,
},
"2022-09-05-12-10-00_0": {
"type": EventKind.STREAM_OUTPUT_START,
"row_id": 5,
"start": "2022-09-05-12-10-00",
"end": "2022-09-05-12-40-00",
"uri": "http://stream.radio.org/main.ogg",
"id": 1,
"show_name": "Show 3",
},
"2022-09-05-12-40-00": {
"type": EventKind.STREAM_BUFFER_END,
"row_id": 5,
"start": "2022-09-05-12-40-00",
"end": "2022-09-05-12-40-00",
"uri": "http://stream.radio.org/main.ogg",
"id": 1,
},
"2022-09-05-12-40-00_0": {
"type": EventKind.STREAM_OUTPUT_END,
"row_id": 5,
"start": "2022-09-05-12-40-00",
"end": "2022-09-05-12-40-00",
"uri": "http://stream.radio.org/main.ogg",
"id": 1,
},
}
@pytest.mark.parametrize(
"schedule",
[
(SCHEDULE),
(random.sample(SCHEDULE, len(SCHEDULE))),
],
)
def test_get_schedule(schedule, requests_mock, api_client: ApiClient):
base_url = api_client.base_url
requests_mock.get(
f"{base_url}/api/v2/stream/preferences",
json={
"input_fade_transition": 2.0,
"message_format": 0,
"message_offline": "",
},
)
requests_mock.get(f"{base_url}/api/v2/schedule", json=schedule)
requests_mock.get(f"{base_url}/api/v2/shows/1", json=SHOW_1)
requests_mock.get(f"{base_url}/api/v2/shows/2", json=SHOW_2)
requests_mock.get(f"{base_url}/api/v2/shows/3", json=SHOW_3)
requests_mock.get(f"{base_url}/api/v2/shows/4", json=SHOW_4)
requests_mock.get(f"{base_url}/api/v2/show-instances/1", json=SHOW_INSTANCE_1)
requests_mock.get(f"{base_url}/api/v2/show-instances/2", json=SHOW_INSTANCE_2)
requests_mock.get(f"{base_url}/api/v2/show-instances/3", json=SHOW_INSTANCE_3)
requests_mock.get(f"{base_url}/api/v2/show-instances/4", json=SHOW_INSTANCE_4)
requests_mock.get(f"{base_url}/api/v2/files/1", json=FILE_1)
requests_mock.get(f"{base_url}/api/v2/files/2", json=FILE_2)
requests_mock.get(f"{base_url}/api/v2/files/3", json=FILE_3)
requests_mock.get(f"{base_url}/api/v2/files/4", json=FILE_4)
requests_mock.get(f"{base_url}/api/v2/files/5", json=FILE_5)
requests_mock.get(f"{base_url}/api/v2/webstreams/1", json=WEBSTREAM_1)
assert get_schedule(api_client) == {
"media": {
"2022-09-05-11-00-00": {
"type": EventKind.FILE,
"row_id": 1,
"start": "2022-09-05-11-00-00",
"end": "2022-09-05-11-05-02",
"uri": None,
"id": 2,
"show_name": "Show 1",
"fade_in": 500.0,
"fade_out": 500.0,
"cue_in": 13.7008,
"cue_out": 315.845,
"metadata": {
"track_title": "My Friend the Forest",
"artist_name": "Nils Frahm",
"mime": "audio/flac",
},
"replay_gain": "11.46",
"filesize": 10000,
},
"2022-09-05-11-05-02": {
"type": EventKind.FILE,
"row_id": 2,
"start": "2022-09-05-11-05-02",
"end": "2022-09-05-11-10-00",
"uri": None,
"id": 4,
"show_name": "Show 1",
"fade_in": 500.0,
"fade_out": 500.0,
"cue_in": 0.0,
"cue_out": 297.8558,
"metadata": {
"track_title": "#2",
"artist_name": "Nils Frahm",
"mime": "audio/flac",
},
"replay_gain": "-1.65",
"filesize": 10000,
},
"2022-09-05-11-10-00": {
"type": EventKind.FILE,
"row_id": 3,
"start": "2022-09-05-11-10-00",
"end": "2022-09-05-12-08-59",
"uri": None,
"id": 5,
"show_name": "Show 2",
"fade_in": 500.0,
"fade_out": 500.0,
"cue_in": 0.0,
"cue_out": 3539.13,
"metadata": {
"track_title": "Democracy Now! 2022-09-05 Monday",
"artist_name": "Democracy Now! Audio",
"mime": "audio/mp3",
},
"replay_gain": "-1.39",
"filesize": 10000,
},
"2022-09-05-12-08-59": {
"type": EventKind.FILE,
"row_id": 4,
"start": "2022-09-05-12-08-59",
"end": "2022-09-05-12-10-00",
"uri": None,
"id": 4,
"show_name": "Show 2",
"fade_in": 500.0,
"fade_out": 500.0,
"cue_in": 0.0,
"cue_out": 61.0,
"metadata": {
"track_title": "#2",
"artist_name": "Nils Frahm",
"mime": "audio/flac",
},
"replay_gain": "-1.65",
"filesize": 10000,
},
"2022-09-05-12-10-00": {
"type": EventKind.STREAM_BUFFER_START,
"row_id": 5,
"start": "2022-09-05-12-09-55",
"end": "2022-09-05-12-09-55",
"uri": "http://stream.radio.org/main.ogg",
"id": 1,
},
"2022-09-05-12-10-00_0": {
"type": EventKind.STREAM_OUTPUT_START,
"row_id": 5,
"start": "2022-09-05-12-10-00",
"end": "2022-09-05-12-40-00",
"uri": "http://stream.radio.org/main.ogg",
"id": 1,
"show_name": "Show 3",
},
"2022-09-05-12-40-00": {
"type": EventKind.STREAM_BUFFER_END,
"row_id": 5,
"start": "2022-09-05-12-40-00",
"end": "2022-09-05-12-40-00",
"uri": "http://stream.radio.org/main.ogg",
"id": 1,
},
"2022-09-05-12-40-00_0": {
"type": EventKind.STREAM_OUTPUT_END,
"row_id": 5,
"start": "2022-09-05-12-40-00",
"end": "2022-09-05-12-40-00",
"uri": "http://stream.radio.org/main.ogg",
"id": 1,
},
"2022-09-05-12-40-00_1": {
"type": EventKind.FILE,
"row_id": 6,
"start": "2022-09-05-12-40-00",
"end": "2022-09-05-12-53-23",
"uri": None,
"id": 3,
"show_name": "Show 3",
"fade_in": 500.0,
"fade_out": 500.0,
"cue_in": 55.1211,
"cue_out": 858.4,
"metadata": {
"track_title": "All Melody",
"artist_name": "Nils Frahm",
"mime": "audio/flac",
},
"replay_gain": "-2.13",
"filesize": 10000,
},
"2022-09-05-12-53-23": {
"type": EventKind.FILE,
"row_id": 7,
"start": "2022-09-05-12-53-23",
"end": "2022-09-05-12-58-25",
"uri": None,
"id": 2,
"show_name": "Show 3",
"fade_in": 500.0,
"fade_out": 500.0,
"cue_in": 13.7008,
"cue_out": 315.845,
"metadata": {
"track_title": "My Friend the Forest",
"artist_name": "Nils Frahm",
"mime": "audio/flac",
},
"replay_gain": "11.46",
"filesize": 10000,
},
"2022-09-05-12-58-25": {
"type": EventKind.FILE,
"row_id": 8,
"start": "2022-09-05-12-58-25",
"end": "2022-09-05-13-00-00",
"uri": None,
"id": 1,
"show_name": "Show 3",
"fade_in": 500.0,
"fade_out": 500.0,
"cue_in": 8.25245,
"cue_out": 95.0,
"metadata": {
"track_title": "The Dane",
"artist_name": "Nils Frahm",
"mime": "audio/flac",
},
"replay_gain": "4.52",
"filesize": 10000,
},
"2022-09-05-12-59-58": {
"type": EventKind.EVENT,
"event_type": "switch_off",
"start": "2022-09-05-12-59-58",
"end": "2022-09-05-12-59-58",
},
"2022-09-05-13-00-00": {
"type": EventKind.EVENT,
"event_type": "kick_out",
"start": "2022-09-05-13-00-00",
"end": "2022-09-05-13-00-00",
},
"2022-09-05-13-00-00_0": {
"type": EventKind.FILE,
"row_id": 9,
"start": "2022-09-05-13-00-00",
"end": "2022-09-05-13-05-02",
"uri": None,
"id": 2,
"show_name": "Show 4",
"fade_in": 500.0,
"fade_out": 500.0,
"cue_in": 13.7008,
"cue_out": 315.845,
"metadata": {
"track_title": "My Friend the Forest",
"artist_name": "Nils Frahm",
"mime": "audio/flac",
},
"replay_gain": "11.46",
"filesize": 10000,
},
"2022-09-05-13-05-02": {
"type": EventKind.FILE,
"row_id": 10,
"start": "2022-09-05-13-05-02",
"end": "2022-09-05-13-10-00",
"uri": None,
"id": 4,
"show_name": "Show 4",
"fade_in": 500.0,
"fade_out": 500.0,
"cue_in": 0.0,
"cue_out": 297.8558,
"metadata": {
"track_title": "#2",
"artist_name": "Nils Frahm",
"mime": "audio/flac",
},
"replay_gain": "-1.65",
"filesize": 10000,
},
}
}