fixing testmain_test

Author: grantnelson-wf

build/build.go

@@ -687,6 +687,14 @@ func (p PackageData) FileModTime() time.Time {
 	return newest
 }
 
+// InternalBuildContext returns the build context that produced the package.
+//
+// WARNING: This function is a part of internal API and will be removed in
+// future.
+func (p *PackageData) InternalBuildContext() *build.Context {
+	return p.bctx
+}
+
 // TestPackage returns a variant of the package with "internal" tests.
 func (p *PackageData) TestPackage() *PackageData {
 	return &PackageData{

internal/testmain/testmain_test.go

@@ -21,7 +21,7 @@ func TestScan(t *testing.T) {
 
 	fset := token.NewFileSet()
 
-	got := TestMain{Package: pkg}
+	got := TestMain{Package: pkg.Package, Context: pkg.InternalBuildContext()}
 	if err := got.Scan(fset); err != nil {
 		t.Fatalf("Got: tm.Scan() returned error: %s. Want: no error.", err)
 	}
@@ -47,6 +47,7 @@ func TestScan(t *testing.T) {
 	}
 	opts := cmp.Options{
 		cmpopts.IgnoreFields(TestMain{}, "Package"), // Inputs.
+		cmpopts.IgnoreFields(TestMain{}, "Context"),
 	}
 	if diff := cmp.Diff(want, got, opts...); diff != "" {
 		t.Errorf("List of test function is different from expected (-want,+got):\n%s", diff)
@@ -66,7 +67,7 @@ func TestSynthesize(t *testing.T) {
 		{
 			descr: "all tests",
 			tm: TestMain{
-				Package: pkg,
+				Package: pkg.Package,
 				Tests: []TestFunc{
 					{Location: LocInPackage, Name: "TestXxx"},
 					{Location: LocExternal, Name: "TestYyy"},
@@ -88,14 +89,14 @@ func TestSynthesize(t *testing.T) {
 		}, {
 			descr: "testmain",
 			tm: TestMain{
-				Package:  pkg,
+				Package:  pkg.Package,
 				TestMain: &TestFunc{Location: LocInPackage, Name: "TestMain"},
 			},
 			wantSrc: testmain,
 		}, {
 			descr: "import only",
 			tm: TestMain{
-				Package: pkg,
+				Package: pkg.Package,
 				Examples: []ExampleFunc{
 					{Location: LocInPackage, Name: "ExampleXxx"},
 				},