unit-testing - 初始化函数中断单元测试

Question

在我要测试的包中，我有一个 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)
    }
}

Answer 1

我不确定是否有更好的方法来做到这一点，但如果你考虑到包级变量在 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