unit-testing - 在 go test 文件中使用包级变量

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 以使其适合测试？

这是使这些值可配置的标准吗，以便它们可以在单元测试期间修改，例如在这种情况下的文件名，或者我们应该修改我们的函数，以便它可以直接处理字节流数据，以便在测试测试期间我们可以提供虚拟数据直接地？

score 1 · Accepted Answer

使用配置/设置结构。在生产中，最好使用 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变量。

1 回答 1