|
from unittest import mock |
|
|
|
import pytest |
|
|
|
import pytube |
|
from pytube import YouTube |
|
from pytube.exceptions import RegexMatchError |
|
|
|
|
|
@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", |
|
proxies=proxies, |
|
) |
|
opener.assert_called() |
|
|
|
|
|
@mock.patch("pytube.request.get") |
|
def test_video_unavailable(get): |
|
get.return_value = "" |
|
youtube = YouTube("https://www.youtube.com/watch?v=9bZkp7q19f0") |
|
with pytest.raises(RegexMatchError): |
|
youtube.check_availability() |
|
|
|
|
|
def test_video_keywords(cipher_signature): |
|
expected = [ |
|
'Rewind', 'Rewind 2019', |
|
'youtube rewind 2019', '#YouTubeRewind', |
|
'MrBeast', 'PewDiePie', 'James Charles', |
|
'Shane Dawson', 'CaseyNeistat', 'RiceGum', |
|
'Simone Giertz', 'JennaMarbles', 'Lilly Singh', |
|
'emma chamberlain', 'The Try Guys', 'Fortnite', |
|
'Minecraft', 'Roblox', 'Marshmello', |
|
'Garena Free Fire', 'GTA V', 'Lachlan', |
|
'Anaysa', 'jeffreestar', 'Noah Schnapp', |
|
'Jennelle Eliana', 'T-Series', 'Azzyland', |
|
'LazarBeam', 'Dude Perfect', 'David Dobrik', |
|
'KSI', 'NikkieTutorials', 'Kurzgesagt', |
|
'Jelly', 'Ariana Grande', 'Billie Eilish', |
|
'BLACKPINK', 'Year in Review' |
|
] |
|
assert cipher_signature.keywords == expected |
|
|
|
|
|
def test_js_caching(cipher_signature): |
|
assert pytube.__js__ is not None |
|
assert pytube.__js_url__ is not None |
|
assert pytube.__js__ == cipher_signature.js |
|
assert pytube.__js_url__ == cipher_signature.js_url |
|
|
|
|
|
def test_channel_id(cipher_signature): |
|
assert cipher_signature.channel_id == 'UCBR8-60-B28hp2BmDPdntcQ' |
|
|
|
|
|
def test_channel_url(cipher_signature): |
|
assert cipher_signature.channel_url == 'https://www.youtube.com/channel/UCBR8-60-B28hp2BmDPdntcQ' |
|
|
