Golang 模拟所有参数，除了答案

【问题标题】：Golang mock for all arguments exceptGolang 模拟所有参数，除了
【发布时间】：2020-01-21 20:21:28
【问题描述】：

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

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

GoMock 是如何实现的？

【问题讨论】：

您能否提供您正在使用的模拟库的链接
@Jesse, godoc.org/github.com/golang/mock/gomock
godoc.org/github.com/golang/mock/gomock#Call.DoAndReturn。话虽如此，如果您需要一些非常具体的事情发生，您总是可以自己编写一个模拟实现。没有什么会强迫您始终使用生成的模拟。
你试过什么？包括您的代码。你遇到了什么问题？
这很简单：不要嘲笑。

标签： go

【解决方案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"))

【讨论】：