def test_download_audio_none(youtube, download): # Given youtube_instance = youtube.return_value youtube_instance.streams.filter.return_value.order_by.return_value.last.return_value = ( None) # When with pytest.raises(SystemExit): cli.download_audio(youtube_instance, "filetype", "target") # Then download.assert_not_called()
def test_download_audio(youtube, download): # Given youtube_instance = youtube.return_value audio_stream = MagicMock() youtube_instance.streams.filter.return_value.order_by.return_value.last.return_value = ( audio_stream) # When cli.download_audio(youtube_instance, "filetype", "target") # Then download.assert_called_with(audio_stream, target="target")