added test_print_available_captions

pytube/query.py

@@ -397,7 +397,7 @@ class CaptionQuery(Mapping):
         return len(self.lang_code_index)
 
     def __iter__(self):
-        return iter(self.lang_code_index)
+        return iter(self.lang_code_index.values())
 
     def __repr__(self) -> str:
         return f"{self.lang_code_index}"

tests/test_captions.py

@@ -26,7 +26,8 @@ def test_caption_query_sequence():
     assert caption_query["en"] == caption1
     assert caption_query["fr"] == caption2
     with pytest.raises(KeyError):
-        caption_query["nada"]
+        not_exists = caption_query["nada"]
+        assert not_exists is not None  # should never reach this
 
 
 def test_caption_query_get_by_language_code_when_exists():

tests/test_cli.py

@@ -100,6 +100,22 @@ def test_download_caption_with_lang_not_found(youtube, print_available):
     print_available.assert_called_with(youtube.captions)
 
 
+def test_print_available_captions(capsys):
+    # Given
+    caption1 = Caption(
+        {"url": "url1", "name": {"simpleText": "name1"}, "languageCode": "en"}
+    )
+    caption2 = Caption(
+        {"url": "url2", "name": {"simpleText": "name2"}, "languageCode": "fr"}
+    )
+    query = CaptionQuery([caption1, caption2])
+    # When
+    cli._print_available_captions(query)
+    # Then
+    captured = capsys.readouterr()
+    assert captured.out == "Available caption codes are: en, fr\n"
+
+
 def test_display_progress_bar(capsys):
     cli.display_progress_bar(bytes_received=25, filesize=100, scale=0.55)
     out, _ = capsys.readouterr()

tests/test_query.py

@@ -44,14 +44,14 @@ def test_get_last(cipher_signature):
     """Ensure :meth:`~pytube.StreamQuery.last` returns the expected
     :class:`Stream <Stream>`.
     """
-    assert cipher_signature.streams.last().itag == 251
+    assert cipher_signature.streams[-1].itag == 251
 
 
 def test_get_first(cipher_signature):
     """Ensure :meth:`~pytube.StreamQuery.first` returns the expected
     :class:`Stream <Stream>`.
     """
-    assert cipher_signature.streams.first().itag == 18
+    assert cipher_signature.streams[0].itag == 18
 
 
 def test_order_by(cipher_signature):
@@ -154,10 +154,10 @@ def test_sequence(cipher_signature):
 
 
 def test_otf(cipher_signature):
-    non_otf = cipher_signature.streams.otf().all()
+    non_otf = cipher_signature.streams.otf()
     assert len(non_otf) == 22
 
-    otf = cipher_signature.streams.otf(True).all()
+    otf = cipher_signature.streams.otf(True)
     assert len(otf) == 0
 
 

tests/test_streams.py

@@ -11,13 +11,13 @@ from pytube import Stream, streams
 def test_filesize(cipher_signature, mocker):
     mocker.patch.object(request, "head")
     request.head.return_value = {"content-length": "6796391"}
-    assert cipher_signature.streams.first().filesize == 6796391
+    assert cipher_signature.streams[0].filesize == 6796391
 
 
 def test_filesize_approx(cipher_signature, mocker):
     mocker.patch.object(request, "head")
     request.head.return_value = {"content-length": "123"}
-    stream = cipher_signature.streams.first()
+    stream = cipher_signature.streams[0]
     assert stream.filesize_approx == 22350604
     stream.bitrate = None
     assert stream.filesize_approx == 123
@@ -25,7 +25,7 @@ def test_filesize_approx(cipher_signature, mocker):
 
 def test_default_filename(cipher_signature):
     expected = "PSY - GANGNAM STYLE(강남스타일) MV.mp4"
-    stream = cipher_signature.streams.first()
+    stream = cipher_signature.streams[0]
     assert stream.default_filename == expected
 
 
@@ -103,7 +103,7 @@ def test_download(cipher_signature, mocker):
     mocker.patch.object(request, "stream")
     request.stream.return_value = iter([str(random.getrandbits(8 * 1024))])
     with mock.patch("pytube.streams.open", mock.mock_open(), create=True):
-        stream = cipher_signature.streams.first()
+        stream = cipher_signature.streams[0]
         stream.download()
 
 
@@ -114,7 +114,7 @@ def test_download_with_prefix(cipher_signature, mocker):
     request.stream.return_value = iter([str(random.getrandbits(8 * 1024))])
     streams.target_directory = MagicMock(return_value="/target")
     with mock.patch("pytube.streams.open", mock.mock_open(), create=True):
-        stream = cipher_signature.streams.first()
+        stream = cipher_signature.streams[0]
         file_path = stream.download(filename_prefix="prefix")
         assert file_path == "/target/prefixPSY - GANGNAM STYLE(강남스타일) MV.mp4"
 
