added unit tests

pytube/cli.py

@@ -3,6 +3,7 @@
 from __future__ import print_function
 
 import argparse
+import json
 import os
 import sys
 
@@ -24,16 +25,37 @@ def main():
             'available to download'
         ),
     )
+    parser.add_argument(
+        '--build-debug-report', action='store_true', help=(
+            'Save the html and js to disk'
+        ),
+    )
     args = parser.parse_args()
     if not args.url:
         parser.print_help()
         sys.exit(1)
     if args.list:
         display_streams(args.url)
+    elif args.build_debug_report:
+        build_debug_report(args.url)
     elif args.itag:
         download(args.url, args.itag)
 
 
+def build_debug_report(url):
+    yt = YouTube(url)
+    fp = os.path.join(
+        os.getcwd(),
+        'yt-video-{yt.video_id}.json'.format(yt=yt),
+    )
+    with open(fp, 'w') as fh:
+        fh.write(json.dumps({
+            'js': yt.js,
+            'watch_html': yt.watch_html,
+            'video_info': yt.vid_info,
+        }))
+
+
 def get_terminal_size():
     """Return the terminal size in rows and columns."""
     rows, columns = os.popen('stty size', 'r').read().split()

setup.cfg

@@ -12,3 +12,7 @@ description-file = README.md
 [bumpversion:file:setup.py]
 
 [bumpversion:file:pytube/__init__.py]
+
+
+[coverage:run]
+source = pytube

tests/__init__.py

@@ -0,0 +1,2 @@
+# -*- coding: utf-8 -*-
+"""Package init | pydocstyle ignore isn't working :/."""

tests/conftest.py

@@ -0,0 +1,27 @@
+# -*- coding: utf-8 -*-
+"""Reusable dependency injected testing components."""
+import json
+import os
+
+import pytest
+
+from pytube import YouTube
+
+
+@pytest.fixture
+def gangnam_style():
+    """Youtube instance initialized with video id 9bZkp7q19f0."""
+    cur_dir = os.path.dirname(os.path.realpath(__file__))
+    fp = os.path.join(cur_dir, 'mocks', 'yt-video-9bZkp7q19f0.json')
+    video = None
+    with open(fp, 'r') as fh:
+        video = json.loads(fh.read())
+    yt = YouTube(
+        url='https://www.youtube.com/watch?v=9bZkp7q19f0',
+        defer_init=True,
+    )
+    yt.watch_html = video['watch_html']
+    yt.js = video['js']
+    yt.vid_info = video['video_info']
+    yt.init()
+    return yt

tests/test_extract.py

@@ -1,10 +1,5 @@
 # -*- coding: utf-8 -*-
-"""
-tests.test_extract
-~~~~~~~~~~~~~~~~~~
-
-Unit tests for the :module:`extract <extract>` module.
-"""
+"""Unit tests for the :module:`extract <extract>` module."""
 from pytube import extract
 
 
@@ -17,4 +12,18 @@ def test_extract_video_id():
 def test_extract_watch_url():
     video_id = '9bZkp7q19f0'
     watch_url = extract.watch_url(video_id)
-    assert watch_url == 'https://www.youtube.com/watch?v=9bZkp7q19f0'
+    assert watch_url == 'https://youtube.com/watch?v=9bZkp7q19f0'
+
+
+def test_info_url(gangnam_style):
+    video_info_url = extract.video_info_url(
+        video_id=gangnam_style.video_id,
+        watch_url=gangnam_style.watch_url,
+        watch_html=gangnam_style.watch_html,
+    )
+    expected = (
+        'https://youtube.com/get_video_info?video_id=9bZkp7q19f0&el=%24el'
+        '&ps=default&eurl=https%253A%2F%2Fyoutube.com%2Fwatch%253Fv%'
+        '253D9bZkp7q19f0&hl=en_US&t=%252C%2522t%2522%253A%25221%2522'
+    )
+    assert video_info_url == expected