112 lines
4.2 KiB
Python
112 lines
4.2 KiB
Python
|
from youtube_dl.jsinterp.jsgrammar import Token
|
||
|
|
from youtube_dl.jsinterp.tstream import _OPERATORS
|
||
|
|
|
||
|
|
skip = {'i': 'Interpreting function call not yet implemented'}
|
||
|
|
|
||
|
|
tests = [
|
||
|
|
{
|
||
|
|
'code': '''
|
||
|
|
function x() { return 2; }
|
||
|
|
function y(a) { return x() + a; }
|
||
|
|
function z() { return y(3); }
|
||
|
|
''',
|
||
|
|
'asserts': [{'value': 5, 'call': ('z',)}],
|
||
|
|
'ast': [
|
||
|
|
(Token.FUNC, 'x',
|
||
|
|
[],
|
||
|
|
(Token.BLOCK, [
|
||
|
|
(Token.RETURN, (Token.EXPR, [
|
||
|
|
(Token.ASSIGN, None, (Token.OPEXPR, [(Token.MEMBER, (Token.INT, 2), None, None)]), None)
|
||
|
|
])
|
||
|
|
)
|
||
|
|
])),
|
||
|
|
(Token.FUNC, 'y',
|
||
|
|
['a'],
|
||
|
|
(Token.BLOCK, [
|
||
|
|
(Token.RETURN, (Token.EXPR, [
|
||
|
|
(Token.ASSIGN, None,
|
||
|
|
(Token.OPEXPR, [
|
||
|
|
(Token.MEMBER, (Token.ID, 'x'), None, (Token.CALL, [], None)),
|
||
|
|
(Token.MEMBER, (Token.ID, 'a'), None, None),
|
||
|
|
(Token.OP, _OPERATORS['+'][1])
|
||
|
|
]), None)
|
||
|
|
])
|
||
|
|
)
|
||
|
|
])),
|
||
|
|
(Token.FUNC, 'z',
|
||
|
|
[],
|
||
|
|
(Token.BLOCK, [
|
||
|
|
(Token.RETURN, (Token.EXPR, [
|
||
|
|
(Token.ASSIGN, None, (Token.OPEXPR, [
|
||
|
|
(Token.MEMBER, (Token.ID, 'y'), None, (Token.CALL, [
|
||
|
|
(Token.ASSIGN, None, (Token.OPEXPR, [(Token.MEMBER, (Token.INT, 3), None, None)]), None)
|
||
|
|
], None))
|
||
|
|
]), None)
|
||
|
|
])
|
||
|
|
)
|
||
|
|
]))
|
||
|
|
]
|
||
|
|
}, {
|
||
|
|
'code': 'function x(a) { return a.split(""); }',
|
||
|
|
'asserts': [{'value': ["a", "b", "c"], 'call': ('x',)}],
|
||
|
|
'ast': [
|
||
|
|
(Token.FUNC, 'x',
|
||
|
|
['a'],
|
||
|
|
(Token.BLOCK, [
|
||
|
|
(Token.RETURN, (Token.EXPR, [
|
||
|
|
(Token.ASSIGN, None, (Token.OPEXPR, [
|
||
|
|
(Token.MEMBER, (Token.ID, 'a'), None,
|
||
|
|
(Token.FIELD, 'split',
|
||
|
|
(Token.CALL, [
|
||
|
|
(Token.ASSIGN, None, (Token.OPEXPR, [(Token.MEMBER, (Token.STR, ''), None, None)]), None)
|
||
|
|
], None))
|
||
|
|
)]),
|
||
|
|
None)
|
||
|
|
])
|
||
|
|
)
|
||
|
|
]))
|
||
|
|
]
|
||
|
|
}, {
|
||
|
|
'code': '''
|
||
|
|
function a(x) { return x; }
|
||
|
|
function b(x) { return x; }
|
||
|
|
function c() { return [a, b][0](0); }
|
||
|
|
''',
|
||
|
|
'asserts': [{'value': 0}],
|
||
|
|
'ast': [
|
||
|
|
(Token.FUNC, 'a', ['x'],
|
||
|
|
(Token.BLOCK, [
|
||
|
|
(Token.RETURN, (Token.EXPR, [
|
||
|
|
(Token.ASSIGN, None, (Token.OPEXPR, [(Token.MEMBER, (Token.ID, 'x'), None, None)]), None)
|
||
|
|
])
|
||
|
|
)
|
||
|
|
])),
|
||
|
|
(Token.FUNC, 'b', ['x'],
|
||
|
|
(Token.BLOCK, [
|
||
|
|
(Token.RETURN, (Token.EXPR, [
|
||
|
|
(Token.ASSIGN, None, (Token.OPEXPR, [(Token.MEMBER, (Token.ID, 'x'), None, None)]), None)
|
||
|
|
])
|
||
|
|
)
|
||
|
|
])),
|
||
|
|
(Token.FUNC, 'c', [],
|
||
|
|
(Token.BLOCK, [
|
||
|
|
(Token.RETURN, (Token.EXPR, [
|
||
|
|
(Token.ASSIGN, None, (Token.OPEXPR, [
|
||
|
|
(Token.MEMBER, (Token.ARRAY, [
|
||
|
|
(Token.ASSIGN, None, (Token.OPEXPR, [
|
||
|
|
(Token.MEMBER, (Token.ID, 'a'), None, None)]), None),
|
||
|
|
(Token.ASSIGN, None, (Token.OPEXPR, [
|
||
|
|
(Token.MEMBER, (Token.ID, 'b'), None, None)]), None)
|
||
|
|
]), None, (Token.ELEM, (Token.EXPR, [
|
||
|
|
(Token.ASSIGN, None, (Token.OPEXPR, [(Token.MEMBER, (Token.INT, 0), None, None)]), None)
|
||
|
|
]), (Token.CALL, [
|
||
|
|
(Token.ASSIGN, None, (Token.OPEXPR, [(Token.MEMBER, (Token.INT, 0), None, None)]), None)
|
||
|
|
], None)))
|
||
|
|
]), None)
|
||
|
|
])
|
||
|
|
)
|
||
|
|
]))
|
||
|
|
]
|
||
|
|
}
|
||
|
|
]
|