"""Tests for ``app.dl_formats`` format selectors and yt-dlp option mapping.""" from __future__ import annotations import copy import unittest from app.dl_formats import ( _normalize_caption_mode, _normalize_subtitle_language, get_format, get_opts, ) class DlFormatsTests(unittest.TestCase): def test_audio_unknown_format_raises_value_error(self): with self.assertRaises(ValueError): get_format("audio", "auto", "invalid", "best") def test_wav_does_not_enable_thumbnail_postprocessing(self): opts = get_opts("audio", "auto", "wav", "best", {}) self.assertNotIn("writethumbnail", opts) def test_mp3_enables_thumbnail_postprocessing(self): opts = get_opts("audio", "auto", "mp3", "best", {}) self.assertTrue(opts.get("writethumbnail")) def test_custom_format_passthrough(self): self.assertEqual(get_format("video", "auto", "custom:bestvideo+bestaudio", "best"), "bestvideo+bestaudio") def test_thumbnail_and_captions_format_strings(self): self.assertEqual(get_format("thumbnail", "auto", "jpg", "best"), "bestaudio/best") self.assertEqual(get_format("captions", "auto", "srt", "best"), "bestaudio/best") def test_audio_formats(self): for fmt in ("m4a", "mp3", "opus", "wav", "flac"): with self.subTest(fmt=fmt): self.assertIn(f"ext={fmt}", get_format("audio", "auto", fmt, "best")) def test_video_unknown_format_raises(self): with self.assertRaises(ValueError): get_format("video", "auto", "mkv", "best") def test_unknown_download_type_raises(self): with self.assertRaises(ValueError): get_format("unknown", "auto", "any", "best") def test_video_any_mp4_ios_with_height_quality(self): self.assertIn("height<=1080", get_format("video", "auto", "any", "1080")) self.assertNotIn("height<=", get_format("video", "auto", "any", "best")) self.assertNotIn("height<=", get_format("video", "auto", "any", "worst")) def test_video_codec_filters(self): self.assertIn("h264", get_format("video", "h264", "any", "best")) self.assertIn("hevc", get_format("video", "h265", "any", "best")) self.assertIn("av0?1", get_format("video", "av1", "any", "best")) self.assertIn("vp0?9", get_format("video", "vp9", "any", "best")) def test_video_mp4_includes_m4a_audio(self): s = get_format("video", "auto", "mp4", "720") self.assertIn("[ext=m4a]", s) def test_video_ios_selector_contains_avc_pattern(self): s = get_format("video", "auto", "ios", "best") self.assertIn("h26[45]", s) def test_get_opts_deepcopy_does_not_mutate_input(self): base = {"postprocessors": [{"key": "Existing"}]} orig = copy.deepcopy(base) get_opts("audio", "auto", "mp3", "best", base) self.assertEqual(base, orig) def test_get_opts_audio_m4a_postprocessors(self): opts = get_opts("audio", "auto", "m4a", "best", {}) keys = [p["key"] for p in opts["postprocessors"]] self.assertIn("FFmpegExtractAudio", keys) def test_get_opts_audio_mp3_quality_not_best(self): opts = get_opts("audio", "auto", "mp3", "192", {}) ext = next(p for p in opts["postprocessors"] if p["key"] == "FFmpegExtractAudio") self.assertEqual(ext["preferredquality"], "192") def test_get_opts_thumbnail_skip_download(self): opts = get_opts("thumbnail", "auto", "jpg", "best", {}) self.assertTrue(opts.get("skip_download")) self.assertTrue(opts.get("writethumbnail")) def test_get_opts_captions_manual_only(self): opts = get_opts( "captions", "auto", "vtt", "best", {}, subtitle_language="fr", subtitle_mode="manual_only" ) self.assertTrue(opts.get("writesubtitles")) self.assertFalse(opts.get("writeautomaticsub")) self.assertEqual(opts["subtitleslangs"], ["fr", "fr-FR", "fr-CA", "fr-BE", "fr-CH"]) def test_get_opts_captions_auto_only(self): opts = get_opts( "captions", "auto", "srt", "best", {}, subtitle_language="de", subtitle_mode="auto_only" ) self.assertFalse(opts.get("writesubtitles")) self.assertTrue(opts.get("writeautomaticsub")) self.assertEqual(opts["subtitleslangs"], ["de", "de-DE", "de-AT", "de-CH", "de-orig"]) def test_get_opts_captions_prefer_auto(self): opts = get_opts( "captions", "auto", "srt", "best", {}, subtitle_language="es", subtitle_mode="prefer_auto" ) self.assertTrue(opts.get("writesubtitles")) self.assertTrue(opts.get("writeautomaticsub")) self.assertEqual(opts["subtitleslangs"], ["es", "es-ES", "es-MX", "es-AR", "es-CO", "es-CL", "es-PE", "es-VE", "es-orig"]) def test_get_opts_captions_prefer_manual_default_branch(self): opts = get_opts( "captions", "auto", "srt", "best", {}, subtitle_language="it", subtitle_mode="prefer_manual" ) self.assertEqual(opts["subtitleslangs"], ["it", "it-IT", "it-SM", "it-orig"]) def test_get_opts_captions_txt_maps_to_srt_format(self): opts = get_opts("captions", "auto", "txt", "best", {}) self.assertEqual(opts["subtitlesformat"], "srt") def test_get_opts_merges_existing_postprocessors(self): opts = get_opts("audio", "auto", "opus", "best", {"postprocessors": [{"key": "SponsorBlock"}]}) keys = [p["key"] for p in opts["postprocessors"]] self.assertIn("SponsorBlock", keys) self.assertIn("FFmpegExtractAudio", keys) def test_normalize_caption_mode_invalid_defaults(self): self.assertEqual(_normalize_caption_mode(""), "prefer_manual") self.assertEqual(_normalize_caption_mode("not_a_mode"), "prefer_manual") def test_normalize_subtitle_language_empty_defaults_en(self): self.assertEqual(_normalize_subtitle_language(""), "en") self.assertEqual(_normalize_subtitle_language(" "), "en") def test_subtitle_lang_list_with_region_variant_input(self): """zh-Hans should expand to include zh base and zh-CN, zh-TW, etc.""" from app.dl_formats import _subtitle_lang_list result = _subtitle_lang_list("zh-Hans", include_orig=True) self.assertEqual(result[0], "zh-Hans") # original input first self.assertIn("zh", result) # base language fallback self.assertIn("zh-CN", result) # regional variant self.assertIn("zh-orig", result) # orig suffix for auto-captions def test_subtitle_lang_list_dedup_region_variant(self): """pt-BR should not duplicate pt-BR in the variant expansion.""" from app.dl_formats import _subtitle_lang_list result = _subtitle_lang_list("pt-BR", include_orig=True) self.assertEqual(result.count("pt-BR"), 1) # no duplicate self.assertIn("pt", result) # base language fallback self.assertIn("pt-PT", result) # other regional variant if __name__ == "__main__": unittest.main()