1 files changed, 78 insertions, 0 deletions
diff --git a/test/irb/test_workspace.rb b/test/irb/test_workspace.rb
new file mode 100644
index 0000000000..f725c55c06
--- /dev/null
+++ b/test/irb/test_workspace.rb
@@ -0,0 +1,78 @@
+# frozen_string_literal: false
+require 'test/unit'
+require 'tempfile'
+require 'irb/workspace'
+
+module TestIRB
+  class TestWorkSpace < Test::Unit::TestCase
+    def test_code_around_bindinig
+      Tempfile.create do |f|
+        code = <<~RUBY
+          # 1
+          # 2
+          IRB::WorkSpace.new(binding) # 3
+          # 4
+          # 5
+        RUBY
+        f.print(code)
+        f.close
+
+        workspace = eval(code, binding, f.path)
+        assert_equal(<<~EOS, workspace.code_around_binding)
+
+          From: #{f.path} @ line 3 :
+
+              1: # 1
+              2: # 2
+           => 3: IRB::WorkSpace.new(binding) # 3
+              4: # 4
+              5: # 5
+
+        EOS
+      end
+    end
+
+    def test_code_around_bindinig_with_script_lines__
+      with_script_lines do |script_lines|
+        Tempfile.create do |f|
+          code = "IRB::WorkSpace.new(binding)\n"
+          script_lines[f.path] = code
+
+          workspace = eval(code, binding, f.path)
+          assert_equal(<<~EOS, workspace.code_around_binding)
+
+            From: #{f.path} @ line 1 :
+
+             => 1: IRB::WorkSpace.new(binding)
+
+          EOS
+        end
+      end
+    end
+
+    def test_code_around_bindinig_on_irb
+      workspace = eval("IRB::WorkSpace.new(binding)", binding, "(irb)")
+      assert_equal(nil, workspace.code_around_binding)
+    end
+
+    private
+
+    def with_script_lines
+      script_lines = nil
+      debug_lines = {}
+      Object.class_eval do
+        if defined?(SCRIPT_LINES__)
+          script_lines = SCRIPT_LINES__
+          remove_const :SCRIPT_LINES__
+        end
+        const_set(:SCRIPT_LINES__, debug_lines)
+      end
+      yield debug_lines
+    ensure
+      Object.class_eval do
+        remove_const :SCRIPT_LINES__
+        const_set(:SCRIPT_LINES__, script_lines) if script_lines
+      end
+    end
+  end
+end