3b536690d7
Refractors ast: * function declaration is no longer a statement * function body is no longer a block
110 lines
4.4 KiB
Python
110 lines
4.4 KiB
Python
from youtube_dl.jsinterp.jsgrammar import Token
|
|
from youtube_dl.jsinterp.tstream import _OPERATORS
|
|
|
|
tests = [
|
|
{
|
|
'code': '''
|
|
function x() { return 2; }
|
|
function y(a) { return x() + a; }
|
|
function z() { return y(3); }
|
|
z();
|
|
''',
|
|
'asserts': [{'value': 5}],
|
|
'ast': [
|
|
(Token.FUNC, 'x', [], [
|
|
(Token.RETURN, (Token.EXPR, [
|
|
(Token.ASSIGN, None, (Token.OPEXPR, [(Token.MEMBER, (Token.INT, 2), None, None)]), None)
|
|
]))
|
|
]),
|
|
(Token.FUNC, 'y', ['a'], [
|
|
(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.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)
|
|
])
|
|
)
|
|
]),
|
|
(Token.EXPR, [
|
|
(Token.ASSIGN, None, (Token.OPEXPR, [
|
|
(Token.MEMBER, (Token.ID, 'z'), None, (Token.CALL, [], None))
|
|
]), None)
|
|
])
|
|
]
|
|
}, {
|
|
'code': 'function x(a) { return a.split(""); }',
|
|
# built-in functions not yet implemented
|
|
# 'asserts': [{'value': ["a", "b", "c"], 'call': ('x',"abc")}],
|
|
'ast': [
|
|
(Token.FUNC, 'x', ['a'], [
|
|
(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 + 1; }
|
|
function c() { return [a, b][0](0); }
|
|
c();
|
|
''',
|
|
'asserts': [{'value': 0}],
|
|
'ast': [
|
|
(Token.FUNC, 'a', ['x'], [
|
|
(Token.RETURN, (Token.EXPR, [
|
|
(Token.ASSIGN, None, (Token.OPEXPR, [(Token.MEMBER, (Token.ID, 'x'), None, None)]), None)
|
|
]))
|
|
]),
|
|
(Token.FUNC, 'b', ['x'], [
|
|
(Token.RETURN, (Token.EXPR, [
|
|
(Token.ASSIGN, None, (Token.OPEXPR, [
|
|
(Token.MEMBER, (Token.ID, 'x'), None, None),
|
|
(Token.MEMBER, (Token.INT, 1), None, None),
|
|
(Token.OP, _OPERATORS['+'][1])
|
|
]), None)
|
|
]))
|
|
]),
|
|
(Token.FUNC, 'c', [], [
|
|
(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)
|
|
]))
|
|
]),
|
|
(Token.EXPR, [
|
|
(Token.ASSIGN, None, (Token.OPEXPR, [
|
|
(Token.MEMBER, (Token.ID, 'c'), None, (Token.CALL, [], None))
|
|
]), None)
|
|
])
|
|
]
|
|
}
|
|
]
|