Fix up blocks for index nodes in ripper translation

kddnewton · kddnewton · commit a35eadce478d · 2024-03-06T01:12:57.000-05:00
diff --git a/lib/prism/translation/ripper.rb b/lib/prism/translation/ripper.rb
@@ -1538,7 +1538,7 @@ def visit_in_node(node)
       # ^^^^^^^^^^^^^^^
       def visit_index_operator_write_node(node)
         receiver = visit(node.receiver)
-        arguments = visit(node.arguments)
+        arguments, _ = visit_call_node_arguments(node.arguments, node.block, trailing_comma?(node.arguments&.location || node.location, node.closing_loc))
 
         bounds(node.location)
         target = on_aref_field(receiver, arguments)
@@ -1555,7 +1555,7 @@ def visit_index_operator_write_node(node)
       # ^^^^^^^^^^^^^^^^
       def visit_index_and_write_node(node)
         receiver = visit(node.receiver)
-        arguments = visit(node.arguments)
+        arguments, _ = visit_call_node_arguments(node.arguments, node.block, trailing_comma?(node.arguments&.location || node.location, node.closing_loc))
 
         bounds(node.location)
         target = on_aref_field(receiver, arguments)
@@ -1572,7 +1572,7 @@ def visit_index_and_write_node(node)
       # ^^^^^^^^^^^^^^^^
       def visit_index_or_write_node(node)
         receiver = visit(node.receiver)
-        arguments = visit(node.arguments)
+        arguments, _ = visit_call_node_arguments(node.arguments, node.block, trailing_comma?(node.arguments&.location || node.location, node.closing_loc))
 
         bounds(node.location)
         target = on_aref_field(receiver, arguments)
@@ -1589,7 +1589,7 @@ def visit_index_or_write_node(node)
       # ^^^^^^^^
       def visit_index_target_node(node)
         receiver = visit(node.receiver)
-        arguments = visit(node.arguments)
+        arguments, _ = visit_call_node_arguments(node.arguments, node.block, trailing_comma?(node.arguments&.location || node.location, node.closing_loc))
 
         bounds(node.location)
         on_aref_field(receiver, arguments)
diff --git a/test/prism/ripper_test.rb b/test/prism/ripper_test.rb
@@ -29,7 +29,6 @@ class RipperTest < TestCase
     ]
 
     skips = incorrect | %w[
-      arrays.txt
       dos_endings.txt
       embdoc_no_newline_at_end.txt
       heredocs_leading_whitespace.txt

Original file line number	Diff line number	Diff line change
`@@ -29,7 +29,6 @@ class RipperTest < TestCase`
`29`	`29`	`]`
`30`	`30`
`31`	`31`	`skips = incorrect \| %w[`
`32`		`- arrays.txt`
`33`	`32`	`dos_endings.txt`
`34`	`33`	`embdoc_no_newline_at_end.txt`
`35`	`34`	`heredocs_leading_whitespace.txt`