我有一个定义 Downloadfile 函数的包装器接口。问题是其中一个参数有一个随机的临时目录名,而输出是一个随机的临时文件名。如何使用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)
}