def test_ffmpeg_process_res_none_should_not_download(_ffmpeg_downloader, youtube):
# Given
target = "/target"
streams = MagicMock()
youtube.streams = streams
streams.filter.return_value.first.return_value = None
audio_stream = MagicMock()
streams.get_audio_only.return_value = audio_stream
# When
with pytest.raises(SystemExit):
cli.ffmpeg_process(youtube, "XYZp", target)
# Then
_ffmpeg_downloader.assert_not_called()
def test_ffmpeg_process_audio_none_should_fallback_download( # noqa: PT019
_ffmpeg_downloader, youtube):
# Given
target = "/target"
streams = MagicMock()
youtube.streams = streams
stream = MagicMock()
streams.filter.return_value.order_by.return_value.last.return_value = stream
streams.get_audio_only.return_value = None
# When
cli.ffmpeg_process(youtube, "best", target)
# Then
_ffmpeg_downloader.assert_called_with(audio_stream=stream,
video_stream=stream,
target=target)
def test_ffmpeg_process_res_should_download(_ffmpeg_downloader, youtube):
# Given
target = "/target"
streams = MagicMock()
youtube.streams = streams
video_stream = MagicMock()
streams.filter.return_value.first.return_value = video_stream
audio_stream = MagicMock()
streams.get_audio_only.return_value = audio_stream
# When
cli.ffmpeg_process(youtube, "XYZp", target)
# Then
_ffmpeg_downloader.assert_called_with(
audio_stream=audio_stream, video_stream=video_stream, target=target
)
def test_ffmpeg_process_audio_fallback_none_should_exit(_ffmpeg_downloader, youtube):
# Given
target = "/target"
streams = MagicMock()
youtube.streams = streams
stream = MagicMock()
streams.filter.return_value.order_by.return_value.last.side_effect = [
stream,
stream,
None,
]
streams.get_audio_only.return_value = None
# When
with pytest.raises(SystemExit):
cli.ffmpeg_process(youtube, "best", target)
# Then
_ffmpeg_downloader.assert_not_called()
