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",
]
