fix: Read go.mod from correct location in testarchive (#3551)

Author: shreyas-goenka

integration/testarchive/downloader_test.go

@@ -69,7 +69,7 @@ func TestGoDownloader(t *testing.T) {
 	tmpDir := t.TempDir()
 
 	for _, arch := range []string{"arm64", "amd64"} {
-		err := testarchive.GoDownloader{Arch: arch, BinDir: tmpDir}.Download()
+		err := testarchive.GoDownloader{Arch: arch, BinDir: tmpDir, RepoRoot: "../.."}.Download()
 		require.NoError(t, err)
 
 		entries, err := os.ReadDir(filepath.Join(tmpDir, arch))

internal/testarchive/archive.go

@@ -106,7 +106,7 @@ func CreateArchive(archiveDir, binDir, repoRoot string) error {
 	// The Databricks platform explicitly does not provide any guarantees around
 	// the CPU architecture to keep the door open for future optimizations.
 	downloaders := []downloader{
-		GoDownloader{Arch: "amd64", BinDir: binDir},
+		GoDownloader{Arch: "amd64", BinDir: binDir, RepoRoot: repoRoot},
 		UvDownloader{Arch: "amd64", BinDir: binDir},
 		JqDownloader{Arch: "amd64", BinDir: binDir},
 

internal/testarchive/go.go

@@ -15,12 +15,13 @@ type downloader interface {
 }
 
 type GoDownloader struct {
-	BinDir string
-	Arch   string
+	BinDir   string
+	Arch     string
+	RepoRoot string
 }
 
 func (g GoDownloader) readGoVersionFromMod() (string, error) {
-	goModPath := filepath.Join("..", "..", "go.mod")
+	goModPath := filepath.Join(g.RepoRoot, "go.mod")
 
 	file, err := os.Open(goModPath)
 	if err != nil {