mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2024-12-28 22:25:52 +01:00
ad25aee245
Some functions are defined now inside an object, the jsinterp will search its definition if the variable is not defined in the local namespace.
96 lines
2.8 KiB
Python
96 lines
2.8 KiB
Python
#!/usr/bin/env python
|
|
|
|
# Allow direct execution
|
|
import os
|
|
import sys
|
|
import unittest
|
|
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
|
|
|
|
import io
|
|
import re
|
|
import string
|
|
|
|
from youtube_dl.extractor import YoutubeIE
|
|
from youtube_dl.utils import compat_str, compat_urlretrieve
|
|
|
|
_TESTS = [
|
|
(
|
|
u'https://s.ytimg.com/yts/jsbin/html5player-vflHOr_nV.js',
|
|
u'js',
|
|
86,
|
|
u'>=<;:/.-[+*)(\'&%$#"!ZYX0VUTSRQPONMLKJIHGFEDCBA\\yxwvutsrqponmlkjihgfedcba987654321',
|
|
),
|
|
(
|
|
u'https://s.ytimg.com/yts/jsbin/html5player-vfldJ8xgI.js',
|
|
u'js',
|
|
85,
|
|
u'3456789a0cdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRS[UVWXYZ!"#$%&\'()*+,-./:;<=>?@',
|
|
),
|
|
(
|
|
u'https://s.ytimg.com/yts/jsbin/html5player-vfle-mVwz.js',
|
|
u'js',
|
|
90,
|
|
u']\\[@?>=<;:/.-,+*)(\'&%$#"hZYXWVUTSRQPONMLKJIHGFEDCBAzyxwvutsrqponmlkjiagfedcb39876',
|
|
),
|
|
(
|
|
u'https://s.ytimg.com/yts/jsbin/html5player-en_US-vfl0Cbn9e.js',
|
|
u'js',
|
|
84,
|
|
u'O1I3456789abcde0ghijklmnopqrstuvwxyzABCDEFGHfJKLMN2PQRSTUVW@YZ!"#$%&\'()*+,-./:;<=',
|
|
),
|
|
(
|
|
u'https://s.ytimg.com/yts/jsbin/html5player-en_US-vflXGBaUN.js',
|
|
u'js',
|
|
u'2ACFC7A61CA478CD21425E5A57EBD73DDC78E22A.2094302436B2D377D14A3BBA23022D023B8BC25AA',
|
|
u'A52CB8B320D22032ABB3A41D773D2B6342034902.A22E87CDD37DBE75A5E52412DC874AC16A7CFCA2',
|
|
),
|
|
]
|
|
|
|
|
|
class TestSignature(unittest.TestCase):
|
|
def setUp(self):
|
|
TEST_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
self.TESTDATA_DIR = os.path.join(TEST_DIR, 'testdata')
|
|
if not os.path.exists(self.TESTDATA_DIR):
|
|
os.mkdir(self.TESTDATA_DIR)
|
|
|
|
|
|
def make_tfunc(url, stype, sig_input, expected_sig):
|
|
basename = url.rpartition('/')[2]
|
|
m = re.match(r'.*-([a-zA-Z0-9_-]+)\.[a-z]+$', basename)
|
|
assert m, '%r should follow URL format' % basename
|
|
test_id = m.group(1)
|
|
|
|
def test_func(self):
|
|
fn = os.path.join(self.TESTDATA_DIR, basename)
|
|
|
|
if not os.path.exists(fn):
|
|
compat_urlretrieve(url, fn)
|
|
|
|
ie = YoutubeIE()
|
|
if stype == 'js':
|
|
with io.open(fn, encoding='utf-8') as testf:
|
|
jscode = testf.read()
|
|
func = ie._parse_sig_js(jscode)
|
|
else:
|
|
assert stype == 'swf'
|
|
with open(fn, 'rb') as testf:
|
|
swfcode = testf.read()
|
|
func = ie._parse_sig_swf(swfcode)
|
|
src_sig = (
|
|
compat_str(string.printable[:sig_input])
|
|
if isinstance(sig_input, int) else sig_input)
|
|
got_sig = func(src_sig)
|
|
self.assertEqual(got_sig, expected_sig)
|
|
|
|
test_func.__name__ = str('test_signature_' + stype + '_' + test_id)
|
|
setattr(TestSignature, test_func.__name__, test_func)
|
|
|
|
for test_spec in _TESTS:
|
|
make_tfunc(*test_spec)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|