testing - 如何在运行时获取测试环境

Question

我想检查代码是否正在运行go test，以便我可以进行一些配置。

有什么功能可以做到吗？喜欢:

运行时.IsBeingTested()

Answer 1

只需指定您在测试的 init 中运行测试。例如，在 pkg.go 中:

package pkg

var isTesting = false

// ...

在 pkg_test.go 中:

package pkg

func init() {
    isTesting = true
}

// ...

此技术不仅可以用于 bool，还可以用于任何数据或函数。如果你的包中有一些变量(在你的例子中，一个配置变量)，你可以在 init 中覆盖它。