test_jsinterp.py 15 KB

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