added cli test
Browse files- tests/contrib/test_playlist.py +1 -1
- tests/test_cli.py +10 -0
tests/contrib/test_playlist.py
CHANGED
|
@@ -88,7 +88,7 @@ def test_video_urls(request_get, playlist_html):
|
|
| 88 |
|
| 89 |
@mock.patch("pytube.contrib.playlist.request.get")
|
| 90 |
@mock.patch("pytube.cli.YouTube.__init__", return_value=None)
|
| 91 |
-
def
|
| 92 |
url = "https://www.fakeurl.com/playlist?list=whatever"
|
| 93 |
request_get.return_value = playlist_html
|
| 94 |
playlist = Playlist(url)
|
|
|
|
| 88 |
|
| 89 |
@mock.patch("pytube.contrib.playlist.request.get")
|
| 90 |
@mock.patch("pytube.cli.YouTube.__init__", return_value=None)
|
| 91 |
+
def test_videos(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)
|
tests/test_cli.py
CHANGED
|
@@ -177,3 +177,13 @@ def test_download_by_resolution(youtube):
|
|
| 177 |
cli.main()
|
| 178 |
youtube.assert_called()
|
| 179 |
cli.download_by_resolution.assert_called()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 177 |
cli.main()
|
| 178 |
youtube.assert_called()
|
| 179 |
cli.download_by_resolution.assert_called()
|
| 180 |
+
|
| 181 |
+
|
| 182 |
+
@mock.patch("pytube.cli.Playlist")
|
| 183 |
+
def test_download_by_resolution(playlist):
|
| 184 |
+
parser = argparse.ArgumentParser()
|
| 185 |
+
args = parse_args(parser, ["https://www.youtube.com/playlist?list=PLyn", "-r", "320p"])
|
| 186 |
+
cli._parse_args = MagicMock(return_value=args)
|
| 187 |
+
cli.download_by_resolution = MagicMock()
|
| 188 |
+
cli.main()
|
| 189 |
+
playlist.assert_called()
|