1 files changed, 16 insertions, 0 deletions
diff --git a/gopls/internal/lsp/testdata/codelens/codelens_test.go b/gopls/internal/lsp/testdata/codelens/codelens_test.go
new file mode 100644
index 000000000..f6c696416
--- /dev/null
+++ b/gopls/internal/lsp/testdata/codelens/codelens_test.go
@@ -0,0 +1,16 @@
+package codelens //@codelens("package codelens", "run file benchmarks", "test")
+
+import "testing"
+
+func TestMain(m *testing.M) {} // no code lens for TestMain
+
+func TestFuncWithCodeLens(t *testing.T) { //@codelens("func", "run test", "test")
+}
+
+func thisShouldNotHaveACodeLens(t *testing.T) {
+}
+
+func BenchmarkFuncWithCodeLens(b *testing.B) { //@codelens("func", "run benchmark", "test")
+}
+
+func helper() {} // expect no code lens