0

我有一个定义 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)
}
4

1 回答 1

0

了解匹配器的工作原理。这是我需要做的

m.EXPECT().DownloadFile("myfile", gomock.Any()).Return("/var/folders/ls/ok", nil)
于 2018-08-19T20:29:12.733 回答