测试命令行参数

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

因此，流入的流量将是：main

func main() {

  conf, output, err := parseArgs(os.Args[0], os.Args[1:])

  if err != nil {

    // ...

  }

  realMain(conf)

}

现在，您可以在单元测试中进行测试并完全单独进行测试。这也开辟了拥有多个配置源的可能性：例如，您也可以从env vars或配置文件或命令行中读取它。多个读取器可以填充一个结构，并且所有读取器都可以单独测试。parseArgsrealMainConfig