added playlist tests

tests/conftest.py

@@ -50,3 +50,15 @@ def age_restricted():
     """Youtube instance initialized with video id zRbsm3e2ltw."""
     filename = "yt-video-zRbsm3e2ltw-1507777044.json.gz"
     return load_playback_file(filename)
+
+
+@pytest.fixture
+def playlist_html():
+    """Youtube playlist HTML loaded on 2020-01-25 from
+    https://www.youtube.com/playlist?list=PLzMcBGfZo4-mP7qA9cagf68V06sko5otr"""
+    file_path = os.path.join(
+        os.path.dirname(os.path.realpath(__file__)), "mocks", "playlist.html"
+    )
+    with open(file_path, encoding="utf-8") as f:
+        read_data = f.read()
+    return read_data

tests/contrib/test_playlist.py

@@ -1,5 +1,6 @@
 # -*- coding: utf-8 -*-
 from unittest import mock
+from unittest.mock import MagicMock
 
 from pytube import Playlist
 
@@ -32,3 +33,51 @@ def test_init_with_watch_url():
         playlist.playlist_url
         == "https://www.youtube.com/playlist?list=PLynhp4cZEpTbRs_PYISQ8v_uwO0_mDg_X"
     )
+
+
+@mock.patch("pytube.contrib.playlist.request.get")
+def test_parse_links(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)
+    links = playlist.parse_links()
+    request_get.assert_called()
+    assert links == [
+        "/watch?v=ujTCoH21GlA",
+        "/watch?v=45ryDIPHdGg",
+        "/watch?v=1BYu65vLKdA",
+        "/watch?v=3AQ_74xrch8",
+        "/watch?v=ddqQUz9mZaM",
+        "/watch?v=vwLT6bZrHEE",
+        "/watch?v=TQKI0KE-JYY",
+        "/watch?v=dNBvQ38MlT8",
+        "/watch?v=JHxyrMgOUWI",
+        "/watch?v=l2I8NycJMCY",
+        "/watch?v=g1Zbuk1gAfk",
+        "/watch?v=zixd-si9Q-o",
+    ]
+
+
+@mock.patch("pytube.contrib.playlist.request.get")
+def test_populate_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)
+    playlist.populate_video_urls()
+    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",
+    ]

tests/test_cli.py

@@ -161,7 +161,7 @@ def test_main_download_caption(youtube):
 @mock.patch("pytube.cli.YouTube.__init__", return_value=None)
 def test_download_by_resolution(youtube):
     parser = argparse.ArgumentParser()
-    args = parse_args(parser, ["urlhere", "-r", "720p"])
+    args = parse_args(parser, ["urlhere", "-r", "320p"])
     cli._parse_args = MagicMock(return_value=args)
     cli.download_by_resolution = MagicMock()
     cli.main()