Implement Index Operator Write Node locations

lib/ast_to_prism/parser.rb

@@ -1242,6 +1242,7 @@ def convert_node(node, block: nil)
         not_supported(node)
       when :OP_ASGN1
         nd_recv, nd_mid, nd_index, nd_rvalue = node.children
+        loc, call_operator_loc, opening_loc, closing_loc, binary_operator_loc = node.locations
         receiver = convert_node(nd_recv)
         arguments, _ = convert_arguments(nd_index)
         # NOTE: Can block not be nil?
@@ -1250,18 +1251,18 @@ def convert_node(node, block: nil)
         value = convert_node(nd_rvalue)
 
         Prism::IndexOperatorWriteNode.new(
-          source,          # source
-          0,               # flags
-          receiver,        # receiver
-          null_location,   # call_operator_loc
-          null_location,   # opening_loc
-          arguments,       # arguments
-          null_location,   # closing_loc
-          block,           # block
-          binary_operator, # binary_operator
-          null_location,   # binary_operator_loc
-          value,           # value
-          location(node)   # location
+          source,                        # source
+          0,                             # flags
+          receiver,                      # receiver
+          location(call_operator_loc),   # call_operator_loc
+          location(opening_loc),         # opening_loc
+          arguments,                     # arguments
+          location(closing_loc),         # closing_loc
+          block,                         # block
+          binary_operator,               # binary_operator
+          location(binary_operator_loc), # binary_operator_loc
+          value,                         # value
+          location(loc)                  # location
         )
       when :OP_ASGN2
         nd_recv, nd_aid, nd_vid, nd_mid, nd_value = node.children

spec/basic_spec.rb

@@ -651,16 +651,12 @@ def test_code(code)
 
   describe "array assignment with operator (OP_ASGN1)" do
     it "tests" do
-      pending "IndexOperatorWriteNode locations are not supported"
-
       test_code(<<~CODE)
         ary[1] += foo
       CODE
     end
 
     it "tests" do
-      pending "IndexOperatorWriteNode locations are not supported"
-
       test_code(<<~CODE)
         ary[1, 2] += foo
       CODE