diff options
| -rw-r--r-- | ast.c | 4 | ||||
| -rw-r--r-- | test/ruby/test_ast.rb | 8 |
2 files changed, 12 insertions, 0 deletions
@@ -824,6 +824,10 @@ node_locations(VALUE ast_value, const NODE *node) return rb_ary_new_from_args(2, location_new(nd_code_loc(node)), location_new(&RNODE_FLIP2(node)->operator_loc)); + case NODE_FLIP3: + return rb_ary_new_from_args(2, + location_new(nd_code_loc(node)), + location_new(&RNODE_FLIP3(node)->operator_loc)); case NODE_LAMBDA: return rb_ary_new_from_args(4, location_new(nd_code_loc(node)), diff --git a/test/ruby/test_ast.rb b/test/ruby/test_ast.rb index cb48efb75a..898c4fcc3d 100644 --- a/test/ruby/test_ast.rb +++ b/test/ruby/test_ast.rb @@ -1420,6 +1420,14 @@ dummy assert_locations(node.children[-1].children[0].locations, [[1, 3, 1, 7], [1, 4, 1, 6]]) end + def test_flip3_locations + node = ast_parse("if 'a'...('z'); foo; end") + assert_locations(node.children[-1].children[0].locations, [[1, 3, 1, 14], [1, 6, 1, 9]]) + + node = ast_parse('if 1...5; foo; end') + assert_locations(node.children[-1].children[0].locations, [[1, 3, 1, 8], [1, 4, 1, 7]]) + end + def test_lambda_locations node = ast_parse("-> (a, b) { foo }") assert_locations(node.children[-1].locations, [[1, 0, 1, 17], [1, 0, 1, 2], [1, 10, 1, 11], [1, 16, 1, 17]]) |