| // Copyright 2020 The Go Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style |
| // license that can be found in the LICENSE file. |
| |
| package regtest |
| |
| import ( |
| "fmt" |
| "io/ioutil" |
| "os" |
| "path/filepath" |
| "strings" |
| "testing" |
| |
| "golang.org/x/tools/internal/lsp" |
| "golang.org/x/tools/internal/lsp/fake" |
| "golang.org/x/tools/internal/testenv" |
| ) |
| |
| const workspaceProxy = ` |
| -- example.com@v1.2.3/go.mod -- |
| module example.com |
| |
| go 1.12 |
| -- example.com@v1.2.3/blah/blah.go -- |
| package blah |
| |
| func SaySomething() { |
| fmt.Println("something") |
| } |
| -- random.org@v1.2.3/go.mod -- |
| module random.org |
| |
| go 1.12 |
| -- random.org@v1.2.3/bye/bye.go -- |
| package bye |
| |
| func Goodbye() { |
| println("Bye") |
| } |
| ` |
| |
| // TODO: Add a replace directive. |
| const workspaceModule = ` |
| -- pkg/go.mod -- |
| module mod.com |
| |
| go 1.14 |
| |
| require ( |
| example.com v1.2.3 |
| random.org v1.2.3 |
| ) |
| -- pkg/main.go -- |
| package main |
| |
| import ( |
| "example.com/blah" |
| "mod.com/inner" |
| "random.org/bye" |
| ) |
| |
| func main() { |
| blah.SaySomething() |
| inner.Hi() |
| bye.Goodbye() |
| } |
| -- pkg/main2.go -- |
| package main |
| |
| import "fmt" |
| |
| func _() { |
| fmt.Print("%s") |
| } |
| -- pkg/inner/inner.go -- |
| package inner |
| |
| import "example.com/blah" |
| |
| func Hi() { |
| blah.SaySomething() |
| } |
| -- goodbye/bye/bye.go -- |
| package bye |
| |
| func Bye() {} |
| -- goodbye/go.mod -- |
| module random.org |
| |
| go 1.12 |
| ` |
| |
| // Confirm that find references returns all of the references in the module, |
| // regardless of what the workspace root is. |
| func TestReferences(t *testing.T) { |
| for _, tt := range []struct { |
| name, rootPath string |
| }{ |
| { |
| name: "module root", |
| rootPath: "pkg", |
| }, |
| { |
| name: "subdirectory", |
| rootPath: "pkg/inner", |
| }, |
| } { |
| t.Run(tt.name, func(t *testing.T) { |
| opts := []RunOption{WithProxyFiles(workspaceProxy)} |
| if tt.rootPath != "" { |
| opts = append(opts, WithRootPath(tt.rootPath)) |
| } |
| withOptions(opts...).run(t, workspaceModule, func(t *testing.T, env *Env) { |
| f := "pkg/inner/inner.go" |
| env.OpenFile(f) |
| locations := env.References(f, env.RegexpSearch(f, `SaySomething`)) |
| want := 3 |
| if got := len(locations); got != want { |
| t.Fatalf("expected %v locations, got %v", want, got) |
| } |
| }) |
| }) |
| } |
| } |
| |
| // Make sure that analysis diagnostics are cleared for the whole package when |
| // the only opened file is closed. This test was inspired by the experience in |
| // VS Code, where clicking on a reference result triggers a |
| // textDocument/didOpen without a corresponding textDocument/didClose. |
| func TestClearAnalysisDiagnostics(t *testing.T) { |
| withOptions(WithProxyFiles(workspaceProxy), WithRootPath("pkg/inner")).run(t, workspaceModule, func(t *testing.T, env *Env) { |
| env.OpenFile("pkg/main.go") |
| env.Await( |
| env.DiagnosticAtRegexp("pkg/main2.go", "fmt.Print"), |
| ) |
| env.CloseBuffer("pkg/main.go") |
| env.Await( |
| EmptyDiagnostics("pkg/main2.go"), |
| ) |
| }) |
| } |
| |
| // This test checks that gopls updates the set of files it watches when a |
| // replace target is added to the go.mod. |
| func TestWatchReplaceTargets(t *testing.T) { |
| withOptions(WithProxyFiles(workspaceProxy), WithRootPath("pkg")).run(t, workspaceModule, func(t *testing.T, env *Env) { |
| // Add a replace directive and expect the files that gopls is watching |
| // to change. |
| dir := env.Sandbox.Workdir.URI("goodbye").SpanURI().Filename() |
| goModWithReplace := fmt.Sprintf(`%s |
| replace random.org => %s |
| `, env.ReadWorkspaceFile("pkg/go.mod"), dir) |
| env.WriteWorkspaceFile("pkg/go.mod", goModWithReplace) |
| env.Await( |
| CompletedWork(lsp.DiagnosticWorkTitle(lsp.FromDidChangeWatchedFiles), 1), |
| UnregistrationMatching("didChangeWatchedFiles"), |
| RegistrationMatching("didChangeWatchedFiles"), |
| ) |
| }) |
| } |
| |
| const workspaceModuleProxy = ` |
| -- example.com@v1.2.3/go.mod -- |
| module example.com |
| |
| go 1.12 |
| -- example.com@v1.2.3/blah/blah.go -- |
| package blah |
| |
| func SaySomething() { |
| fmt.Println("something") |
| } |
| -- b.com@v1.2.3/go.mod -- |
| module b.com |
| |
| go 1.12 |
| -- b.com@v1.2.3/b/b.go -- |
| package b |
| |
| func Hello() {} |
| ` |
| |
| func TestAutomaticWorkspaceModule_Interdependent(t *testing.T) { |
| const multiModule = ` |
| -- moda/a/go.mod -- |
| module a.com |
| |
| require b.com v1.2.3 |
| |
| -- moda/a/a.go -- |
| package a |
| |
| import ( |
| "b.com/b" |
| ) |
| |
| func main() { |
| var x int |
| _ = b.Hello() |
| } |
| -- modb/go.mod -- |
| module b.com |
| |
| -- modb/b/b.go -- |
| package b |
| |
| func Hello() int { |
| var x int |
| } |
| ` |
| withOptions( |
| WithProxyFiles(workspaceModuleProxy), |
| WithModes(Experimental), |
| ).run(t, multiModule, func(t *testing.T, env *Env) { |
| env.Await( |
| env.DiagnosticAtRegexp("moda/a/a.go", "x"), |
| env.DiagnosticAtRegexp("modb/b/b.go", "x"), |
| env.NoDiagnosticAtRegexp("moda/a/a.go", `"b.com/b"`), |
| ) |
| }) |
| } |
| |
| // This change tests that the version of the module used changes after it has |
| // been deleted from the workspace. |
| func TestDeleteModule_Interdependent(t *testing.T) { |
| const multiModule = ` |
| -- moda/a/go.mod -- |
| module a.com |
| |
| require b.com v1.2.3 |
| |
| -- moda/a/a.go -- |
| package a |
| |
| import ( |
| "b.com/b" |
| ) |
| |
| func main() { |
| var x int |
| _ = b.Hello() |
| } |
| -- modb/go.mod -- |
| module b.com |
| |
| -- modb/b/b.go -- |
| package b |
| |
| func Hello() int { |
| var x int |
| } |
| ` |
| withOptions( |
| WithProxyFiles(workspaceModuleProxy), |
| WithModes(Experimental), |
| ).run(t, multiModule, func(t *testing.T, env *Env) { |
| env.OpenFile("moda/a/a.go") |
| |
| original, _ := env.GoToDefinition("moda/a/a.go", env.RegexpSearch("moda/a/a.go", "Hello")) |
| if want := "modb/b/b.go"; !strings.HasSuffix(original, want) { |
| t.Errorf("expected %s, got %v", want, original) |
| } |
| env.CloseBuffer(original) |
| env.RemoveWorkspaceFile("modb/b/b.go") |
| env.RemoveWorkspaceFile("modb/go.mod") |
| env.Await( |
| CompletedWork(lsp.DiagnosticWorkTitle(lsp.FromDidChangeWatchedFiles), 2), |
| ) |
| got, _ := env.GoToDefinition("moda/a/a.go", env.RegexpSearch("moda/a/a.go", "Hello")) |
| if want := "b.com@v1.2.3/b/b.go"; !strings.HasSuffix(got, want) { |
| t.Errorf("expected %s, got %v", want, got) |
| } |
| }) |
| } |
| |
| // Tests that the version of the module used changes after it has been added |
| // to the workspace. |
| func TestCreateModule_Interdependent(t *testing.T) { |
| const multiModule = ` |
| -- moda/a/go.mod -- |
| module a.com |
| |
| require b.com v1.2.3 |
| |
| -- moda/a/a.go -- |
| package a |
| |
| import ( |
| "b.com/b" |
| ) |
| |
| func main() { |
| var x int |
| _ = b.Hello() |
| } |
| ` |
| withOptions( |
| WithModes(Experimental), |
| WithProxyFiles(workspaceModuleProxy), |
| ).run(t, multiModule, func(t *testing.T, env *Env) { |
| env.OpenFile("moda/a/a.go") |
| original, _ := env.GoToDefinition("moda/a/a.go", env.RegexpSearch("moda/a/a.go", "Hello")) |
| if want := "b.com@v1.2.3/b/b.go"; !strings.HasSuffix(original, want) { |
| t.Errorf("expected %s, got %v", want, original) |
| } |
| env.CloseBuffer(original) |
| env.WriteWorkspaceFiles(map[string]string{ |
| "modb/go.mod": "module b.com", |
| "modb/b/b.go": `package b |
| |
| func Hello() int { |
| var x int |
| } |
| `, |
| }) |
| env.Await( |
| OnceMet( |
| CompletedWork(lsp.DiagnosticWorkTitle(lsp.FromDidChangeWatchedFiles), 1), |
| env.DiagnosticAtRegexp("modb/b/b.go", "x"), |
| ), |
| ) |
| got, _ := env.GoToDefinition("moda/a/a.go", env.RegexpSearch("moda/a/a.go", "Hello")) |
| if want := "modb/b/b.go"; !strings.HasSuffix(got, want) { |
| t.Errorf("expected %s, got %v", want, original) |
| } |
| }) |
| } |
| |
| // This test confirms that a gopls workspace can recover from initialization |
| // with one invalid module. |
| func TestOneBrokenModule(t *testing.T) { |
| const multiModule = ` |
| -- moda/a/go.mod -- |
| module a.com |
| |
| require b.com v1.2.3 |
| |
| -- moda/a/a.go -- |
| package a |
| |
| import ( |
| "b.com/b" |
| ) |
| |
| func main() { |
| var x int |
| _ = b.Hello() |
| } |
| -- modb/go.mod -- |
| modul b.com // typo here |
| |
| -- modb/b/b.go -- |
| package b |
| |
| func Hello() int { |
| var x int |
| } |
| ` |
| withOptions( |
| WithProxyFiles(workspaceModuleProxy), |
| WithModes(Experimental), |
| ).run(t, multiModule, func(t *testing.T, env *Env) { |
| env.OpenFile("modb/go.mod") |
| env.Await( |
| OnceMet( |
| CompletedWork(lsp.DiagnosticWorkTitle(lsp.FromDidOpen), 1), |
| DiagnosticAt("modb/go.mod", 0, 0), |
| ), |
| ) |
| env.RegexpReplace("modb/go.mod", "modul", "module") |
| env.Editor.SaveBufferWithoutActions(env.Ctx, "modb/go.mod") |
| env.Await( |
| env.DiagnosticAtRegexp("modb/b/b.go", "x"), |
| ) |
| }) |
| } |
| |
| func TestUseGoplsMod(t *testing.T) { |
| // This test validates certain functionality related to using a gopls.mod |
| // file to specify workspace modules. |
| testenv.NeedsGo1Point(t, 14) |
| const multiModule = ` |
| -- moda/a/go.mod -- |
| module a.com |
| |
| require b.com v1.2.3 |
| |
| -- moda/a/a.go -- |
| package a |
| |
| import ( |
| "b.com/b" |
| ) |
| |
| func main() { |
| var x int |
| _ = b.Hello() |
| } |
| -- modb/go.mod -- |
| module b.com |
| |
| require example.com v1.2.3 |
| -- modb/b/b.go -- |
| package b |
| |
| func Hello() int { |
| var x int |
| } |
| -- gopls.mod -- |
| module gopls-workspace |
| |
| require ( |
| a.com v0.0.0-goplsworkspace |
| b.com v1.2.3 |
| ) |
| |
| replace a.com => $SANDBOX_WORKDIR/moda/a |
| ` |
| withOptions( |
| WithProxyFiles(workspaceModuleProxy), |
| WithModes(Experimental), |
| ).run(t, multiModule, func(t *testing.T, env *Env) { |
| // Initially, the gopls.mod should cause only the a.com module to be |
| // loaded. Validate this by jumping to a definition in b.com and ensuring |
| // that we go to the module cache. |
| env.OpenFile("moda/a/a.go") |
| location, _ := env.GoToDefinition("moda/a/a.go", env.RegexpSearch("moda/a/a.go", "Hello")) |
| if want := "b.com@v1.2.3/b/b.go"; !strings.HasSuffix(location, want) { |
| t.Errorf("expected %s, got %v", want, location) |
| } |
| workdir := env.Sandbox.Workdir.RootURI().SpanURI().Filename() |
| |
| // Now, modify the gopls.mod file on disk to activate the b.com module in |
| // the workspace. |
| env.WriteWorkspaceFile("gopls.mod", fmt.Sprintf(`module gopls-workspace |
| |
| require ( |
| a.com v0.0.0-goplsworkspace |
| b.com v0.0.0-goplsworkspace |
| ) |
| |
| replace a.com => %s/moda/a |
| replace b.com => %s/modb |
| `, workdir, workdir)) |
| env.Await( |
| OnceMet( |
| CompletedWork(lsp.DiagnosticWorkTitle(lsp.FromDidChangeWatchedFiles), 1), |
| env.DiagnosticAtRegexp("modb/b/b.go", "x"), |
| ), |
| ) |
| env.OpenFile("modb/go.mod") |
| // Check that go.mod diagnostics picked up the newly active mod file. |
| env.Await(env.DiagnosticAtRegexp("modb/go.mod", `require example.com v1.2.3`)) |
| // ...and that jumping to definition now goes to b.com in the workspace. |
| location, _ = env.GoToDefinition("moda/a/a.go", env.RegexpSearch("moda/a/a.go", "Hello")) |
| if want := "modb/b/b.go"; !strings.HasSuffix(location, want) { |
| t.Errorf("expected %s, got %v", want, location) |
| } |
| |
| // Now, let's modify the gopls.mod *overlay* (not on disk), and verify that |
| // this change is also picked up. |
| env.OpenFile("gopls.mod") |
| env.SetBufferContent("gopls.mod", fmt.Sprintf(`module gopls-workspace |
| |
| require ( |
| a.com v0.0.0-goplsworkspace |
| ) |
| |
| replace a.com => %s/moda/a |
| `, workdir)) |
| env.Await(CompletedWork(lsp.DiagnosticWorkTitle(lsp.FromDidChange), 1)) |
| // TODO: diagnostics are not being cleared from the old go.mod location, |
| // because it's not treated as a 'deleted' file. Uncomment this after |
| // fixing. |
| /* |
| env.Await(OnceMet( |
| CompletedWork(lsp.DiagnosticWorkTitle(lsp.FromDidChange), 1), |
| EmptyDiagnostics("modb/go.mod"), |
| )) |
| */ |
| |
| // Just as before, check that we now jump to the module cache. |
| location, _ = env.GoToDefinition("moda/a/a.go", env.RegexpSearch("moda/a/a.go", "Hello")) |
| if want := "b.com@v1.2.3/b/b.go"; !strings.HasSuffix(location, want) { |
| t.Errorf("expected %s, got %v", want, location) |
| } |
| }) |
| } |
| |
| func TestNonWorkspaceFileCreation(t *testing.T) { |
| testenv.NeedsGo1Point(t, 13) |
| |
| const files = ` |
| -- go.mod -- |
| module mod.com |
| |
| -- x.go -- |
| package x |
| ` |
| |
| const code = ` |
| package foo |
| import "fmt" |
| var _ = fmt.Printf |
| ` |
| run(t, files, func(t *testing.T, env *Env) { |
| env.CreateBuffer("/tmp/foo.go", "") |
| env.EditBuffer("/tmp/foo.go", fake.NewEdit(0, 0, 0, 0, code)) |
| env.GoToDefinition("/tmp/foo.go", env.RegexpSearch("/tmp/foo.go", `Printf`)) |
| }) |
| } |
| |
| func TestMultiModuleV2(t *testing.T) { |
| const multiModule = ` |
| -- moda/a/go.mod -- |
| module a.com |
| |
| require b.com/v2 v2.0.0 |
| -- moda/a/a.go -- |
| package a |
| |
| import ( |
| "b.com/v2/b" |
| ) |
| |
| func main() { |
| var x int |
| _ = b.Hi() |
| } |
| -- modb/go.mod -- |
| module b.com |
| |
| -- modb/b/b.go -- |
| package b |
| |
| func Hello() int { |
| var x int |
| } |
| -- modb/v2/go.mod -- |
| module b.com/v2 |
| |
| -- modb/v2/b/b.go -- |
| package b |
| |
| func Hi() int { |
| var x int |
| } |
| -- modc/go.mod -- |
| module gopkg.in/yaml.v1 // test gopkg.in versions |
| -- modc/main.go -- |
| package main |
| |
| func main() { |
| var x int |
| } |
| ` |
| withOptions( |
| WithModes(Experimental), |
| ).run(t, multiModule, func(t *testing.T, env *Env) { |
| env.Await( |
| env.DiagnosticAtRegexp("moda/a/a.go", "x"), |
| env.DiagnosticAtRegexp("modb/b/b.go", "x"), |
| env.DiagnosticAtRegexp("modb/v2/b/b.go", "x"), |
| env.DiagnosticAtRegexp("modc/main.go", "x"), |
| ) |
| }) |
| } |
| |
| func TestWorkspaceDirAccess(t *testing.T) { |
| const multiModule = ` |
| -- moda/a/go.mod -- |
| module a.com |
| |
| -- moda/a/a.go -- |
| package main |
| |
| func main() { |
| fmt.Println("Hello") |
| } |
| -- modb/go.mod -- |
| module b.com |
| -- modb/b/b.go -- |
| package main |
| |
| func main() { |
| fmt.Println("World") |
| } |
| ` |
| withOptions( |
| WithModes(Experimental), |
| SendPID(), |
| ).run(t, multiModule, func(t *testing.T, env *Env) { |
| pid := os.Getpid() |
| // Don't factor this out of Server.addFolders. vscode-go expects this |
| // directory. |
| modPath := filepath.Join(os.TempDir(), fmt.Sprintf("gopls-%d.workspace", pid), "go.mod") |
| gotb, err := ioutil.ReadFile(modPath) |
| if err != nil { |
| t.Fatalf("reading expected workspace modfile: %v", err) |
| } |
| got := string(gotb) |
| for _, want := range []string{"a.com v0.0.0-goplsworkspace", "b.com v0.0.0-goplsworkspace"} { |
| if !strings.Contains(got, want) { |
| // want before got here, since the go.mod is multi-line |
| t.Fatalf("workspace go.mod missing %q. got:\n%s", want, got) |
| } |
| } |
| workdir := env.Sandbox.Workdir.RootURI().SpanURI().Filename() |
| env.WriteWorkspaceFile("gopls.mod", fmt.Sprintf(` |
| module gopls-workspace |
| |
| require ( |
| a.com v0.0.0-goplsworkspace |
| ) |
| |
| replace a.com => %s/moda/a |
| `, workdir)) |
| env.Await(CompletedWork(lsp.DiagnosticWorkTitle(lsp.FromDidChangeWatchedFiles), 1)) |
| gotb, err = ioutil.ReadFile(modPath) |
| if err != nil { |
| t.Fatalf("reading expected workspace modfile: %v", err) |
| } |
| got = string(gotb) |
| want := "b.com v0.0.0-goplsworkspace" |
| if strings.Contains(got, want) { |
| t.Fatalf("workspace go.mod contains unexpected %q. got:\n%s", want, got) |
| } |
| }) |
| } |