route/vendor/github.com/magefile/mage/sh/testmain_test.go

47 lines
748 B
Go
Raw Normal View History

2018-01-17 04:32:31 +00:00
package sh
import (
"flag"
"fmt"
"os"
"testing"
)
var (
helperCmd bool
printArgs bool
stderr string
stdout string
exitCode int
printVar string
)
func init() {
flag.BoolVar(&helperCmd, "helper", false, "")
flag.BoolVar(&printArgs, "printArgs", false, "")
flag.StringVar(&stderr, "stderr", "", "")
flag.StringVar(&stdout, "stdout", "", "")
flag.IntVar(&exitCode, "exit", 0, "")
flag.StringVar(&printVar, "printVar", "", "")
}
func TestMain(m *testing.M) {
flag.Parse()
if printArgs {
fmt.Println(flag.Args())
return
}
if printVar != "" {
fmt.Println(os.Getenv(printVar))
return
}
if helperCmd {
fmt.Fprintln(os.Stderr, stderr)
fmt.Fprintln(os.Stdout, stdout)
os.Exit(exitCode)
}
os.Exit(m.Run())
}