Fix an error on multi next statements

ruby · Aug 22, 2024 · 3dad2f7 · 3dad2f7
1 parent 39e434f
commit 3dad2f7
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/lib/typeprof/core/graph/box.rb b/lib/typeprof/core/graph/box.rb
@@ -254,7 +254,7 @@ def show
   class EscapeBox < Box
     def initialize(node, genv, a_ret, f_ret)
       super(node)
-      @a_ret = a_ret
+      @a_ret = a_ret.new_vertex(genv, node)
       @f_ret = f_ret
       @f_ret.add_edge(genv, self)
     end

diff --git a/scenario/regressions/multi-next.rb b/scenario/regressions/multi-next.rb
@@ -0,0 +1,13 @@
+## update
+def check
+  ["str"].map do |x|
+    next x
+    next x
+    1
+  end
+end
+
+## assert
+class Object
+  def check: -> Array[Integer | String]
+end