Голанг издевается над всеми аргументами, кроме

Question

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

Предположим, если someStringArgument "привет", то я хочу вернуть "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"))