|
|
|
from unittest import mock |
|
|
|
import pytest |
|
|
|
from pytube import YouTube |
|
from pytube.exceptions import VideoUnavailable |
|
|
|
|
|
@mock.patch("pytube.__main__.YouTube") |
|
def test_prefetch_deferred(youtube): |
|
instance = youtube.return_value |
|
instance.prefetch_descramble.return_value = None |
|
YouTube("https://www.youtube.com/watch?v=9bZkp7q19f0", True) |
|
assert not instance.prefetch_descramble.called |
|
|
|
|
|
@mock.patch("urllib.request.install_opener") |
|
def test_install_proxy(opener): |
|
proxies = {"http": "http://www.example.com:3128/"} |
|
YouTube( |
|
"https://www.youtube.com/watch?v=9bZkp7q19f0", |
|
defer_prefetch_init=True, |
|
proxies=proxies, |
|
) |
|
opener.assert_called() |
|
|
|
|
|
@mock.patch("pytube.request.get") |
|
def test_video_unavailable(get): |
|
get.return_value = None |
|
youtube = YouTube( |
|
"https://www.youtube.com/watch?v=9bZkp7q19f0", defer_prefetch_init=True |
|
) |
|
with pytest.raises(VideoUnavailable): |
|
youtube.prefetch() |
|
|
