在 go 测试文件中使用包级变量

Question

我正在尝试为这样的文件编写单元测试//file1.govar (    fileName = "/path/to/original/file.json")func DoSomething(){    // Read <fileName> and do some stuff}但是对于单元测试，我想使用 /path/to/dummy/file.json 供 DoSomething() 使用 file1_test.go 将与 file.go 在同一个包下编写测试用例的正确方法是什么这个功能？将 fileName 变量修改为 /path/to/dummy/file.json 是个好主意吗？还是我们应该修改 file1.go 以使其适合测试？这是使这些值可配置的标准吗，以便它们可以在单元测试期间修改，例如在这种情况下的文件名，或者我们应该修改我们的函数，以便它可以直接处理字节流数据，以便在测试测试期间我们可以提供虚拟数据直接地？

Answer 1

使用配置/设置结构。在生产中，最好使用 YAML 来保存配置值

type Config struct{FileName string}

在 main.go

var conf Config

func main(){

   conf = LoadConfigFromYaml()

   DoSomething()

}

func DoSomething(){

// Read <fileName> from conf.FileName

}

在 main_test.go

func TestSomething(t *testing.T){

   conf = Config{FileName: "/path/to/dummy/file.json"}

   DoSomething()

}

它还允许您为每个测试设置不同的配置。

更好的方法是创建一个在初始化时接受 Config 并从 main 和 test 注入配置并避免使用main package变量的库。