kotovalexarian-likes-github/ruby--ruby
1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2022-11-09 12:17:21 -05:00
Code Releases Activity

ast.c: Fix defs

Browse source 
* ast.c (node_children): Add mid to children

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@63782 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
This commit is contained in:
yui-knk 2018-06-28 14:33:28 +00:00
parent 4aaceb29bc
commit 63e72a5850
2 changed files with 21 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
ast.c
View file

@ -374,9 +374,9 @@ node_children(rb_ast_t *ast, NODE *node)
case NODE_BLOCK_PASS:
return rb_ary_new_from_node_args(ast, 2, node->nd_head, node->nd_body);
case NODE_DEFN:
return rb_ary_new_from_node_args(ast, 1, node->nd_defn);
return rb_ary_new_from_args(2, ID2SYM(node->nd_mid), NEW_CHILD(ast, node->nd_defn));
case NODE_DEFS:
return rb_ary_new_from_node_args(ast, 2, node->nd_recv, node->nd_defn);
return rb_ary_new_from_args(3, NEW_CHILD(ast, node->nd_recv), ID2SYM(node->nd_mid), NEW_CHILD(ast, node->nd_defn));
case NODE_ALIAS:
return rb_ary_new_from_node_args(ast, 2, node->nd_1st, node->nd_2nd);
case NODE_VALIAS:

19
test/ruby/test_ast.rb
View file

@ -204,4 +204,23 @@ class TestAst < Test::Unit::TestCase
assert_equal(:foo, mid)
assert_nil(args)
end
def test_defn
node = RubyVM::AST.parse("def a; end")
_, _, body = *node.children
assert_equal("NODE_DEFN", body.type)
mid, defn = body.children
assert_equal(:a, mid)
assert_equal("NODE_SCOPE", defn.type)
end
def test_defs
node = RubyVM::AST.parse("def a.b; end")
_, _, body = *node.children
assert_equal("NODE_DEFS", body.type)
recv, mid, defn = body.children
assert_equal("NODE_VCALL", recv.type)
assert_equal(:b, mid)
assert_equal("NODE_SCOPE", defn.type)
end
end