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)
            ])
        ]
    }
]