File size: 2,538 Bytes
d2c119a cd0bfbf d2c119a 0bbb369 b4ef682 d2c119a 0bbb369 d2c119a 82321d6 846239d 5afc31c 82321d6 5afc31c 9fb6343 25de36d 9fb6343 0bbb369 b043411 b4ef682 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 |
# -*- coding: utf-8 -*-
from unittest import mock
import pytest
import pytube
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()
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
with mock.patch('pytube.request.urlopen') as mock_urlopen:
mock_urlopen_object = mock.Mock()
# We should never read the js from this
mock_urlopen_object.read.side_effect = [
cipher_signature.watch_html.encode('utf-8'),
cipher_signature.vid_info_raw.encode('utf-8'),
cipher_signature.js.encode('utf-8')
]
mock_urlopen.return_value = mock_urlopen_object
cipher_signature.prefetch()
assert mock_urlopen.call_count == 2
|