diff options
author | manga_osyo <manga.osyo@gmail.com> | 2020-07-05 20:58:55 +0900 |
---|---|---|
committer | Nobuyoshi Nakada <nobu@ruby-lang.org> | 2020-07-06 00:48:15 +0900 |
commit | 8e189df32cacef4b21e357a590d349330532baf5 (patch) | |
tree | 16c98959902e4a00baced105bf566d6bcf58beb0 | |
parent | 988b4bc9eab1e22e4bef7812e878c8ef4ed5ac0f (diff) | |
download | ruby-8e189df32cacef4b21e357a590d349330532baf5.tar.gz |
Add operator info to `OP_ASGN2` of `RubyVM::AbstractSyntaxTree`.
-rw-r--r-- | ast.c | 3 | ||||
-rw-r--r-- | test/ruby/test_ast.rb | 11 |
2 files changed, 13 insertions, 1 deletions
@@ -398,9 +398,10 @@ node_children(rb_ast_t *ast, NODE *node) NEW_CHILD(ast, node->nd_args->nd_head), NEW_CHILD(ast, node->nd_args->nd_body)); case NODE_OP_ASGN2: - return rb_ary_new_from_args(4, NEW_CHILD(ast, node->nd_recv), + return rb_ary_new_from_args(5, NEW_CHILD(ast, node->nd_recv), node->nd_next->nd_aid ? Qtrue : Qfalse, ID2SYM(node->nd_next->nd_vid), + ID2SYM(node->nd_next->nd_mid), NEW_CHILD(ast, node->nd_value)); case NODE_OP_ASGN_AND: return rb_ary_new_from_args(3, NEW_CHILD(ast, node->nd_head), ID2SYM(idANDOP), diff --git a/test/ruby/test_ast.rb b/test/ruby/test_ast.rb index 147d05eadf..da7dde4a38 100644 --- a/test/ruby/test_ast.rb +++ b/test/ruby/test_ast.rb @@ -319,4 +319,15 @@ class TestAst < Test::Unit::TestCase helper.validate_range assert_equal([], helper.errors) end + + def test_op_asgn2 + node = RubyVM::AbstractSyntaxTree.parse("struct.field += foo") + _, _, body = *node.children + assert_equal(:OP_ASGN2, body.type) + recv, _, mid, op, value = body.children + assert_equal(:VCALL, recv.type) + assert_equal(:field, mid) + assert_equal(:+, op) + assert_equal(:VCALL, value.type) + end end |