cli tests

README.rst

@@ -14,7 +14,7 @@ pytube
   :alt: Documentation Status
   :target: http://python-pytube.readthedocs.io/en/latest/?badge=latest
 
-.. image:: https://coveralls.io/repos/github/nficano/pytube/badge.svg?branch=master#23e6f7ac5623e
+.. image:: https://coveralls.io/repos/github/nficano/pytube/badge.svg?branch=master#23e6f7ac56dd3e
   :alt: Code Coverage
   :target: https://coveralls.io/github/nficano/pytube?branch=master
 

pytube/cli.py

@@ -12,7 +12,6 @@ import os
 import sys
 
 from pytube import __version__
-from pytube import request
 from pytube import YouTube
 
 
@@ -78,9 +77,9 @@ def build_playback_report(url):
         'yt-video-{yt.video_id}-{ts}.json.tar.gz'.format(yt=yt, ts=ts),
     )
 
-    js = request.get(yt.js_url)
-    watch_html = request.get(yt.watch_html)
-    vid_info = request.get(yt.vid_info_url)
+    js = yt.js
+    watch_html = yt.watch_html
+    vid_info = yt.vid_info
 
     with gzip.open(fp, 'wb') as fh:
         fh.write(

tests/test_cli.py

@@ -0,0 +1,13 @@
+# -*- coding: utf-8 -*-
+import mock
+
+from pytube import cli
+
+
+@mock.patch('pytube.cli.YouTube')
+@mock.patch('pytube.cli.sys')
+def test_download(MockYouTube, mock_sys):
+    instance = MockYouTube.return_value
+    instance.prefetch_init.return_value = None
+    instance.streams = mock.Mock()
+    cli.download('asdf', 'asdf')