在我要测试的包中,我有一个 init 函数,它加载配置文件,其中包含一些我想用来运行我的应用程序的东西。但是,我不想在运行单元测试时触发此 init 函数。
有什么方法可以跳过或阻止在单元测试期间调用此 init 函数吗?
一些片段来说明这个问题:
func init() {
var err error // Necessary to prevent config variable shadowing
config, err = loadConfig("./client/config.yml")
if err != nil {
log.Fatal(err)
}
}
func loadConfig(filepath string) (*Config, error) {
viper.SetConfigFile(filepath)
if err := viper.ReadInConfig(); err != nil {
return nil, fmt.Errorf("Error loading config file: %s", err)
}
(...)
}
// New returns a Config value(!)
func New() Config {
return *config
}
一个测试用例:
func TestNew(t *testing.T) {
expected := &Config{}
observed := New()
if !reflect.DeepEqual(observed, expected) {
t.Errorf("observed %+v. expecting %+v\n", observed, expected)
}
}
最佳答案
我不确定是否有更好的方法来做到这一点,但如果你考虑到包级变量在 init
函数运行之前初始化的事实,你可以使用一个标志来告诉你是否正在运行测试。
var _testing = false
func init() {
if _testing {
return
}
var err error // Necessary to prevent config variable shadowing
config, err = loadConfig("./client/config.yml")
if err != nil {
log.Fatal(err)
}
}
// ...
在你的测试文件中你可以做这样的事情:
// not nice but works
var _ = (func() interface{} {
_testing = true
return nil
}())
func TestNew(t *testing.T) {
expected := &Config{}
observed := New()
if !reflect.DeepEqual(observed, expected) {
t.Errorf("observed %+v. expecting %+v\n", observed, expected)
}
}
您可以在此处阅读有关初始化顺序的更多信息:https://golang.org/ref/spec#Program_initialization_and_execution
关于unit-testing - 初始化函数中断单元测试,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/43409919/