class TestFunctionContext:
def setUp(self):
self.c = FunctionContext(None)
def test_arg_names_attr(self):
assert self.c.arg_names == []
def test_str(self):
assert str(self.c) == '''(function() {
})'''
def test_str_with_body_and_vars(self):
self.c.scope.declare('a', True)
self.c.scope.declare('b', True)
self.c.scope.declare('c', False)
self.c.add(' foo\n')
assert str(self.c) == '''(function() {
def setUp(self):
self.c = FunctionContext(None)
def setUp(self):
self.c = FunctionContext(None, arg_names=['a', 'b', 'c'])
def visit_FunctionDef(self, node):
if node.name:
self.stack[-1].scope.declare(node.name)
args = [get_arg_id(arg) for arg in node.args.args]
if node.args.kwarg is not None:
raise Exception('**kwargs not supported')
func = FunctionContext(self.stack[-1], args)
self.push(func)
# Emit vararg
if node.args.vararg is not None:
self.visit(
ast.Assign([ast.Name(node.args.vararg, ast.Store())],
ast_call(
ast_load('Array.prototype.slice.call'),
ast.Name('arguments', ast.Load()),
ast.Num(len(args)),
)))
# Emit default arguments
def_args = node.args.defaults
for arg_name, arg_val in zip(args[-len(def_args):], def_args):
self.block([
ast.If(
ast.Compare(
ast_call(
ast.Name('type', ast.Load()),
ast.Name(arg_name, ast.Load()),
),
[
ast.Eq(),
],
[
ast.Str('undefined'),
],
),
[
ast.Assign([ast.Name(arg_name, ast.Store())], arg_val),
],
[],
)
])
# Emit function body
self.block(node.body)
body = ast_call(
ast_load('JS'),
ast.Str(str(self.stack.pop())),
)
for decorator in node.decorator_list:
body = ast_call(decorator, body)
if not node.name:
self.visit(body)
else:
self.visit(ast.Assign(
[ast_load(node.name)],
body,
))
