]> jfr.im git - yt-dlp.git/blame - test/test_jsinterp.py
[extractor/youtube] Fallback regex for nsig code extraction
[yt-dlp.git] / test / test_jsinterp.py
CommitLineData
cc52de43 1#!/usr/bin/env python3
54007a45 2
9e3f1991
PH
3# Allow direct execution
4import os
5import sys
6import unittest
f8271158 7
9e3f1991
PH
8sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
9
be13a6e5 10import math
11import re
54007a45 12
be13a6e5 13from yt_dlp.jsinterp import JS_Undefined, JSInterpreter
9e3f1991
PH
14
15
16class TestJSInterpreter(unittest.TestCase):
17 def test_basic(self):
18 jsi = JSInterpreter('function x(){;}')
19 self.assertEqual(jsi.call_function('x'), None)
20
21 jsi = JSInterpreter('function x3(){return 42;}')
22 self.assertEqual(jsi.call_function('x3'), 42)
23
8f53dc44 24 jsi = JSInterpreter('function x3(){42}')
25 self.assertEqual(jsi.call_function('x3'), None)
26
ff29bf81
YCH
27 jsi = JSInterpreter('var x5 = function(){return 42;}')
28 self.assertEqual(jsi.call_function('x5'), 42)
29
9e3f1991
PH
30 def test_calc(self):
31 jsi = JSInterpreter('function x4(a){return 2*a+1;}')
32 self.assertEqual(jsi.call_function('x4', 3), 7)
33
34 def test_empty_return(self):
35 jsi = JSInterpreter('function f(){return; y()}')
36 self.assertEqual(jsi.call_function('f'), None)
37
38 def test_morespace(self):
39 jsi = JSInterpreter('function x (a) { return 2 * a + 1 ; }')
40 self.assertEqual(jsi.call_function('x', 3), 7)
41
42 jsi = JSInterpreter('function f () { x = 2 ; return x; }')
43 self.assertEqual(jsi.call_function('f'), 2)
44
45 def test_strange_chars(self):
46 jsi = JSInterpreter('function $_xY1 ($_axY1) { var $_axY2 = $_axY1 + 1; return $_axY2; }')
47 self.assertEqual(jsi.call_function('$_xY1', 20), 21)
48
49 def test_operators(self):
50 jsi = JSInterpreter('function f(){return 1 << 5;}')
51 self.assertEqual(jsi.call_function('f'), 32)
52
49b4ceae 53 jsi = JSInterpreter('function f(){return 2 ** 5}')
54 self.assertEqual(jsi.call_function('f'), 32)
55
9e3f1991
PH
56 jsi = JSInterpreter('function f(){return 19 & 21;}')
57 self.assertEqual(jsi.call_function('f'), 17)
58
59 jsi = JSInterpreter('function f(){return 11 >> 2;}')
60 self.assertEqual(jsi.call_function('f'), 2)
61
8f53dc44 62 jsi = JSInterpreter('function f(){return []? 2+3: 4;}')
63 self.assertEqual(jsi.call_function('f'), 5)
64
49b4ceae 65 jsi = JSInterpreter('function f(){return 1 == 2}')
66 self.assertEqual(jsi.call_function('f'), False)
67
68 jsi = JSInterpreter('function f(){return 0 && 1 || 2;}')
69 self.assertEqual(jsi.call_function('f'), 2)
70
be13a6e5 71 jsi = JSInterpreter('function f(){return 0 ?? 42;}')
72 self.assertEqual(jsi.call_function('f'), 0)
73
9e3f1991 74 def test_array_access(self):
8f53dc44 75 jsi = JSInterpreter('function f(){var x = [1,2,3]; x[0] = 4; x[0] = 5; x[2.0] = 7; return x;}')
9e3f1991
PH
76 self.assertEqual(jsi.call_function('f'), [5, 2, 7])
77
78 def test_parens(self):
79 jsi = JSInterpreter('function f(){return (1) + (2) * ((( (( (((((3)))))) )) ));}')
80 self.assertEqual(jsi.call_function('f'), 7)
81
82 jsi = JSInterpreter('function f(){return (1 + 2) * 3;}')
83 self.assertEqual(jsi.call_function('f'), 9)
84
8f53dc44 85 def test_quotes(self):
86 jsi = JSInterpreter(R'function f(){return "a\"\\("}')
87 self.assertEqual(jsi.call_function('f'), R'a"\(')
88
9e3f1991
PH
89 def test_assignments(self):
90 jsi = JSInterpreter('function f(){var x = 20; x = 30 + 1; return x;}')
91 self.assertEqual(jsi.call_function('f'), 31)
92
93 jsi = JSInterpreter('function f(){var x = 20; x += 30 + 1; return x;}')
94 self.assertEqual(jsi.call_function('f'), 51)
95
96 jsi = JSInterpreter('function f(){var x = 20; x -= 30 + 1; return x;}')
97 self.assertEqual(jsi.call_function('f'), -11)
98
99 def test_comments(self):
3eff81fb
PH
100 'Skipping: Not yet fully implemented'
101 return
9e3f1991
PH
102 jsi = JSInterpreter('''
103 function x() {
104 var x = /* 1 + */ 2;
105 var y = /* 30
106 * 40 */ 50;
107 return x + y;
108 }
109 ''')
110 self.assertEqual(jsi.call_function('x'), 52)
111
3eff81fb
PH
112 jsi = JSInterpreter('''
113 function f() {
114 var x = "/*";
115 var y = 1 /* comment */ + 2;
116 return y;
117 }
118 ''')
119 self.assertEqual(jsi.call_function('f'), 3)
120
9e3f1991
PH
121 def test_precedence(self):
122 jsi = JSInterpreter('''
123 function x() {
124 var a = [10, 20, 30, 40, 50];
125 var b = 6;
126 a[0]=a[b%a.length];
127 return a;
128 }''')
129 self.assertEqual(jsi.call_function('x'), [20, 20, 30, 40, 50])
130
49b4ceae 131 def test_builtins(self):
d81ba7d4 132 jsi = JSInterpreter('''
133 function x() { return NaN }
134 ''')
135 self.assertTrue(math.isnan(jsi.call_function('x')))
136
49b4ceae 137 jsi = JSInterpreter('''
138 function x() { return new Date('Wednesday 31 December 1969 18:01:26 MDT') - 0; }
139 ''')
140 self.assertEqual(jsi.call_function('x'), 86000)
141 jsi = JSInterpreter('''
142 function x(dt) { return new Date(dt) - 0; }
143 ''')
144 self.assertEqual(jsi.call_function('x', 'Wednesday 31 December 1969 18:01:26 MDT'), 86000)
145
189935f1
KM
146 def test_call(self):
147 jsi = JSInterpreter('''
148 function x() { return 2; }
8f53dc44 149 function y(a) { return x() + (a?a:0); }
189935f1
KM
150 function z() { return y(3); }
151 ''')
152 self.assertEqual(jsi.call_function('z'), 5)
8f53dc44 153 self.assertEqual(jsi.call_function('y'), 2)
9e3f1991 154
404f611f 155 def test_for_loop(self):
156 jsi = JSInterpreter('''
8f53dc44 157 function x() { a=0; for (i=0; i-10; i++) {a++} return a }
404f611f 158 ''')
159 self.assertEqual(jsi.call_function('x'), 10)
160
161 def test_switch(self):
162 jsi = JSInterpreter('''
163 function x(f) { switch(f){
164 case 1:f+=1;
165 case 2:f+=2;
166 case 3:f+=3;break;
167 case 4:f+=4;
168 default:f=0;
169 } return f }
170 ''')
171 self.assertEqual(jsi.call_function('x', 1), 7)
172 self.assertEqual(jsi.call_function('x', 3), 6)
173 self.assertEqual(jsi.call_function('x', 5), 0)
174
a1fc7ca0 175 def test_switch_default(self):
176 jsi = JSInterpreter('''
177 function x(f) { switch(f){
178 case 2: f+=2;
179 default: f-=1;
180 case 5:
181 case 6: f+=6;
182 case 0: break;
183 case 1: f+=1;
184 } return f }
185 ''')
186 self.assertEqual(jsi.call_function('x', 1), 2)
187 self.assertEqual(jsi.call_function('x', 5), 11)
188 self.assertEqual(jsi.call_function('x', 9), 14)
189
404f611f 190 def test_try(self):
191 jsi = JSInterpreter('''
192 function x() { try{return 10} catch(e){return 5} }
193 ''')
194 self.assertEqual(jsi.call_function('x'), 10)
195
196 def test_for_loop_continue(self):
197 jsi = JSInterpreter('''
8f53dc44 198 function x() { a=0; for (i=0; i-10; i++) { continue; a++ } return a }
404f611f 199 ''')
200 self.assertEqual(jsi.call_function('x'), 0)
201
202 def test_for_loop_break(self):
203 jsi = JSInterpreter('''
8f53dc44 204 function x() { a=0; for (i=0; i-10; i++) { break; a++ } return a }
404f611f 205 ''')
206 self.assertEqual(jsi.call_function('x'), 0)
207
208 def test_literal_list(self):
209 jsi = JSInterpreter('''
8f53dc44 210 function x() { return [1, 2, "asdf", [5, 6, 7]][3] }
404f611f 211 ''')
212 self.assertEqual(jsi.call_function('x'), [5, 6, 7])
213
214 def test_comma(self):
215 jsi = JSInterpreter('''
216 function x() { a=5; a -= 1, a+=3; return a }
217 ''')
218 self.assertEqual(jsi.call_function('x'), 7)
219
49b4ceae 220 jsi = JSInterpreter('''
221 function x() { a=5; return (a -= 1, a+=3, a); }
222 ''')
223 self.assertEqual(jsi.call_function('x'), 7)
224
6d3e7424 225 jsi = JSInterpreter('''
226 function x() { return (l=[0,1,2,3], function(a, b){return a+b})((l[1], l[2]), l[3]) }
227 ''')
228 self.assertEqual(jsi.call_function('x'), 5)
229
49b4ceae 230 def test_void(self):
231 jsi = JSInterpreter('''
232 function x() { return void 42; }
233 ''')
234 self.assertEqual(jsi.call_function('x'), None)
235
8f53dc44 236 def test_return_function(self):
237 jsi = JSInterpreter('''
238 function x() { return [1, function(){return 1}][1] }
239 ''')
240 self.assertEqual(jsi.call_function('x')([]), 1)
241
be13a6e5 242 def test_null(self):
243 jsi = JSInterpreter('''
244 function x() { return null; }
245 ''')
246 self.assertEqual(jsi.call_function('x'), None)
247
248 jsi = JSInterpreter('''
249 function x() { return [null > 0, null < 0, null == 0, null === 0]; }
250 ''')
251 self.assertEqual(jsi.call_function('x'), [False, False, False, False])
252
253 jsi = JSInterpreter('''
254 function x() { return [null >= 0, null <= 0]; }
255 ''')
256 self.assertEqual(jsi.call_function('x'), [True, True])
257
258 def test_undefined(self):
259 jsi = JSInterpreter('''
260 function x() { return undefined === undefined; }
261 ''')
262 self.assertEqual(jsi.call_function('x'), True)
263
264 jsi = JSInterpreter('''
265 function x() { return undefined; }
266 ''')
267 self.assertEqual(jsi.call_function('x'), JS_Undefined)
268
269 jsi = JSInterpreter('''
270 function x() { let v; return v; }
271 ''')
272 self.assertEqual(jsi.call_function('x'), JS_Undefined)
273
274 jsi = JSInterpreter('''
275 function x() { return [undefined === undefined, undefined == undefined, undefined < undefined, undefined > undefined]; }
276 ''')
277 self.assertEqual(jsi.call_function('x'), [True, True, False, False])
278
279 jsi = JSInterpreter('''
280 function x() { return [undefined === 0, undefined == 0, undefined < 0, undefined > 0]; }
281 ''')
282 self.assertEqual(jsi.call_function('x'), [False, False, False, False])
283
284 jsi = JSInterpreter('''
285 function x() { return [undefined >= 0, undefined <= 0]; }
286 ''')
287 self.assertEqual(jsi.call_function('x'), [False, False])
288
289 jsi = JSInterpreter('''
290 function x() { return [undefined > null, undefined < null, undefined == null, undefined === null]; }
291 ''')
292 self.assertEqual(jsi.call_function('x'), [False, False, True, False])
293
294 jsi = JSInterpreter('''
295 function x() { return [undefined === null, undefined == null, undefined < null, undefined > null]; }
296 ''')
297 self.assertEqual(jsi.call_function('x'), [False, True, False, False])
298
299 jsi = JSInterpreter('''
300 function x() { let v; return [42+v, v+42, v**42, 42**v, 0**v]; }
301 ''')
302 for y in jsi.call_function('x'):
303 self.assertTrue(math.isnan(y))
304
305 jsi = JSInterpreter('''
306 function x() { let v; return v**0; }
307 ''')
308 self.assertEqual(jsi.call_function('x'), 1)
309
310 jsi = JSInterpreter('''
311 function x() { let v; return [v>42, v<=42, v&&42, 42&&v]; }
312 ''')
313 self.assertEqual(jsi.call_function('x'), [False, False, JS_Undefined, JS_Undefined])
314
315 jsi = JSInterpreter('function x(){return undefined ?? 42; }')
316 self.assertEqual(jsi.call_function('x'), 42)
317
318 def test_object(self):
319 jsi = JSInterpreter('''
320 function x() { return {}; }
321 ''')
322 self.assertEqual(jsi.call_function('x'), {})
323
324 jsi = JSInterpreter('''
325 function x() { let a = {m1: 42, m2: 0 }; return [a["m1"], a.m2]; }
326 ''')
327 self.assertEqual(jsi.call_function('x'), [42, 0])
328
329 jsi = JSInterpreter('''
330 function x() { let a; return a?.qq; }
331 ''')
332 self.assertEqual(jsi.call_function('x'), JS_Undefined)
333
334 jsi = JSInterpreter('''
335 function x() { let a = {m1: 42, m2: 0 }; return a?.qq; }
336 ''')
337 self.assertEqual(jsi.call_function('x'), JS_Undefined)
338
339 def test_regex(self):
340 jsi = JSInterpreter('''
341 function x() { let a=/,,[/,913,/](,)}/; }
342 ''')
343 self.assertEqual(jsi.call_function('x'), None)
344
345 jsi = JSInterpreter('''
346 function x() { let a=/,,[/,913,/](,)}/; return a; }
347 ''')
348 self.assertIsInstance(jsi.call_function('x'), re.Pattern)
349
350 jsi = JSInterpreter('''
351 function x() { let a=/,,[/,913,/](,)}/i; return a; }
352 ''')
353 self.assertEqual(jsi.call_function('x').flags & re.I, re.I)
354
f26af78a
E
355 def test_char_code_at(self):
356 jsi = JSInterpreter('function x(i){return "test".charCodeAt(i)}')
357 self.assertEqual(jsi.call_function('x', 0), 116)
358 self.assertEqual(jsi.call_function('x', 1), 101)
359 self.assertEqual(jsi.call_function('x', 2), 115)
360 self.assertEqual(jsi.call_function('x', 3), 116)
361 self.assertEqual(jsi.call_function('x', 4), None)
362 self.assertEqual(jsi.call_function('x', 'not_a_number'), 116)
363
364 def test_bitwise_operators_overflow(self):
365 jsi = JSInterpreter('function x(){return -524999584 << 5}')
366 self.assertEqual(jsi.call_function('x'), 379882496)
367
368 jsi = JSInterpreter('function x(){return 1236566549 << 5}')
369 self.assertEqual(jsi.call_function('x'), 915423904)
370
582be358 371
9e3f1991
PH
372if __name__ == '__main__':
373 unittest.main()