From b8df6b9e01d8971235ea42fc9098ce392f791928 Mon Sep 17 00:00:00 2001
From: Jemma Issroff <jemmaissroff@gmail.com>
Date: Thu, 7 Dec 2023 10:08:16 -0500
Subject: [PRISM] Don't pop arguments on a return node

Since ReturnNodes don't get popped, their arguments shouldn't either
---
 test/ruby/test_compile_prism.rb | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'test')

diff --git a/test/ruby/test_compile_prism.rb b/test/ruby/test_compile_prism.rb
index 97ee5c2f70..42e2ebbf00 100644
--- a/test/ruby/test_compile_prism.rb
+++ b/test/ruby/test_compile_prism.rb
@@ -1009,6 +1009,15 @@ module Prism
         end
         prism_test_return_node
       CODE
+      assert_prism_eval(<<-CODE)
+        def self.prism_test_return_node
+          [1].map do |i|
+            return i if i == 1
+            2
+          end
+        end
+        prism_test_return_node
+      CODE
     end
 
     ############################################################################
-- 
cgit v1.2.3