update:codelens for proj test classfile

goplus · May 10, 2024 · a1907d5 · a1907d5
1 parent 7f8a235
commit a1907d5
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/gopls/internal/lsp/source/code_lens_gox.go b/gopls/internal/lsp/source/code_lens_gox.go
@@ -215,6 +215,14 @@ func goxTestCodeLens(pgf *ParsedGopFile, classType string) ([]protocol.CodeLens,
  if err != nil {
  return nil, err
  }
+ if pgf.File.IsProj && classType == "main" { //goxls:proj gox test
+ return []protocol.CodeLens{
+ {Range: rng, Command: &protocol.Command{
+ Title: "run test",
+ Command: "gop.test",
+ }},
+ }, nil
+ }
  args, err := command.MarshalArgs(
  map[string]string{
  "functionName": "Test_" + classType,
@@ -232,7 +240,6 @@ func goxTestCodeLens(pgf *ParsedGopFile, classType string) ([]protocol.CodeLens,
  Title: "run file tests",
  Command: "gop.test.cursor",
  Arguments: args,
- }},
- }
+ }}}
  return codelens, nil
 }