Make tracer integration skip IRB traces

ruby · Feb 7, 2024 · ee0ad19 · ee0ad19
1 parent ea4f147
commit ee0ad19
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/lib/irb/ext/tracer.rb b/lib/irb/ext/tracer.rb
@@ -13,6 +13,13 @@
 end
 
 module IRB
+  class CallTracer < ::CallTracer
+    IRB_DIR = File.expand_path('../..', __dir__)
+
+    def skip?(tp)
+      super || tp.path.match?(IRB_DIR) || tp.path.match?('<internal:prelude>')
+    end
+  end
   class WorkSpace
     alias __evaluate__ evaluate
     # Evaluate the context of this workspace and use the Tracer library to

diff --git a/test/irb/test_tracer.rb b/test/irb/test_tracer.rb
@@ -48,7 +48,7 @@ def test_use_tracer_enabled_when_gem_is_unavailable
 
       output = run_ruby_file do
         type "bar(Foo)"
-        type "exit!"
+        type "exit"
       end
 
       assert_include(output, "Tracer extension of IRB is enabled but tracer gem wasn't found.")
@@ -63,21 +63,24 @@ def test_use_tracer_enabled_when_gem_is_available
 
       output = run_ruby_file do
         type "bar(Foo)"
-        type "exit!"
+        type "exit"
       end
 
       assert_include(output, "Object#bar at")
       assert_include(output, "Foo.foo at")
       assert_include(output, "Foo.foo #=> 100")
       assert_include(output, "Object#bar #=> 100")
+
+      # Test that the tracer output does not include IRB's own files
+      assert_not_include(output, "irb/workspace.rb")
     end
 
     def test_use_tracer_is_disabled_by_default
       write_ruby example_ruby_file
 
       output = run_ruby_file do
         type "bar(Foo)"
-        type "exit!"
+        type "exit"
       end
 
       assert_not_include(output, "#depth:")