测试命令行参数

Question

你测试命令行参数吗？

我可以这样做：

func TestMainFunc(t *testing.T) {

    _, filename, _, _ := runtime.Caller(0)

    dir := filepath.Dir(filename)

    os.Args[1] = dir

    main()

}

但是，我在参数列表中覆盖了“test.v”：

os.Args[0]={string} "/private/var/folder/.../___appname.go"

os.Args[1]={string} "test.v"

os.Args[2]={string} "-test.panicontext0"

os.Args[3]={string} "^QTestMainFunc\E$"

我不知道这些参数在做什么，但我怀疑覆盖其中任何一个都不是好的 - 那么你怎么能测试参数呢？

Answer 1

为了扩展 Adrian 的评论，我将标志解析与配置完全分开。下面是一个示例：type Config struct {    verbose  bool    greeting string    level    int}func parseArgs(progname string, args []string) (config *Config, output string, err error) {    flags := flag.NewFlagSet(progname, flag.ContinueOnError)    var buf bytes.Buffer    flags.SetOutput(&buf)    var conf Config    flags.BoolVar(&conf.verbose, "verbose", false, "set verbosity")    flags.StringVar(&conf.greeting, "greeting", "", "set greeting")    flags.IntVar(&conf.level, "level", 0, "set level")    err = flags.Parse(args)    if err != nil {        return nil, buf.String(), err    }    return &conf, "", nil}该函数从命令行参数分析程序的配置，并且可以轻松独立于程序的其余部分进行测试。parseArgs因此，流入的流量将是：mainfunc main() {  conf, output, err := parseArgs(os.Args[0], os.Args[1:])  if err != nil {    // ...  }  realMain(conf)}现在，您可以在单元测试中进行测试并完全单独进行测试。这也开辟了拥有多个配置源的可能性：例如，您也可以从env vars或配置文件或命令行中读取它。多个读取器可以填充一个结构，并且所有读取器都可以单独测试。parseArgsrealMainConfig