66 lines
1.7 KiB
Python
66 lines
1.7 KiB
Python
#!/usr/bin/env python
|
|
|
|
from __future__ import unicode_literals
|
|
|
|
import os
|
|
import sys
|
|
import copy
|
|
|
|
if sys.version_info < (2, 7):
|
|
import unittest2 as unittest
|
|
else:
|
|
import unittest
|
|
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
|
|
from youtube_dl.jsinterp import JSInterpreter
|
|
from test.jstests import gettestcases
|
|
|
|
|
|
def traverse(node, tree_types=(list, tuple)):
|
|
if type(node) == zip:
|
|
node = list(copy.deepcopy(node))
|
|
if isinstance(node, tree_types):
|
|
tree = []
|
|
for value in node:
|
|
tree.append(traverse(value, tree_types))
|
|
return tree
|
|
else:
|
|
return node
|
|
|
|
|
|
defs = gettestcases()
|
|
|
|
|
|
class TestJSInterpreterParse(unittest.TestCase):
|
|
def setUp(self):
|
|
self.defs = defs
|
|
|
|
|
|
def generator(test_case):
|
|
def test_template(self):
|
|
for a in test_case['subtests']:
|
|
jsi = JSInterpreter(a['code'], variables=None if 'globals' not in a else a['globals'])
|
|
parsed = list(jsi.statements())
|
|
if 'ast' in a:
|
|
self.assertEqual(traverse(parsed), traverse(a['ast']))
|
|
|
|
if 'skip' not in test_case or 'p' not in test_case['skip']:
|
|
reason = False
|
|
else:
|
|
reason = test_case['skip']['p']
|
|
|
|
return test_template if not reason else unittest.skip(reason)(test_template)
|
|
|
|
|
|
# And add them to TestJSInterpreter
|
|
for n, tc in enumerate(defs):
|
|
test_method = generator(tc)
|
|
tname = 'test_' + str(tc['name'])
|
|
i = 1
|
|
while hasattr(TestJSInterpreterParse, tname):
|
|
tname = 'test_%s_%d' % (tc['name'], i)
|
|
i += 1
|
|
test_method.__name__ = str(tname)
|
|
setattr(TestJSInterpreterParse, test_method.__name__, test_method)
|
|
del test_method
|