Golang 模拟所有参数，除了

Question

mock.on("FunctionName", "someStringArgument").Return(...)

假设如果 someStringArgument 是“hello”，那么我想返回“1”。但是，如果 someStringArgument 是任何其他字符串，我想返回“2”。

GoMock 如何实现这一点？

Answer 1

您要做的是编写一个自定义函数，它将返回您想要的输出。

这是我所做的一个简单示例。

定义自定义响应函数

func FunctionNameResponse(arg String) string{
    if arg == "hellp" {
        // I used quotes because you mentioned "1" and not 1
        return "1"
    }
    return "2"
}

在任何需要的地方调用自定义函数。

mock.on("FunctionName", mock.Anything).Return(FunctionNameResponse("someStringArgument"))