在 go test 中处理命令行参数

Question

有没有办法在 go "tests" 中获取命令行参数，
当您调用go test 时，显然您的main 没有运行，所以有没有办法处理命令行参数，

一种方法是使用 flags 包并检查每个测试或正在测试的函数中的命令行参数，但这并不理想，因为您需要在很多地方执行此操作，不像当您运行应用程序时，您只需在main 中即可。

有人可能会认为这样做是错误的，并且违反了单元测试的纯度：

1. 并非所有测试都是单元测试
2. 不依赖“ENV”变量并在命令行中实际将这些东西作为参数传递是非常实用的，

---

为了记录，我最终将init() 函数放在我的一个_test 文件中，并设置在以这种方式调用main 时通过标志设置的变量。

Answer 1

根据我的经验，环境配置最好保存在环境变量中。您可以像这样依赖全局变量：

var envSetting = os.Getenv("TEST_ENV")

或者，如果需要使用标志，您可以将初始化代码放在名为 init() 的函数中。

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

Answer 2

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

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

Answer 3

另一种方法是让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)
    ...
}

Answer 4

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

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

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

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.
}