Merge pull request #765 from Shopify/fix-code-lens

Skip test code lens on methods unless its inside a test class

Author: st0012

lib/ruby_lsp/requests/code_lens.rb

@@ -80,11 +80,13 @@ def after_class(node)
 
       sig { params(node: SyntaxTree::DefNode).void }
       def on_def(node)
+        class_name = @class_stack.last
+        return unless class_name
+
         visibility, _ = @visibility_stack.last
         if visibility == "public"
           method_name = node.name.value
           if method_name.start_with?("test_")
-            class_name = T.must(@class_stack.last)
             add_test_code_lens(
               node,
               name: method_name,

test/fixtures/minitest_tests.rb

@@ -25,3 +25,7 @@ def test_with_q?; end
 class AnotherTest < Minitest::Test
   def test_public; end
 end
+
+class Foo
+  def test_public; end
+end