我有一个定义Downloadfile函数的包装器接口。问题是其中一个参数具有随机的临时 directoriename,而输出是随机的临时文件名。如何使用 https://github.com/golang/mock 模拟它,以便我可以测试 addFile 函数
var wrapper
func Test(t *testing.T) {
ctrl, _ := gomock.WithContext(context.Background(), t)
defer ctrl.Finish()
m := mock.NewMockWrapperInterface(ctrl)
m.EXPECT().DownloadFile("myfile", "/var/folders/ls/???").Return("/var/folders/ls/???", nil)
wrapper = m
path, err := addFile(myfile, key)
...
}
func addFile(myfile, key) (string, error) {
tmpDirectory := helper.CreateTemporaryDirectory(objectKey)
defer helper.CleanUpDirectory(tmpDirectory)
return wrapper.DownloadFile(myfile, tmpDirectory)
}
了解匹配器的工作原理。这就是我需要做的
m.EXPECT().DownloadFile("myfile", gomock.Any()).Return("/var/folders/ls/ok", nil)