danvics/ytdl-sub
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

hdh

Browse source
This commit is contained in:
Jesse Bannon 2024-09-29 22:57:33 -07:00
parent 9343f39366
commit 01d8aba109
1 changed files with 19 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
tests/unit/plugins/test_music_tags.py
View file

@ -9,37 +9,40 @@ from ytdl_sub.utils.exceptions import ValidationException
@pytest.fixture @pytest.fixture
def single_song_video_dict(output_directory): def single_song_video_dict(output_directory):
return { return {
"download": "https://www.youtube.com/watch?v=2lAe1cqCOXo", "download": "https://your.name.here",
"output_options": {"output_directory": output_directory, "file_name": "will_error.mp4"}, "output_options": {"output_directory": output_directory, "file_name": "will_error.mp4"},
# test multi-tags # test multi-tags compile
"music_tags": {"genres": ["multi_tag_1", "multi_tag_2"]}, "music_tags": {"genres": ["multi_tag_1", "multi_tag_2"]},
"format": "worst[ext=mp4]", # download the worst format so it is fast
"ytdl_options": {
"postprocessor_args": {"ffmpeg": ["-bitexact"]}, # Must add this for reproducibility
},
} }
class TestMusicTags: class TestMusicTags:
def test_music_tags_errors_on_video( def test_music_tags_errors_on_video(
self, self,
default_config, config,
single_song_video_dict, single_song_video_dict,
output_directory, output_directory,
subscription_name,
mock_download_collection_entries,
): ):
subscription = Subscription.from_dict( subscription = Subscription.from_dict(
config=default_config, config=config,
preset_name="single_song_test", preset_name=subscription_name,
preset_dict=single_song_video_dict, preset_dict=single_song_video_dict,
) )
with pytest.raises( with (
mock_download_collection_entries(
is_youtube_channel=False, num_urls=1, is_extracted_audio=False, is_dry_run=True
),
pytest.raises(
ValidationException, ValidationException,
match=re.escape( match=re.escape(
"Validation error in single_song_test.music_tags: music_tags plugin received a " f"Validation error in {subscription_name}.music_tags: music_tags plugin received a "
"video with the extension 'mp4'. Only audio files are supported for setting music " "video with the extension 'mp4'. Only audio files are supported for setting music "
"tags. Ensure you are converting the video to audio using the audio_extract " "tags. Ensure you are converting the video to audio using the audio_extract "
"plugin." "plugin."
), ),
),
): ):
subscription.download(dry_run=True) subscription.download(dry_run=True)