在go测试中处理命令行参数

Question

有没有办法在"测试"中获取命令行参数,
当你go test显然调用时你main没有运行,那么有没有办法处理命令行参数,

一种方法是使用flags包并检查每个被测试的测试或函数中的命令行参数,但这对于你需要在很多很多地方执行此操作并不理想,这与你在main当您运行该应用程序时.

编辑:主要是向下选民,我想你是在贬低这个问题,因为你认为这是一个错误的事情,并且反对单位测试的纯度,

1. 并非所有测试都是单元测试
2. 它非常实用,不依赖于"ENV"变量,实际上将这些东西作为参数传递给命令行,
3. 请将您的意见放在答案中,以便后代能够受益,

编辑2:为了记录,我最终init()在我的一个_test文件中放置了一个函数,并设置了通过这种方式调用main时通过flags设置的变量.

Answer 1

根据我的经验,环境变量最好保存在环境变量中.您可以依赖全局变量,如下所示:

var envSetting = os.Getenv("TEST_ENV")

或者,如果需要使用flags,则可以将初始化代码放在名为init()的函数中.

func init() {
    flags.Parse()
    myEnv = *envFlag
    // ...
}

Answer 2

您可以直接测试 main 函数并传递参数。

显示标志和一对位置参数的简单示例

注意：不要将其称为“TestMain”，因为它对 Go 1.8 中的测试框架具有特殊含义。

package main

import (
    "os"
    "testing"
)

func TestMainFunc(t *testing.T) {

    os.Args = append(os.Args, "--addr=http://b.com:566/something.avsc")
    os.Args = append(os.Args, "Get")
    os.Args = append(os.Args, `./some/resource/fred`)

    main()

    // Test results here, and decide pass/fail.
}

Answer 3

os.Args[1] = "-conf=my.conf"
flag.Parse()

请注意，配置文件名是硬编码的。

Answer 4

另一种方法是使它main()成为存根，仅在参数由处理后才调用另一个函数flag.Parse()，例如：

var flagvar int
func init() {
    flag.IntVar(&flagvar, "flagname", 1234, "help for flagname")
}

func main() {
    flag.Parse()
    submain(flag.Args)
}

func submain(args []string) {
   ...
}

然后在测试中，可以在调用submain(...)模拟标志和参数的命令行建立之前设置标志变量并建立参数。此方法可用于最大化测试覆盖范围，而无需实际使用命令行。例如，在main_test.go中，您可以编写：

func TestSomething(t *testing.T) {
    flagvar = 23
    args := []string{"a", "b", "c"}
    submain(args)
    ...
}