l1ving_youtube-dl/test/test_jsinterp.py

#!/usr/bin/env python

from __future__ import unicode_literals

import os
import sys

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

defs = gettestcases()


class TestJSInterpreter(unittest.TestCase):
    def setUp(self):
        self.defs = defs


def generator(test_case):
    def test_template(self):
        for test in test_case['subtests']:
            jsi = JSInterpreter(test['code'], variables=None if 'globals' not in test else test['globals'])
            if 'asserts' in test:
                for a in test['asserts']:
                    if 'call' in a:
                        self.assertEqual(jsi.call_function(*a['call']), a['value'])
                    else:
                        self.assertEqual(jsi.run(), a['value'])

    if 'skip' not in test_case or 'i' not in test_case['skip']:
        reason = False
    else:
        reason = test_case['skip']['i']

    return test_template if not reason else unittest.skip(reason)(test_template)


# And add them to TestJSInterpreter
for n, tc in enumerate(defs):
    if any('asserts' in test for test in tc['subtests']):
        test_method = generator(tc)
        tname = 'test_' + str(tc['name'])
        i = 1
        while hasattr(TestJSInterpreter, tname):
            tname = 'test_%s_%d' % (tc['name'], i)
            i += 1
        test_method.__name__ = str(tname)
        setattr(TestJSInterpreter, test_method.__name__, test_method)
        del test_method