我们在go的模块化开发的时候,运行模块里面的单元测试时,当前整个模块的 init 函数都会被执行,在某些情况下我们运行独立的单元测试并不需要执行其他文件里面的init函数,这时就需要判断当前是否是在运行单元测试,判断方式很简单,只需要判断当前运行参数是否是以-test开始即可,if len(os.Args) > 1 && strings.HasPrefix(os.Args[1], "-test") { } 。
完整判断代码如下:
Go
func init() {
// 如果是运行的单元测试,则跳过viper加载配置文件 避免找不到配置文件的异常
// 单元测试 的情况下不加载viper配置
if len(os.Args) > 1 && strings.HasPrefix(os.Args[1], "-test") {
return // 当前运行当前测试,直接返回
}
}