test/test_jsinterp.py

   1 #!/usr/bin/env python3
   2
   3 # Allow direct execution
   4 import os
   5 import sys
   6 import unittest
   7
   8 sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
   9
  10 import math
  11 import re
  12
  13 from yt_dlp.jsinterp import JS_Undefined, JSInterpreter
  14
  15
  16 class 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
  24         jsi = JSInterpreter('function x3(){42}')
  25         self.assertEqual(jsi.call_function('x3'), None)
  26
  27         jsi = JSInterpreter('var x5 = function(){return 42;}')
  28         self.assertEqual(jsi.call_function('x5'), 42)
  29
  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
  53         jsi = JSInterpreter('function f(){return 2 ** 5}')
  54         self.assertEqual(jsi.call_function('f'), 32)
  55
  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
  62         jsi = JSInterpreter('function f(){return []? 2+3: 4;}')
  63         self.assertEqual(jsi.call_function('f'), 5)
  64
  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
  71         jsi = JSInterpreter('function f(){return 0 ?? 42;}')
  72         self.assertEqual(jsi.call_function('f'), 0)
  73
  74     def test_array_access(self):
  75         jsi = JSInterpreter('function f(){var x = [1,2,3]; x[0] = 4; x[0] = 5; x[2.0] = 7; return x;}')
  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
  85     def test_quotes(self):
  86         jsi = JSInterpreter(R'function f(){return "a\"\\("}')
  87         self.assertEqual(jsi.call_function('f'), R'a"\(')
  88
  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):
 100         'Skipping: Not yet fully implemented'
 101         return
 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
 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
 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
 131     def test_builtins(self):
 132         jsi = JSInterpreter('''
 133         function x() { return new Date('Wednesday 31 December 1969 18:01:26 MDT') - 0; }
 134         ''')
 135         self.assertEqual(jsi.call_function('x'), 86000)
 136         jsi = JSInterpreter('''
 137         function x(dt) { return new Date(dt) - 0; }
 138         ''')
 139         self.assertEqual(jsi.call_function('x', 'Wednesday 31 December 1969 18:01:26 MDT'), 86000)
 140
 141     def test_call(self):
 142         jsi = JSInterpreter('''
 143         function x() { return 2; }
 144         function y(a) { return x() + (a?a:0); }
 145         function z() { return y(3); }
 146         ''')
 147         self.assertEqual(jsi.call_function('z'), 5)
 148         self.assertEqual(jsi.call_function('y'), 2)
 149
 150     def test_for_loop(self):
 151         jsi = JSInterpreter('''
 152         function x() { a=0; for (i=0; i-10; i++) {a++} return a }
 153         ''')
 154         self.assertEqual(jsi.call_function('x'), 10)
 155
 156     def test_switch(self):
 157         jsi = JSInterpreter('''
 158         function x(f) { switch(f){
 159             case 1:f+=1;
 160             case 2:f+=2;
 161             case 3:f+=3;break;
 162             case 4:f+=4;
 163             default:f=0;
 164         } return f }
 165         ''')
 166         self.assertEqual(jsi.call_function('x', 1), 7)
 167         self.assertEqual(jsi.call_function('x', 3), 6)
 168         self.assertEqual(jsi.call_function('x', 5), 0)
 169
 170     def test_switch_default(self):
 171         jsi = JSInterpreter('''
 172         function x(f) { switch(f){
 173             case 2: f+=2;
 174             default: f-=1;
 175             case 5:
 176             case 6: f+=6;
 177             case 0: break;
 178             case 1: f+=1;
 179         } return f }
 180         ''')
 181         self.assertEqual(jsi.call_function('x', 1), 2)
 182         self.assertEqual(jsi.call_function('x', 5), 11)
 183         self.assertEqual(jsi.call_function('x', 9), 14)
 184
 185     def test_try(self):
 186         jsi = JSInterpreter('''
 187         function x() { try{return 10} catch(e){return 5} }
 188         ''')
 189         self.assertEqual(jsi.call_function('x'), 10)
 190
 191     def test_for_loop_continue(self):
 192         jsi = JSInterpreter('''
 193         function x() { a=0; for (i=0; i-10; i++) { continue; a++ } return a }
 194         ''')
 195         self.assertEqual(jsi.call_function('x'), 0)
 196
 197     def test_for_loop_break(self):
 198         jsi = JSInterpreter('''
 199         function x() { a=0; for (i=0; i-10; i++) { break; a++ } return a }
 200         ''')
 201         self.assertEqual(jsi.call_function('x'), 0)
 202
 203     def test_literal_list(self):
 204         jsi = JSInterpreter('''
 205         function x() { return [1, 2, "asdf", [5, 6, 7]][3] }
 206         ''')
 207         self.assertEqual(jsi.call_function('x'), [5, 6, 7])
 208
 209     def test_comma(self):
 210         jsi = JSInterpreter('''
 211         function x() { a=5; a -= 1, a+=3; return a }
 212         ''')
 213         self.assertEqual(jsi.call_function('x'), 7)
 214
 215         jsi = JSInterpreter('''
 216         function x() { a=5; return (a -= 1, a+=3, a); }
 217         ''')
 218         self.assertEqual(jsi.call_function('x'), 7)
 219
 220         jsi = JSInterpreter('''
 221         function x() { return (l=[0,1,2,3], function(a, b){return a+b})((l[1], l[2]), l[3]) }
 222         ''')
 223         self.assertEqual(jsi.call_function('x'), 5)
 224
 225     def test_void(self):
 226         jsi = JSInterpreter('''
 227         function x() { return void 42; }
 228         ''')
 229         self.assertEqual(jsi.call_function('x'), None)
 230
 231     def test_return_function(self):
 232         jsi = JSInterpreter('''
 233         function x() { return [1, function(){return 1}][1] }
 234         ''')
 235         self.assertEqual(jsi.call_function('x')([]), 1)
 236
 237     def test_null(self):
 238         jsi = JSInterpreter('''
 239         function x() { return null; }
 240         ''')
 241         self.assertEqual(jsi.call_function('x'), None)
 242
 243         jsi = JSInterpreter('''
 244         function x() { return [null > 0, null < 0, null == 0, null === 0]; }
 245         ''')
 246         self.assertEqual(jsi.call_function('x'), [False, False, False, False])
 247
 248         jsi = JSInterpreter('''
 249         function x() { return [null >= 0, null <= 0]; }
 250         ''')
 251         self.assertEqual(jsi.call_function('x'), [True, True])
 252
 253     def test_undefined(self):
 254         jsi = JSInterpreter('''
 255         function x() { return undefined === undefined; }
 256         ''')
 257         self.assertEqual(jsi.call_function('x'), True)
 258
 259         jsi = JSInterpreter('''
 260         function x() { return undefined; }
 261         ''')
 262         self.assertEqual(jsi.call_function('x'), JS_Undefined)
 263
 264         jsi = JSInterpreter('''
 265         function x() { let v; return v; }
 266         ''')
 267         self.assertEqual(jsi.call_function('x'), JS_Undefined)
 268
 269         jsi = JSInterpreter('''
 270         function x() { return [undefined === undefined, undefined == undefined, undefined < undefined, undefined > undefined]; }
 271         ''')
 272         self.assertEqual(jsi.call_function('x'), [True, True, False, False])
 273
 274         jsi = JSInterpreter('''
 275         function x() { return [undefined === 0, undefined == 0, undefined < 0, undefined > 0]; }
 276         ''')
 277         self.assertEqual(jsi.call_function('x'), [False, False, False, False])
 278
 279         jsi = JSInterpreter('''
 280         function x() { return [undefined >= 0, undefined <= 0]; }
 281         ''')
 282         self.assertEqual(jsi.call_function('x'), [False, False])
 283
 284         jsi = JSInterpreter('''
 285         function x() { return [undefined > null, undefined < null, undefined == null, undefined === null]; }
 286         ''')
 287         self.assertEqual(jsi.call_function('x'), [False, False, True, 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, True, False, False])
 293
 294         jsi = JSInterpreter('''
 295         function x() { let v; return [42+v, v+42, v**42, 42**v, 0**v]; }
 296         ''')
 297         for y in jsi.call_function('x'):
 298             self.assertTrue(math.isnan(y))
 299
 300         jsi = JSInterpreter('''
 301         function x() { let v; return v**0; }
 302         ''')
 303         self.assertEqual(jsi.call_function('x'), 1)
 304
 305         jsi = JSInterpreter('''
 306         function x() { let v; return [v>42, v<=42, v&&42, 42&&v]; }
 307         ''')
 308         self.assertEqual(jsi.call_function('x'), [False, False, JS_Undefined, JS_Undefined])
 309
 310         jsi = JSInterpreter('function x(){return undefined ?? 42; }')
 311         self.assertEqual(jsi.call_function('x'), 42)
 312
 313     def test_object(self):
 314         jsi = JSInterpreter('''
 315         function x() { return {}; }
 316         ''')
 317         self.assertEqual(jsi.call_function('x'), {})
 318
 319         jsi = JSInterpreter('''
 320         function x() { let a = {m1: 42, m2: 0 }; return [a["m1"], a.m2]; }
 321         ''')
 322         self.assertEqual(jsi.call_function('x'), [42, 0])
 323
 324         jsi = JSInterpreter('''
 325         function x() { let a; return a?.qq; }
 326         ''')
 327         self.assertEqual(jsi.call_function('x'), JS_Undefined)
 328
 329         jsi = JSInterpreter('''
 330         function x() { let a = {m1: 42, m2: 0 }; return a?.qq; }
 331         ''')
 332         self.assertEqual(jsi.call_function('x'), JS_Undefined)
 333
 334     def test_regex(self):
 335         jsi = JSInterpreter('''
 336         function x() { let a=/,,[/,913,/](,)}/; }
 337         ''')
 338         self.assertEqual(jsi.call_function('x'), None)
 339
 340         jsi = JSInterpreter('''
 341         function x() { let a=/,,[/,913,/](,)}/; return a; }
 342         ''')
 343         self.assertIsInstance(jsi.call_function('x'), re.Pattern)
 344
 345         jsi = JSInterpreter('''
 346         function x() { let a=/,,[/,913,/](,)}/i; return a; }
 347         ''')
 348         self.assertEqual(jsi.call_function('x').flags & re.I, re.I)
 349
 350
 351 if __name__ == '__main__':
 352     unittest.main()