diff --git a/tests/temp_config.py b/tests/temp_config.py new file mode 100644 index 0000000..80abbf1 --- /dev/null +++ b/tests/temp_config.py @@ -0,0 +1,19 @@ +import os + + +class TempConfig: + """Small test configuration object used by unit and integration tests. + + Adjust attributes via environment variables where appropriate. + """ + yt_dlp_path = os.getenv("YTDLP_PATH", "yt-dlp") + ffmpeg_path = os.getenv("FFMPEG_PATH", "ffmpeg") + aria2c_path = os.getenv("ARIA2C_PATH", "aria2c") + max_parallel_downloads = int(os.getenv("TEST_MAX_PARALLEL", "2")) + aria2c_connections = int(os.getenv("TEST_ARIA2C_CONN", "2")) + download_mode = os.getenv("TEST_DOWNLOAD_MODE", "audio") + max_video_quality = os.getenv("TEST_MAX_VIDEO_QUALITY", "1080p") + # runtime flags + debug = False + non_interactive = False + prune = False diff --git a/tests/test_cli_flags.py b/tests/test_cli_flags.py index 09ae655..a149b3f 100644 --- a/tests/test_cli_flags.py +++ b/tests/test_cli_flags.py @@ -1,29 +1,21 @@ import logging from ytplaylist.manager import PlaylistManager +from tests.temp_config import TempConfig -class test: - playlists=[{"url": None, "save_path":"./tmp_test", "archive":"archive.txt"}] - yt_dlp_path="./bin/yt-dlp" - ffmpeg_path="./bin/ffmpeg" - aria2c_path="./bin/ aria2c" - max_parallel_downloads=2 - aria2c_connections=2 - download_mode = "audio" - max_video_quality = "1080p" - debug=False - non_interactive=False - prune=False + +class TestConfig(TempConfig): + playlists = [{"url": None, "save_path": "./tmp_test", "archive": "archive.txt"}] if __name__ == '__main__': logging.basicConfig(level=logging.INFO, format="%(levelname)s:%(message)s") print('--- Running with prune=False ---') - cfg=test() + cfg=TestConfig() m=PlaylistManager(cfg, debug=False) m.run() print('Run complete prune=False') print('\n--- Running with prune=True, non_interactive=True ---') - cfg2=test() + cfg2=TestConfig() cfg2.prune=True cfg2.non_interactive=True m2=PlaylistManager(cfg2, debug=False)