Fix pushing incorrect test item as code lens

Author: vinistock

lib/ruby_lsp/listeners/test_style.rb

@@ -202,7 +202,7 @@ def on_def_node_enter(node)
           framework: @framework,
         )
         test_item.add(example_item)
-        @response_builder.add_code_lens(test_item)
+        @response_builder.add_code_lens(example_item)
       end
 
       #: (Prism::CallNode node) -> void

lib/ruby_lsp/response_builders/test_collection.rb

@@ -25,8 +25,12 @@ def add(item)
 
       #: (ResponseType item) -> void
       def add_code_lens(item)
-        range = item.range
         arguments = [item.uri.to_standardized_path, item.id]
+        start = item.range.start
+        range = Interface::Range.new(
+          start: start,
+          end: Interface::Position.new(line: start.line, character: start.character + 1),
+        )
 
         @code_lens << Interface::CodeLens.new(
           range: range,
@@ -40,7 +44,7 @@ def add_code_lens(item)
         @code_lens << Interface::CodeLens.new(
           range: range,
           command: Interface::Command.new(
-            title: "▶ Run In Terminal",
+            title: "▶ Run in terminal",
             command: "rubyLsp.runTestInTerminal",
             arguments: arguments,
           ),

test/requests/discover_tests_test.rb

@@ -97,6 +97,90 @@ def test_hello; end
       end
     end
 
+    def test_collects_code_lenses
+      source = <<~RUBY
+        module Foo
+          class MyTest < Test::Unit::TestCase
+            def test_something; end
+
+            def test_something_else; end
+          end
+        end
+      RUBY
+
+      with_server(source) do |server, uri|
+        server.global_state.index.index_single(URI("/other_file.rb"), <<~RUBY)
+          module Test
+            module Unit
+              class TestCase; end
+            end
+          end
+        RUBY
+
+        server.global_state.stubs(:enabled_feature?).returns(true)
+
+        server.process_message(id: 1, method: "textDocument/codeLens", params: {
+          textDocument: { uri: uri },
+        })
+
+        # Discard the indexing log message
+        server.pop_response
+        items = get_response(server)
+        assert_equal(9, items.length)
+
+        # MyTest
+        assert_equal("▶ Run", items[0].command.title)
+        assert_equal(
+          { start: { line: 1, character: 2 }, end: { line: 1, character: 3 } },
+          JSON.parse(items[0].range.to_json, symbolize_names: true),
+        )
+        assert_equal("▶ Run in terminal", items[1].command.title)
+        assert_equal(
+          { start: { line: 1, character: 2 }, end: { line: 1, character: 3 } },
+          JSON.parse(items[1].range.to_json, symbolize_names: true),
+        )
+        assert_equal("⚙ Debug", items[2].command.title)
+        assert_equal(
+          { start: { line: 1, character: 2 }, end: { line: 1, character: 3 } },
+          JSON.parse(items[2].range.to_json, symbolize_names: true),
+        )
+
+        # test_something
+        assert_equal("▶ Run", items[3].command.title)
+        assert_equal(
+          { start: { line: 2, character: 4 }, end: { line: 2, character: 5 } },
+          JSON.parse(items[3].range.to_json, symbolize_names: true),
+        )
+        assert_equal("▶ Run in terminal", items[4].command.title)
+        assert_equal(
+          { start: { line: 2, character: 4 }, end: { line: 2, character: 5 } },
+          JSON.parse(items[4].range.to_json, symbolize_names: true),
+        )
+        assert_equal("⚙ Debug", items[5].command.title)
+        assert_equal(
+          { start: { line: 2, character: 4 }, end: { line: 2, character: 5 } },
+          JSON.parse(items[5].range.to_json, symbolize_names: true),
+        )
+
+        # test_something_else
+        assert_equal("▶ Run", items[6].command.title)
+        assert_equal(
+          { start: { line: 4, character: 4 }, end: { line: 4, character: 5 } },
+          JSON.parse(items[6].range.to_json, symbolize_names: true),
+        )
+        assert_equal("▶ Run in terminal", items[7].command.title)
+        assert_equal(
+          { start: { line: 4, character: 4 }, end: { line: 4, character: 5 } },
+          JSON.parse(items[7].range.to_json, symbolize_names: true),
+        )
+        assert_equal("⚙ Debug", items[8].command.title)
+        assert_equal(
+          { start: { line: 4, character: 4 }, end: { line: 4, character: 5 } },
+          JSON.parse(items[8].range.to_json, symbolize_names: true),
+        )
+      end
+    end
+
     def test_ignores_minitest_tests_that_extend_active_support_declarative
       source = <<~RUBY
         class MyTest < ActiveSupport::TestCase