]>
jfr.im git - yt-dlp.git/blob - test/test_youtube_signature.py
3 # Allow direct execution
7 sys
.path
.insert(0, os
.path
.dirname(os
.path
.dirname(os
.path
.abspath(__file__
))))
14 from youtube_dl
.extractor
import YoutubeIE
15 from youtube_dl
.utils
import compat_str
, compat_urlretrieve
19 u
'https://s.ytimg.com/yts/jsbin/html5player-vflHOr_nV.js',
22 u
'>=<;:/.-[+*)(\'&%$#"!ZYX0VUTSRQPONMLKJIHGFEDCBA\\yxwvutsrqponmlkjihgfedcba987654321',
25 u
'https://s.ytimg.com/yts/jsbin/html5player-vfldJ8xgI.js',
28 u
'3456789a0cdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRS[UVWXYZ!"#$%&\'()*+,-./:;<=>?@',
31 u
'https://s.ytimg.com/yts/jsbin/html5player-vfle-mVwz.js',
34 u
']\\[@?>=<;:/.-,+*)(\'&%$#"hZYXWVUTSRQPONMLKJIHGFEDCBAzyxwvutsrqponmlkjiagfedcb39876',
37 u
'https://s.ytimg.com/yts/jsbin/html5player-en_US-vfl0Cbn9e.js',
40 u
'O1I3456789abcde0ghijklmnopqrstuvwxyzABCDEFGHfJKLMN2PQRSTUVW@YZ!"#$%&\'()*+,-./:;<=',
43 u
'https://s.ytimg.com/yts/jsbin/html5player-en_US-vflXGBaUN.js',
45 u
'2ACFC7A61CA478CD21425E5A57EBD73DDC78E22A.2094302436B2D377D14A3BBA23022D023B8BC25AA',
46 u
'A52CB8B320D22032ABB3A41D773D2B6342034902.A22E87CDD37DBE75A5E52412DC874AC16A7CFCA2',
49 u
'http://s.ytimg.com/yts/swfbin/player-vfl5vIhK2/watch_as3.swf',
52 u
'O1I3456789abcde0ghijklmnopqrstuvwxyzABCDEFGHfJKLMN2PQRSTUVWXY\\!"#$%&\'()*+,-./:;<=>?'
55 u
'http://s.ytimg.com/yts/swfbin/player-vflmDyk47/watch_as3.swf',
57 u
'F375F75BF2AFDAAF2666E43868D46816F83F13E81C46.3725A8218E446A0DECD33F79DC282994D6AA92C92C9',
58 u
'9C29AA6D499282CD97F33DCED0A644E8128A5273.64C18E31F38361864D86834E6662FAADFA2FB57F'
61 u
'https://s.ytimg.com/yts/jsbin/html5player-en_US-vflBb0OQx.js',
64 u
'123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQ0STUVWXYZ!"#$%&\'()*+,@./:;<=>'
67 u
'https://s.ytimg.com/yts/jsbin/html5player-en_US-vfl9FYC6l.js',
70 u
'123456789abcdefghijklmnopqr0tuvwxyzABCDETGHIJKLMNOPQRS>UVWXYZ!"#$%&\'()*+,-./:;<=F'
75 class TestSignature(unittest
.TestCase
):
77 TEST_DIR
= os
.path
.dirname(os
.path
.abspath(__file__
))
78 self
.TESTDATA_DIR
= os
.path
.join(TEST_DIR
, 'testdata')
79 if not os
.path
.exists(self
.TESTDATA_DIR
):
80 os
.mkdir(self
.TESTDATA_DIR
)
83 def make_tfunc(url
, stype
, sig_input
, expected_sig
):
84 m
= re
.match(r
'.*-([a-zA-Z0-9_-]+)(?:/watch_as3)?\.[a-z]+$', url
)
85 assert m
, '%r should follow URL format' % url
89 basename
= 'player-%s.%s' % (test_id
, stype
)
90 fn
= os
.path
.join(self
.TESTDATA_DIR
, basename
)
92 if not os
.path
.exists(fn
):
93 compat_urlretrieve(url
, fn
)
97 with io
.open(fn
, encoding
='utf-8') as testf
:
99 func
= ie
._parse
_sig
_js
(jscode
)
101 assert stype
== 'swf'
102 with open(fn
, 'rb') as testf
:
103 swfcode
= testf
.read()
104 func
= ie
._parse
_sig
_swf
(swfcode
)
106 compat_str(string
.printable
[:sig_input
])
107 if isinstance(sig_input
, int) else sig_input
)
108 got_sig
= func(src_sig
)
109 self
.assertEqual(got_sig
, expected_sig
)
111 test_func
.__name
__ = str('test_signature_' + stype
+ '_' + test_id
)
112 setattr(TestSignature
, test_func
.__name
__, test_func
)
114 for test_spec
in _TESTS
:
115 make_tfunc(*test_spec
)
118 if __name__
== '__main__':