TMP<

lib/spoom/cfg.rb

@@ -605,6 +605,9 @@ def visit_or_node(node)
 
       sig { override.params(node: Prism::RescueNode).void }
       def visit_rescue_node(node)
+        puts node.inspect
+        visit_all(node.exceptions)
+        visit(node.reference)
         visit(node.statements)
       end
 

sorbet/config

@@ -1,4 +1,4 @@
 .
 --ignore=vendor/
---ignore=test/spoom/cfg/
+--ignore=test/spoom/cfg/fixtures
 --enable-experimental-requires-ancestor

test.rb

@@ -2,20 +2,6 @@
 # frozen_string_literal: true
 
 begin
-  x = 1
-  puts "inside"
-  if foo?
-    return
-    puts "dead"
-  else
-    return
-    puts "dead"
-  end
-
-  puts "dead"
-rescue
-  puts x
-  puts "rescue"
-  return
-  puts "dead"
+rescue MyException.new.class => e
+  puts e
 end

test/spoom/cfg/fixtures/begin_rescue_args.cfg

@@ -0,0 +1,13 @@
+cfg: <main>
+
+  bb#0
+    -> bb#3
+    -> bb#1
+
+  bb#3
+    MyException.new.class()
+    e
+    <self>.puts(e)
+    -> bb#1
+
+  bb#1

test/spoom/cfg/fixtures/begin_rescue_args.rb

@@ -0,0 +1,4 @@
+begin
+rescue MyException.new.class => e
+  puts e
+end

test/spoom/cfg/fixtures_test.rb

@@ -16,9 +16,7 @@ class CFGFixturesTest < Minitest::Test
       test_name = "test_#{name}"
 
       define_method(test_name) do
-        expected_path = File.join(dir, "#{name}.cfg.exp")
-        assert(File.exist?(expected_path), "expectation file missing for #{file}")
-        expected_output = File.read(expected_path)
+        T.bind(self, Minitest::Test)
 
         node = Spoom.parse_ruby(File.read(file), file: file)
         cfgs_cluster = Spoom::CFG.from_node(node)
@@ -31,7 +29,13 @@ class CFGFixturesTest < Minitest::Test
           puts
           puts actual_output
           cfgs_cluster.show_dot
-        elsif ENV["CFG_UPDATE_FIXTURES"]
+        end
+
+        expected_path = File.join(dir, "#{name}.cfg")
+        assert(File.exist?(expected_path), "expectation file missing for #{file}")
+        expected_output = File.read(expected_path)
+
+        if ENV["CFG_UPDATE_FIXTURES"]
           File.write(expected_path, actual_output)
         else
           unless expected_output == actual_output