|
|
|
import os |
|
|
|
from unittest import mock |
|
|
|
from pytube import request |
|
|
|
|
|
@mock.patch("pytube.request.urlopen") |
|
def test_get_streaming(mock_urlopen): |
|
fake_stream_binary = [ |
|
iter(os.urandom(8 * 1024)), |
|
iter(os.urandom(8 * 1024)), |
|
iter(os.urandom(8 * 1024)), |
|
None, |
|
] |
|
response = mock.Mock() |
|
response.read.side_effect = fake_stream_binary |
|
mock_urlopen.return_value = response |
|
response = request.get("http://fakeassurl.gov", streaming=True) |
|
call_count = 0 |
|
for i in response: |
|
call_count += 1 |
|
assert call_count == 3 |
|
|
|
|
|
@mock.patch("pytube.request.urlopen") |
|
def test_get_headers(mock_urlopen): |
|
response = mock.Mock() |
|
response.info.return_value = {"content-length": "16384"} |
|
mock_urlopen.return_value = response |
|
response = request.get("http://fakeassurl.gov", headers=True) |
|
assert response == {"content-length": "16384"} |
|
|
|
|
|
@mock.patch("pytube.request.urlopen") |
|
def test_get(mock_urlopen): |
|
response = mock.Mock() |
|
response.read.return_value = "<html></html>".encode("utf-8") |
|
mock_urlopen.return_value = response |
|
response = request.get("http://fakeassurl.gov") |
|
assert response == "<html></html>" |
|
|
