main_test.go

package main

import (
	"encoding/json"
	"os"
	"path/filepath"
	"testing"

	"github.com/rogpeppe/go-internal/gotooltest"
	"github.com/rogpeppe/go-internal/testscript"
	"github.com/stretchr/testify/require"
	exec "golang.org/x/sys/execabs"
)

func TestMain(m *testing.M) {
	os.Exit(testscript.RunMain(m, map[string]func() int{
		"allowtags": func() int {
			main()

			// Won't reach here because main will exit early.
			return 0
		},
	}))
}

func TestScripts(t *testing.T) {
	t.Parallel()

	var goEnv struct {
		GOCACHE    string
		GOMODCACHE string
		GOMOD      string
	}

	out, err := exec.Command("go", "env", "-json").CombinedOutput()
	require.NoError(t, err)

	require.NoError(t, json.Unmarshal(out, &goEnv))

	p := testscript.Params{
		Dir: filepath.Join("testdata", "script"),
		Setup: func(env *testscript.Env) error {
			env.Setenv("GOCACHE", goEnv.GOCACHE)
			env.Setenv("GOMODCACHE", goEnv.GOMODCACHE)
			env.Setenv("GOMOD_DIR", filepath.Dir(goEnv.GOMOD))

			return nil
		},
	}

	require.NoError(t, gotooltest.Setup(&p))

	testscript.Run(t, p)
}