inline go script and run go mod commands in container

Author: RafaelCenzano

internal/test/compilecheck/compile_check_test.go

@@ -7,6 +7,7 @@
 package main
 
 import (
+	"bytes"
 	"context"
 	"fmt"
 	"io"
@@ -21,6 +22,24 @@ import (
 	"github.com/testcontainers/testcontainers-go"
 )
 
+const mainGo = `package main
+
+import (
+	"fmt"
+
+	"go.mongodb.org/mongo-driver/v2/bson"
+	"go.mongodb.org/mongo-driver/v2/mongo"
+	"go.mongodb.org/mongo-driver/v2/mongo/options"
+)
+
+func main() {
+	_, _ = mongo.Connect(options.Client())
+	fmt.Println(bson.D{{Key: "key", Value: "value"}})
+}
+`
+
+var architectures = []string{"386", "arm", "arm64", "ppc64le", "s390x"}
+
 func getVersions(t *testing.T) []string {
 	t.Helper()
 
@@ -49,16 +68,15 @@ func TestCompileCheck(t *testing.T) {
 			req := testcontainers.ContainerRequest{
 				Image:      image,
 				Cmd:        []string{"tail", "-f", "/dev/null"},
-				WorkingDir: "/workspace",
+				WorkingDir: "/app",
 				HostConfigModifier: func(hostConfig *container.HostConfig) {
-					hostConfig.Binds = []string{fmt.Sprintf("%s:/workspace", rootDir)}
+					hostConfig.Binds = []string{fmt.Sprintf("%s:/driver", rootDir)}
 				},
-				Env: map[string]string{
-					"GC": "go",
-					// Compilation modules are not part of the workspace as testcontainers requires
-					// a version of klauspost/compress not supported by the Go Driver / other modules
-					// in the workspace.
-					"GOWORK": "off",
+				Files: []testcontainers.ContainerFile{
+					{
+						ContainerFilePath: "/app/main.go",
+						Reader:            bytes.NewReader([]byte(mainGo)),
+					},
 				},
 			}
 
@@ -75,14 +93,47 @@ func TestCompileCheck(t *testing.T) {
 				require.NoError(t, err)
 			}()
 
-			exitCode, outputReader, err := container.Exec(context.Background(), []string{"bash", "etc/compile_check.sh"})
+			// Initialize go module and set up replace directive to use local driver.
+			setupCmds := [][]string{
+				{"go", "mod", "init", "app"},
+				{"go", "mod", "edit", "-replace", "go.mongodb.org/mongo-driver/v2=/driver"},
+				{"go", "mod", "tidy"},
+			}
+
+			for _, cmd := range setupCmds {
+				exitCode, outputReader, err := container.Exec(context.Background(), cmd)
+				require.NoError(t, err)
+
+				output, err := io.ReadAll(outputReader)
+				require.NoError(t, err)
+
+				require.Equal(t, 0, exitCode, "command %v failed: %s", cmd, output)
+			}
+
+			// Standard build.
+			exitCode, outputReader, err := container.Exec(context.Background(), []string{"go", "build", "./..."})
 			require.NoError(t, err)
 
 			output, err := io.ReadAll(outputReader)
 			require.NoError(t, err)
 
-			t.Logf("output: %s", output)
-			assert.Equal(t, 0, exitCode)
+			require.Equal(t, 0, exitCode, "standard build failed: %s", output)
+
+			// Check build with various architectures.
+			for _, arch := range architectures {
+				exitCode, outputReader, err := container.Exec(
+					context.Background(),
+					[]string{"sh", "-c", fmt.Sprintf("GOOS=linux GOARCH=%s go build ./...", arch)},
+				)
+				require.NoError(t, err)
+
+				output, err := io.ReadAll(outputReader)
+				require.NoError(t, err)
+
+				assert.Equal(t, 0, exitCode, "build for GOARCH=%s failed: %s", arch, output)
+			}
+
+			t.Logf("compilation checks passed for %s", image)
 		})
 	}
 }