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

Question

我正在尝试为这样的文件编写单元测试

//file1.go
var (
    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

使用Config/Settings结构。在生产中，最好使用 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()
}

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

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

Using package level variables in go test files