在 Go 测试期间处理 Viper 配置文件路径

问题描述

所以我有一个非常基本的配置，Viper 从我的基本目录中读取 .env 文件。如果没有 .env 文件，我会致命地终止进程。正常运行我的应用程序时一切顺利。当我使用 go test -v ./.. 运行测试时，测试框架似乎会进入每个文件的目录，并且每次都调用我的 config init() 函数，因此 viper.AddConfigPath(".") 指向错误的位置。>

这是我的目录结构：

/
  /restapi
    items.go
    items_test.go
  /util
    env.go
  main.go
  .env

env.go

package util

imports...

// global variables available via util package
var (
  Port     int
  DbURI    string
)

func init() {
  viper.SetDefault(PORT,8080)
  viper.SetConfigFile(".env")
  viper.AddConfigPath(".")
  viper.AutomaticEnv()

  fmt.Println("---------to see in test printout")
  cwd,_ := os.Getwd()
  fmt.Println(cwd)
  fmt.Println("---------")

  if err := viper.ReadInConfig(); err != nil {
    log.Fatal("no environment file!")
  }

  Port = viper.GetInt("PORT")
  DbURI = viper.GetString("DB_URI")
}

每个包基本上都依赖于我的 util 包，因此这个 init 函数为每个测试运行。有没有办法让 viper 总是从基本目录中提取 .env 文件，即使有测试正在运行？我尝试了几个不同的 AddConfigPath() 调用。对 Go 有点陌生。或者这种环境变量的结构设置是否因为每次都失败了我的测试而不起作用？

解决方法

问题是您提供给 viper.AddConfigPath(".") 方法的路径，但是您的 env 文件相对路径不在基于您共享的文件夹结构树的测试文件上，它必须是这样的：{{1} }.

显然 viper.SetConfigFile() 调用不尊重 viper.AddConfigPath() 调用...我将其修改为使用 viper.SetConfigName(".env") 并且它实际上会接听对 AddConfigPath 的调用，所以我可以为当前目录和父目录添加配置路径。

go viper-go