added test_stream_to_buffer

tests/test_request.py

@@ -10,6 +10,7 @@ from pytube import request
 @mock.patch("pytube.request.filesize", return_value=3 * 8 * 1024)
 @mock.patch("pytube.request.urlopen")
 def test_streaming(mock_urlopen, filesize):
+    # Given
     fake_stream_binary = [
         os.urandom(8 * 1024),
         os.urandom(8 * 1024),
@@ -19,9 +20,10 @@ def test_streaming(mock_urlopen, filesize):
     response = mock.Mock()
     response.read.side_effect = fake_stream_binary
     mock_urlopen.return_value = response
+    # When
     response = request.stream("http://fakeassurl.gov")
+    # Then
     call_count = len(list(response))
-
     assert call_count == 3
 
 

tests/test_streams.py

@@ -8,6 +8,24 @@ from pytube import request
 from pytube import Stream, streams
 
 
+@mock.patch("pytube.streams.request")
+def test_stream_to_buffer(mock_request, cipher_signature):
+    # Given
+    stream_bytes = iter(
+        [
+            bytes(os.urandom(8 * 1024)),
+            bytes(os.urandom(8 * 1024)),
+            bytes(os.urandom(8 * 1024)),
+        ]
+    )
+    mock_request.stream.return_value = stream_bytes
+    buffer = MagicMock()
+    # When
+    cipher_signature.streams[0].stream_to_buffer(buffer)
+    # Then
+    assert buffer.write.call_count == 3
+
+
 def test_filesize(cipher_signature, mocker):
     mocker.patch.object(request, "head")
     request.head.return_value = {"content-length": "6796391"}