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()