@@ -126,7 +126,7 @@ def test_download_with_filename(cipher_signature, mocker):
     request.stream.return_value = iter([str(random.getrandbits(8 * 1024))])
     streams.target_directory = MagicMock(return_value="/target")
     with mock.patch("pytube.streams.open", mock.mock_open(), create=True):
-        stream = cipher_signature.streams.first()
+        stream = cipher_signature.streams[0]
         file_path = stream.download(filename="cool name bro")
         assert file_path == "/target/cool name bro.mp4"
 
@@ -139,7 +139,7 @@ def test_download_with_existing(cipher_signature, mocker):
     mocker.patch.object(os.path, "isfile")
     os.path.isfile.return_value = True
     with mock.patch("pytube.streams.open", mock.mock_open(), create=True):
-        stream = cipher_signature.streams.first()
+        stream = cipher_signature.streams[0]
         mocker.patch.object(os.path, "getsize")
         os.path.getsize.return_value = stream.filesize
         file_path = stream.download()
@@ -156,7 +156,7 @@ def test_download_with_existing_no_skip(cipher_signature, mocker):
     mocker.patch.object(os.path, "isfile")
     os.path.isfile.return_value = True
     with mock.patch("pytube.streams.open", mock.mock_open(), create=True):
-        stream = cipher_signature.streams.first()
+        stream = cipher_signature.streams[0]
         mocker.patch.object(os.path, "getsize")
         os.path.getsize.return_value = stream.filesize
         file_path = stream.download(skip_existing=False)
@@ -165,12 +165,12 @@ def test_download_with_existing_no_skip(cipher_signature, mocker):
 
 
 def test_progressive_streams_return_includes_audio_track(cipher_signature):
-    stream = cipher_signature.streams.filter(progressive=True).first()
+    stream = cipher_signature.streams.filter(progressive=True)[0]
     assert stream.includes_audio_track
 
 
 def test_progressive_streams_return_includes_video_track(cipher_signature):
-    stream = cipher_signature.streams.filter(progressive=True).first()
+    stream = cipher_signature.streams.filter(progressive=True)[0]
     assert stream.includes_video_track
 
 
@@ -184,7 +184,7 @@ def test_on_progress_hook(cipher_signature, mocker):
     request.stream.return_value = iter([str(random.getrandbits(8 * 1024))])
 
     with mock.patch("pytube.streams.open", mock.mock_open(), create=True):
-        stream = cipher_signature.streams.first()
+        stream = cipher_signature.streams[0]
         stream.download()
     assert callback_fn.called
     args, _ = callback_fn.call_args
@@ -203,7 +203,7 @@ def test_on_complete_hook(cipher_signature, mocker):
     request.stream.return_value = iter([str(random.getrandbits(8 * 1024))])
 
     with mock.patch("pytube.streams.open", mock.mock_open(), create=True):
-        stream = cipher_signature.streams.first()
+        stream = cipher_signature.streams[0]
         stream.download()
     assert callback_fn.called
 
@@ -233,7 +233,7 @@ def test_thumbnail_when_not_in_details(cipher_signature):
 
 
 def test_repr_for_audio_streams(cipher_signature):
-    stream = str(cipher_signature.streams.filter(only_audio=True).first())
+    stream = str(cipher_signature.streams.filter(only_audio=True)[0])
     expected = (
         '<Stream: itag="140" mime_type="audio/mp4" abr="128kbps" '
         'acodec="mp4a.40.2" progressive="False" type="audio">'
@@ -242,7 +242,7 @@ def test_repr_for_audio_streams(cipher_signature):
 
 
 def test_repr_for_video_streams(cipher_signature):
-    stream = str(cipher_signature.streams.filter(only_video=True).first())
+    stream = str(cipher_signature.streams.filter(only_video=True)[0])
     expected = (
         '<Stream: itag="137" mime_type="video/mp4" res="1080p" fps="30fps" '
         'vcodec="avc1.640028" progressive="False" type="video">'
@@ -251,7 +251,7 @@ def test_repr_for_video_streams(cipher_signature):
 
 
 def test_repr_for_progressive_streams(cipher_signature):
-    stream = str(cipher_signature.streams.filter(progressive=True).first())
+    stream = str(cipher_signature.streams.filter(progressive=True)[0])
     expected = (
         '<Stream: itag="18" mime_type="video/mp4" res="360p" fps="30fps" '
         'vcodec="avc1.42001E" acodec="mp4a.40.2" progressive="True" type="video">'
@@ -260,7 +260,7 @@ def test_repr_for_progressive_streams(cipher_signature):
 
 
 def test_repr_for_adaptive_streams(cipher_signature):
-    stream = str(cipher_signature.streams.filter(adaptive=True).first())
+    stream = str(cipher_signature.streams.filter(adaptive=True)[0])
     expected = (
         '<Stream: itag="137" mime_type="video/mp4" res="1080p" fps="30fps" '
         'vcodec="avc1.640028" progressive="False" type="video">'