初始化函数破坏单元测试

Question

在我要测试的包中，我有一个 init 函数，它加载包含一些我想用来运行我的应用程序的东西的配置文件。但是，我不想在运行单元测试时触发这个初始化函数。

有什么方法可以跳过或阻止在单元测试期间调用这个 init 函数？

一些sn-ps来说明问题：

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 func 之前已初始化的事实，您可以使用一个标志来告诉您是否是是否运行测试。

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