go - 作证模拟函数在函数内部返回

Question

我想模拟一个函数的响应。但是这个函数位于或在另一个函数内部调用。假设我有这个功能

// main.go
func TheFunction() int {
   // Some code
   val := ToMockResponse()
   return val
}

func ToMockResponse() int {
    return 123
}

现在在我的测试文件上

// main_test.go
func TestTheFunction(t *testing.T) {
    mockInstance = new(randomMock)
    mockInstance.On("ToMockResponse").Return(456)

    returned := TheFunction()
    assert.Equal(t, 456, returned)
}

如您所见，在函数 TheFunction() 中调用了函数 ToMockResponse。现在我想测试 TheFunction 但我想模拟 ToMockResponse 的响应我怎样才能实现这一点？

Answer 1

您应该考虑将第二个函数作为参数传递给第一个函数。

您有几个选择可以做到这一点。您可以简单地将其作为参数传递。

func DoX(doY func()) {
   doY()
}

这很简单，但随着核心变得越来越复杂，效果并不好。更好的选择通常是在结构上创建函数方法。

type X struct {
   Y Y
}

type Y interface {
   Do()
}

func (x *X) Do() {
   x.Y.Do()
}

然后 Y 可以用 mock 代替。