l1ving_youtube-dl/test/test_jsinterp.py

#!/usr/bin/env python

# """
# see: `js2tests`
# """

from __future__ import unicode_literals

# Allow direct execution
import os
import sys
import logging

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 .js2tests import gettestcases

defs = gettestcases()
# set level to logging.DEBUG to see messages about missing assertions
logging.basicConfig(stream=sys.stderr, level=logging.WARNING)


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


def generator(test_case, name):
    def test_template(self):
        for test in test_case['subtests']:
            excluded = test.get('exclude')
            if excluded is not None and 'jsinterp' in excluded:
                log_reason = 'jsinterp does not support this subtest:\n%s' % test['code']
            elif 'code' not in test:
                log_reason = 'No code in subtest, skipping'
            elif 'asserts' not in test:
                log_reason = 'No assertion in subtest, skipping'
            else:
                log_reason = None

            if log_reason is None:
                variables = test.get('globals')
                code = test['code']
                call = None

                if variables is not None:
                    code = 'function f(%s){%s}' % ((''.join(variables.keys())), code)
                    call = ('f',) + tuple(v for v in variables.values())

                jsi = JSInterpreter(code, objects=variables)
                for assertion in test['asserts']:
                    if 'value' in assertion:
                        if call is None:
                            call = assertion['call']
                        self.assertEqual(jsi.call_function(*call), assertion['value'])
                    else:
                        log.debug('No value in assertion, skipping')
            else:
                log.debug(log_reason)

    log = logging.getLogger('TestJSInterpreter.%s' % name)
    return test_template


# And add them to TestJSInterpreter
for n, tc in enumerate(defs):
    reason = tc['skip'].get('jsinterp', False)
    tname = 'test_' + str(tc['name'])
    i = 1
    while hasattr(TestJSInterpreter, tname):
        tname = 'test_%s_%d' % (tc['name'], i)
        i += 1

    if reason is not True:
        log_reason = 'Entirely'
    elif not any('asserts' in test for test in tc['subtests']):
        log_reason = '''There isn't any assertion'''
    else:
        log_reason = None

    if log_reason is not None:
        test_method = generator(tc, tname)
        test_method.__name__ = str(tname)
        if reason is not False:
            test_method.__unittest_skip__ = True
            test_method.__unittest_skip_why__ = reason
        setattr(TestJSInterpreter, test_method.__name__, test_method)
        del test_method
    else:
        log = logging.getLogger('TestJSInterpreter')
        log.debug('Skipping %s:%s' % (tname, log_reason))

if __name__ == '__main__':
    unittest.main()