test playlist.videos
Browse files- pytube/contrib/playlist.py +0 -2
- tests/contrib/test_playlist.py +11 -0
pytube/contrib/playlist.py
CHANGED
@@ -88,8 +88,6 @@ class Playlist:
|
|
88 |
|
89 |
@property
|
90 |
def videos(self) -> Iterable[YouTube]:
|
91 |
-
if not self.video_urls:
|
92 |
-
self.populate_video_urls()
|
93 |
for url in self.video_urls:
|
94 |
yield YouTube(url)
|
95 |
|
|
|
88 |
|
89 |
@property
|
90 |
def videos(self) -> Iterable[YouTube]:
|
|
|
|
|
91 |
for url in self.video_urls:
|
92 |
yield YouTube(url)
|
93 |
|
tests/contrib/test_playlist.py
CHANGED
@@ -84,3 +84,14 @@ def test_video_urls(request_get, playlist_html):
|
|
84 |
"https://www.youtube.com/watch?v=g1Zbuk1gAfk",
|
85 |
"https://www.youtube.com/watch?v=zixd-si9Q-o",
|
86 |
]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
84 |
"https://www.youtube.com/watch?v=g1Zbuk1gAfk",
|
85 |
"https://www.youtube.com/watch?v=zixd-si9Q-o",
|
86 |
]
|
87 |
+
|
88 |
+
|
89 |
+
@mock.patch("pytube.contrib.playlist.request.get")
|
90 |
+
@mock.patch("pytube.cli.YouTube.__init__", return_value=None)
|
91 |
+
def test_video_urls(youtube, request_get, playlist_html):
|
92 |
+
url = "https://www.fakeurl.com/playlist?list=whatever"
|
93 |
+
request_get.return_value = playlist_html
|
94 |
+
playlist = Playlist(url)
|
95 |
+
playlist._find_load_more_url = MagicMock(return_value=None)
|
96 |
+
request_get.assert_called()
|
97 |
+
assert len(list(playlist.videos)) == 12
|