wire: make tests work with Go modules (google/go-cloud#331)

Generates go.mod files for test cases to support Go modules.

internal/wire/testdata/BuildTagsMainOnly/foo/wire.go

@@ -17,8 +17,7 @@
 package main
 
 import (
-	"bar"
-
+	"example.com/bar"
 	"github.com/google/go-cloud/wire"
 )
 

internal/wire/testdata/BuildTagsMainOnly/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/BuildTagsRelativePkg/foo/wire.go

@@ -17,8 +17,7 @@
 package main
 
 import (
-	"bar"
-
+	"example.com/bar"
 	"github.com/google/go-cloud/wire"
 )
 

internal/wire/testdata/BuildTagsRelativePkg/pkg

@@ -1 +1 @@
-./foo
+./example.com/foo

internal/wire/testdata/Chain/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/Cleanup/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/CopyOtherDecls/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/Cycle/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/EmptyVar/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/ExportedValue/foo/wire.go

@@ -17,7 +17,7 @@
 package main
 
 import (
-	"bar"
+	"example.com/bar"
 	"github.com/google/go-cloud/wire"
 )
 

internal/wire/testdata/ExportedValue/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/ExportedValue/want/wire_gen.go

@@ -6,7 +6,7 @@
 package main
 
 import (
-	bar "bar"
+	bar "example.com/bar"
 )
 
 // Injectors from wire.go:

internal/wire/testdata/ExportedValueDifferentPackage/foo/wire.go

@@ -19,7 +19,7 @@ package main
 import (
 	"os"
 
-	"bar"
+	"example.com/bar"
 	"github.com/google/go-cloud/wire"
 )
 

internal/wire/testdata/ExportedValueDifferentPackage/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/ImportedInterfaceBinding/bar/bar.go

@@ -17,7 +17,7 @@ package main
 import (
 	"fmt"
 
-	"foo"
+	"example.com/foo"
 	"github.com/google/go-cloud/wire"
 )
 

internal/wire/testdata/ImportedInterfaceBinding/bar/wire.go

@@ -17,7 +17,7 @@
 package main
 
 import (
-	"foo"
+	"example.com/foo"
 	"github.com/google/go-cloud/wire"
 )
 

internal/wire/testdata/ImportedInterfaceBinding/pkg

@@ -1 +1 @@
-bar
+example.com/bar

internal/wire/testdata/ImportedInterfaceBinding/want/wire_gen.go

@@ -6,7 +6,7 @@
 package main
 
 import (
-	foo "foo"
+	foo "example.com/foo"
 )
 
 // Injectors from wire.go:

internal/wire/testdata/InjectInput/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/InjectInputConflict/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/InjectWithPanic/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/InterfaceBinding/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/InterfaceBindingReuse/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/InterfaceValue/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/InterfaceValueInvalidArg0/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/InterfaceValueNotEnoughArgs/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/MultipleMissingInputs/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/MultipleMissingInputs/want/wire_errs.txt

@@ -1,2 +1,2 @@
-no provider found for foo.Foo
-no provider found for foo.Bar
+no provider found for example.com/foo.Foo
+no provider found for example.com/foo.Bar

internal/wire/testdata/NamingWorstCase/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/NamingWorstCaseAllInOne/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/NiladicIdentity/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/NiladicValue/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/NoImplicitInterface/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/NoInjectParamNames/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/NoopBuild/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/PartialCleanup/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/PkgImport/foo/foo.go

@@ -17,7 +17,7 @@ package main
 import (
 	"fmt"
 
-	"bar"
+	"example.com/bar"
 	"github.com/google/go-cloud/wire"
 )
 

internal/wire/testdata/PkgImport/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/PkgImport/want/wire_gen.go

@@ -6,7 +6,7 @@
 package main
 
 import (
-	bar "bar"
+	bar "example.com/bar"
 )
 
 // Injectors from wire.go:

internal/wire/testdata/ReturnError/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/Struct/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/StructPointer/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/TwoDeps/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/UnexportedValue/foo/wire.go

@@ -17,7 +17,7 @@
 package main
 
 import (
-	"bar"
+	"example.com/bar"
 	"github.com/google/go-cloud/wire"
 )
 

internal/wire/testdata/UnexportedValue/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/UnusedProviders/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/UnusedProviders/want/wire_errs.txt

@@ -1,4 +1,4 @@
 unused provider set "unusedSet"
 unused provider "provideUnused"
 unused value of type string
-unused interface binding to type foo.Fooer
+unused interface binding to type example.com/foo.Fooer

internal/wire/testdata/ValueChain/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/ValueConversion/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/ValueIsInterfaceValue/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/VarValue/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/Vendor/foo/wire.go

@@ -17,7 +17,7 @@
 package main
 
 import (
-	"bar"
+	"example.com/bar"
 	"github.com/google/go-cloud/wire"
 )
 

internal/wire/testdata/Vendor/pkg

@@ -1 +1 @@
-foo
+example.com/foo

internal/wire/testdata/Vendor/want/wire_gen.go

@@ -6,7 +6,7 @@
 package main
 
 import (
-	bar "bar"
+	bar "example.com/bar"
 )
 
 // Injectors from wire.go:

internal/wire/wire_test.go

@@ -71,6 +71,10 @@ func TestWire(t *testing.T) {
 	for _, test := range tests {
 		test := test
 		t.Run(test.name, func(t *testing.T) {
+			if test.name == "Vendor" && os.Getenv("GO111MODULE") != "off" {
+				// TODO: Remove the GO111MODULE check when it is not relevant (maybe after Go 1.12).
+				t.Skip("Skipped testing for vendored package for Go module turned on, see https://github.com/google/go-cloud/issues/326")
+			}
 			t.Parallel()
 
 			// Run Wire from a fake build context.
@@ -147,6 +151,9 @@ func goBuildCheck(test *testCase, wd string, bctx *build.Context, gen []byte) er
 			return err
 		}
 	}
+	if err := writeGoMod(gopath); err != nil {
+		return err
+	}
 	testExePath := filepath.Join(gopath, "bin", "testprog")
 	realBuildCtx := &build.Context{
 		GOARCH:      bctx.GOARCH,
@@ -158,7 +165,8 @@ func goBuildCheck(test *testCase, wd string, bctx *build.Context, gen []byte) er
 		BuildTags:   bctx.BuildTags,
 		ReleaseTags: bctx.ReleaseTags,
 	}
-	if err := runGo(realBuildCtx, "build", "-o", testExePath, genPkg.ImportPath); err != nil {
+	buildDir := filepath.Join(gopath, "src", genPkg.ImportPath)
+	if err := runGo(realBuildCtx, buildDir, "build", "-o", testExePath); err != nil {
 		return fmt.Errorf("build: %v", err)
 	}
 
@@ -342,7 +350,7 @@ func loadTestCase(root string, wireGoSrc []byte) (*testCase, error) {
 		if !*setup.Record {
 			wantWireOutput, err = ioutil.ReadFile(filepath.Join(root, "want", "wire_gen.go"))
 			if err != nil {
-				return nil, fmt.Errorf("load test case %s: %v. If this is a new testcase, run with -record to generate the wire_gen.go file.", name, err)
+				return nil, fmt.Errorf("load test case %s: %v, if this is a new testcase, run with -record to generate the wire_gen.go file", name, err)
 			}
 		}
 		wantProgramOutput, err = ioutil.ReadFile(filepath.Join(root, "want", "program_out.txt"))
@@ -368,7 +376,7 @@ func loadTestCase(root string, wireGoSrc []byte) (*testCase, error) {
 		if err != nil {
 			return err
 		}
-		goFiles[rel] = data
+		goFiles[filepath.Join("example.com", rel)] = data
 		return nil
 	})
 	if err != nil {
@@ -576,10 +584,45 @@ func (test *testCase) materialize(gopath string) error {
 	return nil
 }
 
-func runGo(bctx *build.Context, args ...string) error {
+// writeGoMod generates go.mod files for the test package and its dependency.
+// The file structure looks like:
+//
+//	gopath/src/
+//
+//		example.com/
+//
+//			go.mod
+//				replaces dependency with local copied one
+//
+//			... (Packages to be built and tested)
+//				any Go files copied recursively
+//
+//		github.com/google/go-cloud/
+//
+//			go.mod
+//
+//			... (Dependency files copied)
+func writeGoMod(gopath string) error {
+	importPath := "example.com"
+	depPath := "github.com/google/go-cloud"
+	depLoc := filepath.Join(gopath, "src", filepath.FromSlash(depPath))
+	example := fmt.Sprintf("module %s\n\nreplace %s => %s\n", importPath, depPath, depLoc)
+	gomod := filepath.Join(gopath, "src", importPath, "go.mod")
+	if err := ioutil.WriteFile(gomod, []byte(example), 0666); err != nil {
+		return fmt.Errorf("generate go.mod for %s: %v", gomod, err)
+	}
+	if err := ioutil.WriteFile(filepath.Join(depLoc, "go.mod"), []byte("module "+depPath), 0666); err != nil {
+		return fmt.Errorf("generate go.mod for %s: %v", depPath, err)
+	}
+	return nil
+}
+
+// runGo runs a go command in dir.
+func runGo(bctx *build.Context, dir string, args ...string) error {
 	exe := filepath.Join(bctx.GOROOT, "bin", "go")
 	c := exec.Command(exe, args...)
 	c.Env = append(os.Environ(), "GOROOT="+bctx.GOROOT, "GOARCH="+bctx.GOARCH, "GOOS="+bctx.GOOS, "GOPATH="+bctx.GOPATH)
+	c.Dir = dir
 	if bctx.CgoEnabled {
 		c.Env = append(c.Env, "CGO_ENABLED=1")
 	} else {