test_youtube_signature.py (5229B)
1 #!/usr/bin/env python 2 3 from __future__ import unicode_literals 4 5 # Allow direct execution 6 import os 7 import sys 8 import unittest 9 sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) 10 11 import io 12 import re 13 import string 14 15 from test.helper import FakeYDL 16 from youtube_dl.extractor import YoutubeIE 17 from youtube_dl.compat import compat_str, compat_urlretrieve 18 19 _TESTS = [ 20 ( 21 'https://s.ytimg.com/yts/jsbin/html5player-vflHOr_nV.js', 22 86, 23 '>=<;:/.-[+*)(\'&%$#"!ZYX0VUTSRQPONMLKJIHGFEDCBA\\yxwvutsrqponmlkjihgfedcba987654321', 24 ), 25 ( 26 'https://s.ytimg.com/yts/jsbin/html5player-vfldJ8xgI.js', 27 85, 28 '3456789a0cdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRS[UVWXYZ!"#$%&\'()*+,-./:;<=>?@', 29 ), 30 ( 31 'https://s.ytimg.com/yts/jsbin/html5player-vfle-mVwz.js', 32 90, 33 ']\\[@?>=<;:/.-,+*)(\'&%$#"hZYXWVUTSRQPONMLKJIHGFEDCBAzyxwvutsrqponmlkjiagfedcb39876', 34 ), 35 ( 36 'https://s.ytimg.com/yts/jsbin/html5player-en_US-vfl0Cbn9e.js', 37 84, 38 'O1I3456789abcde0ghijklmnopqrstuvwxyzABCDEFGHfJKLMN2PQRSTUVW@YZ!"#$%&\'()*+,-./:;<=', 39 ), 40 ( 41 'https://s.ytimg.com/yts/jsbin/html5player-en_US-vflXGBaUN.js', 42 '2ACFC7A61CA478CD21425E5A57EBD73DDC78E22A.2094302436B2D377D14A3BBA23022D023B8BC25AA', 43 'A52CB8B320D22032ABB3A41D773D2B6342034902.A22E87CDD37DBE75A5E52412DC874AC16A7CFCA2', 44 ), 45 ( 46 'https://s.ytimg.com/yts/jsbin/html5player-en_US-vflBb0OQx.js', 47 84, 48 '123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQ0STUVWXYZ!"#$%&\'()*+,@./:;<=>' 49 ), 50 ( 51 'https://s.ytimg.com/yts/jsbin/html5player-en_US-vfl9FYC6l.js', 52 83, 53 '123456789abcdefghijklmnopqr0tuvwxyzABCDETGHIJKLMNOPQRS>UVWXYZ!"#$%&\'()*+,-./:;<=F' 54 ), 55 ( 56 'https://s.ytimg.com/yts/jsbin/html5player-en_US-vflCGk6yw/html5player.js', 57 '4646B5181C6C3020DF1D9C7FCFEA.AD80ABF70C39BD369CCCAE780AFBB98FA6B6CB42766249D9488C288', 58 '82C8849D94266724DC6B6AF89BBFA087EACCD963.B93C07FBA084ACAEFCF7C9D1FD0203C6C1815B6B' 59 ), 60 ( 61 'https://s.ytimg.com/yts/jsbin/html5player-en_US-vflKjOTVq/html5player.js', 62 '312AA52209E3623129A412D56A40F11CB0AF14AE.3EE09501CB14E3BCDC3B2AE808BF3F1D14E7FBF12', 63 '112AA5220913623229A412D56A40F11CB0AF14AE.3EE0950FCB14EEBCDC3B2AE808BF331D14E7FBF3', 64 ) 65 ] 66 67 68 class TestPlayerInfo(unittest.TestCase): 69 def test_youtube_extract_player_info(self): 70 PLAYER_URLS = ( 71 ('https://www.youtube.com/s/player/64dddad9/player_ias.vflset/en_US/base.js', '64dddad9'), 72 ('https://www.youtube.com/s/player/64dddad9/player_ias.vflset/fr_FR/base.js', '64dddad9'), 73 ('https://www.youtube.com/s/player/64dddad9/player-plasma-ias-phone-en_US.vflset/base.js', '64dddad9'), 74 ('https://www.youtube.com/s/player/64dddad9/player-plasma-ias-phone-de_DE.vflset/base.js', '64dddad9'), 75 ('https://www.youtube.com/s/player/64dddad9/player-plasma-ias-tablet-en_US.vflset/base.js', '64dddad9'), 76 # obsolete 77 ('https://www.youtube.com/yts/jsbin/player_ias-vfle4-e03/en_US/base.js', 'vfle4-e03'), 78 ('https://www.youtube.com/yts/jsbin/player_ias-vfl49f_g4/en_US/base.js', 'vfl49f_g4'), 79 ('https://www.youtube.com/yts/jsbin/player_ias-vflCPQUIL/en_US/base.js', 'vflCPQUIL'), 80 ('https://www.youtube.com/yts/jsbin/player-vflzQZbt7/en_US/base.js', 'vflzQZbt7'), 81 ('https://www.youtube.com/yts/jsbin/player-en_US-vflaxXRn1/base.js', 'vflaxXRn1'), 82 ('https://s.ytimg.com/yts/jsbin/html5player-en_US-vflXGBaUN.js', 'vflXGBaUN'), 83 ('https://s.ytimg.com/yts/jsbin/html5player-en_US-vflKjOTVq/html5player.js', 'vflKjOTVq'), 84 ) 85 for player_url, expected_player_id in PLAYER_URLS: 86 player_id = YoutubeIE._extract_player_info(player_url) 87 self.assertEqual(player_id, expected_player_id) 88 89 90 class TestSignature(unittest.TestCase): 91 def setUp(self): 92 TEST_DIR = os.path.dirname(os.path.abspath(__file__)) 93 self.TESTDATA_DIR = os.path.join(TEST_DIR, 'testdata') 94 if not os.path.exists(self.TESTDATA_DIR): 95 os.mkdir(self.TESTDATA_DIR) 96 97 98 def make_tfunc(url, sig_input, expected_sig): 99 m = re.match(r'.*-([a-zA-Z0-9_-]+)(?:/watch_as3|/html5player)?\.[a-z]+$', url) 100 assert m, '%r should follow URL format' % url 101 test_id = m.group(1) 102 103 def test_func(self): 104 basename = 'player-%s.js' % test_id 105 fn = os.path.join(self.TESTDATA_DIR, basename) 106 107 if not os.path.exists(fn): 108 compat_urlretrieve(url, fn) 109 110 ydl = FakeYDL() 111 ie = YoutubeIE(ydl) 112 with io.open(fn, encoding='utf-8') as testf: 113 jscode = testf.read() 114 func = ie._parse_sig_js(jscode) 115 src_sig = ( 116 compat_str(string.printable[:sig_input]) 117 if isinstance(sig_input, int) else sig_input) 118 got_sig = func(src_sig) 119 self.assertEqual(got_sig, expected_sig) 120 121 test_func.__name__ = str('test_signature_js_' + test_id) 122 setattr(TestSignature, test_func.__name__, test_func) 123 124 125 for test_spec in _TESTS: 126 make_tfunc(*test_spec) 127 128 129 if __name__ == '__main__': 130 unittest.main()