internal/goenv_testmode.go

// Copyright 2022 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.

//go:build testmode
// +build testmode

package internal

import (
	"encoding/json"
	"os"
	"os/exec"
)

// GoEnv returns the value for key in `go env`.
//
// For debugging and testing purposes, the value of
// undocumented environment variable TEST_GOVERSION
// is used for go env GOVERSION.
func GoEnv(key string) (string, error) {
	out, err := exec.Command("go", "env", "-json", key).Output()
	if err != nil {
		return "", err
	}
	env := make(map[string]string)
	if err := json.Unmarshal(out, &env); err != nil {
		return "", err
	}

	if v := os.Getenv("TEST_GOVERSION"); v != "" {
		// Unlikely to happen in practice, mostly used for testing.
		env["GOVERSION"] = v
	}

	return env[key], nil
}