def test_videos(youtube, request_get, playlist_html): url = "https://www.fakeurl.com/playlist?list=whatever" request_get.return_value = playlist_html playlist = Playlist(url) playlist._find_load_more_url = MagicMock(return_value=None) request_get.assert_called() assert len(list(playlist.videos)) == 12
def test_sequence(request_get, playlist_html): url = "https://www.fakeurl.com/playlist?list=whatever" request_get.return_value = playlist_html playlist = Playlist(url) playlist._find_load_more_url = MagicMock(return_value=None) assert playlist[0] == "https://www.youtube.com/watch?v=ujTCoH21GlA" assert len(playlist) == 12
def test_load_more(youtube, request_get, playlist_html): url = "https://www.fakeurl.com/playlist?list=whatever" request_get.side_effect = [ playlist_html, '{"content_html":"", "load_more_widget_html":""}', ] playlist = Playlist(url) playlist._find_load_more_url = MagicMock(side_effect=["dummy", None]) request_get.assert_called() assert len(list(playlist.videos)) == 12
def test_trimmed(request_get, playlist_html): url = "https://www.fakeurl.com/playlist?list=whatever" request_get.return_value = playlist_html playlist = Playlist(url) playlist._find_load_more_url = MagicMock(return_value=None) assert request_get.call_count == 1 trimmed = list(playlist.trimmed("1BYu65vLKdA")) assert trimmed == [ "https://www.youtube.com/watch?v=ujTCoH21GlA", "https://www.youtube.com/watch?v=45ryDIPHdGg", ]
def test_repr(request_get, playlist_html): url = "https://www.fakeurl.com/playlist?list=whatever" request_get.return_value = playlist_html playlist = Playlist(url) playlist._find_load_more_url = MagicMock(return_value=None) request_get.assert_called() assert ( repr(playlist) == "['https://www.youtube.com/watch?v=ujTCoH21GlA', " "'https://www.youtube.com/watch?v=45ryDIPHdGg', " "'https://www.youtube.com/watch?v=1BYu65vLKdA', " "'https://www.youtube.com/watch?v=3AQ_74xrch8', " "'https://www.youtube.com/watch?v=ddqQUz9mZaM', " "'https://www.youtube.com/watch?v=vwLT6bZrHEE', " "'https://www.youtube.com/watch?v=TQKI0KE-JYY', " "'https://www.youtube.com/watch?v=dNBvQ38MlT8', " "'https://www.youtube.com/watch?v=JHxyrMgOUWI', " "'https://www.youtube.com/watch?v=l2I8NycJMCY', " "'https://www.youtube.com/watch?v=g1Zbuk1gAfk', " "'https://www.youtube.com/watch?v=zixd-si9Q-o']" )
def test_video_urls(request_get, playlist_html): url = "https://www.fakeurl.com/playlist?list=whatever" request_get.return_value = playlist_html playlist = Playlist(url) playlist._find_load_more_url = MagicMock(return_value=None) request_get.assert_called() assert playlist.video_urls == [ "https://www.youtube.com/watch?v=ujTCoH21GlA", "https://www.youtube.com/watch?v=45ryDIPHdGg", "https://www.youtube.com/watch?v=1BYu65vLKdA", "https://www.youtube.com/watch?v=3AQ_74xrch8", "https://www.youtube.com/watch?v=ddqQUz9mZaM", "https://www.youtube.com/watch?v=vwLT6bZrHEE", "https://www.youtube.com/watch?v=TQKI0KE-JYY", "https://www.youtube.com/watch?v=dNBvQ38MlT8", "https://www.youtube.com/watch?v=JHxyrMgOUWI", "https://www.youtube.com/watch?v=l2I8NycJMCY", "https://www.youtube.com/watch?v=g1Zbuk1gAfk", "https://www.youtube.com/watch?v=zixd-si9Q-o", ]