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

Question

所以我有一个非常基本的配置，Viper从我的基目录中读取一个.env文件。如果没有.env文件，我将致命地杀死该进程。正常运行我的应用程序时，一切都很顺利。当我使用 运行测试时，测试框架似乎单步执行每个文件的目录，并且每次都调用我的config init（）函数，因此指向错误的位置。go test -v ./..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（）调用。有点新去。还是环境变量的这种结构设置不起作用，因为它每次都无法通过我的测试？

Answer 1

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

Answer 2

所以显然这个电话不尊重这个电话...我将其修改为使用，它实际上会拾取对AddConfigPath的调用，因此我可以为当前目录和父目录添加配置路径。viper.SetConfigFile()viper.AddConfigPath()viper.SetConfigName(".env")