def test_download_when_itag_not_found(youtube): youtube.streams = mock.Mock() youtube.streams.all.return_value = [] youtube.streams.get_by_itag.return_value = None with pytest.raises(SystemExit): cli.download_by_itag(youtube, 123) youtube.streams.get_by_itag.assert_called_with(123)
def test_download_when_itag_not_found(youtube, display_streams): # Given youtube.streams = mock.Mock() youtube.streams.get_by_itag.return_value = None # When with pytest.raises(SystemExit): cli.download_by_itag(youtube, 123) # Then youtube.streams.get_by_itag.assert_called_with(123) display_streams.assert_called_with(youtube)
def test_download_when_itag_is_found(youtube, stream): stream.itag = 123 youtube.streams = StreamQuery([stream]) with patch.object(youtube.streams, "get_by_itag", wraps=youtube.streams.get_by_itag) as wrapped_itag: cli.download_by_itag(youtube, 123) wrapped_itag.assert_called_with(123) youtube.register_on_progress_callback.assert_called_with(cli.on_progress) stream.download.assert_called()