[PRISM] Compile IndexAndWriteNode

1 files changed, 22 insertions, 0 deletions

diff --git a/test/ruby/test_compile_prism.rb b/test/ruby/test_compile_prism.rb
index 96aad1ab5b..2ee078673f 100644
--- a/test/ruby/test_compile_prism.rb
+++ b/test/ruby/test_compile_prism.rb
@@ -238,6 +238,28 @@ module Prism
       assert_prism_eval("$pit = 1")
     end
 
+    def test_IndexAndWriteNode
+      assert_prism_eval("[0][0] &&= 1")
+      assert_prism_eval("[nil][0] &&= 1")
+
+      # Testing `[]` with a block passed in
+      assert_prism_eval(<<-CODE)
+        class CustomHash < Hash
+          def []=(key, value, &block)
+            block ? super(block.call(key), value) : super(key, value)
+          end
+        end
+
+        hash = CustomHash.new
+
+        # Call the custom method with a block that modifies
+        # the key before assignment
+        hash["KEY"] = "test"
+        hash["key", &(Proc.new { _1.upcase })] &&= "value"
+        hash
+      CODE
+    end
+
     def test_IndexOrWriteNode
       assert_prism_eval("[0][0] ||= 1")
       assert_prism_eval("[nil][0] ||= 1")