ffplayout/tests/run_multiple_tests.py
jb-alvarado e3b0375ca4 cleanup
2022-01-01 23:08:56 +01:00

130 lines
2.9 KiB
Python
Executable File

#!/usr/bin/env python3
"""
Test script, for testing different situations, like:
- different day_start times
- different situ where playlist is empty, not long enough or to long
"""
import json
import os
import sys
from datetime import datetime
from threading import Thread
from time import sleep
from unittest.mock import patch
from zoneinfo import ZoneInfo
import time_machine
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
# from ffplayout import playlist
# set time zone
_TZ = ZoneInfo("Europe/Berlin")
def run_at(time_tuple):
dt = datetime(*time_tuple, tzinfo=_TZ).strftime("%Y-%m-%d %H:%M:%S")
@time_machine.travel(dt)
def run_in_time_machine():
desktop.output()
print(f'simulated date and time: {dt}\n')
run_in_time_machine()
def run_time(seconds):
"""
validate json values in new thread
and test if source paths exist
"""
def timer(seconds):
print(f'run test for {seconds} seconds...')
sleep(seconds)
terminate_processes()
print('terminated successfully')
terminator = Thread(name='timer', target=timer, args=(seconds,))
terminator.daemon = True
terminator.start()
def print_separater():
print('\n')
print(79 * '-')
print(79 * '-')
def shorten_playlist(file):
json_object = json.load(file)
del json_object['program'][-1:]
return json_object
def extend_playlist(file):
json_object = json.load(file)
elems = json_object['program'][:2]
json_object['program'].extend(elems)
return json_object
def clear_playlist(file):
return {}
@patch('ffplayout.playlist.valid_json', shorten_playlist)
def run_with_less_elements(time_tuple):
run_at(time_tuple)
@patch('ffplayout.playlist.valid_json', extend_playlist)
def run_with_more_elements(time_tuple):
run_at(time_tuple)
@patch('ffplayout.playlist.valid_json', clear_playlist)
def run_with_no_elements(time_tuple):
run_at(time_tuple)
if __name__ == '__main__':
from ffplayout.output import desktop
from ffplayout.utils import playlist, terminate_processes
print('\ntest playlists, which are empty')
playlist.start = 0
run_time(140)
run_with_no_elements((2021, 2, 15, 23, 59, 53))
print_separater()
print('\ntest playlists, which are to short')
playlist.start = 0
run_time(140)
run_with_less_elements((2021, 2, 15, 23, 58, 3))
print_separater()
print('\ntest playlists, which are to long')
playlist.start = 0
run_time(140)
run_with_more_elements((2021, 2, 15, 23, 59, 33))
print_separater()
print('\ntest transition from playlists, with day_start at: 05:59:25')
playlist.start = 21575
run_time(140)
run_at((2021, 2, 17, 5, 58, 3))
print_separater()
print('\ntest transition from playlists, with day_start at: 20:00:00')
playlist.start = 72000
run_time(140)
run_at((2021, 2, 17, 19, 58, 23))