Mocking 用于在 go 中进行测试的内部函数（或更深）

Question

有没有办法做到这一点？ 我想从另一个 function 调用中模拟 function 调用的响应。 例子

主要.go

type Crs struct {}
func (cr crs)TheFunction() error {
   // Some code
   _ := ToMockResponse(hello string)
   return nil
}

func ToMockResponse() error {
    return somerror
}

在我的测试文件中

main_test.go

func TestTheFunction(t *testing.T) {
    cr = Crs{}
    mockInstance = new(randomMock)
    mockInstance.On("ToMockResponse").Return(nil)

    err := cr.TheFunction()
    assert.NoError(t, err)
}

我不确定我做的是否正确。 我想要实现的是，只要在 TheFunction 中调用它，我只想对 ToMockResponse 进行模拟响应

我已经看过并阅读了大部分教程，但它们都显示了来自 class 方法的模拟响应。但是没有函数到函数 mocking 示例。 希望您能为此提供解决方案。

Answer 1

无需考虑您是否应该这样做，并假设您有充分的理由这样做，模拟函数的一种方法是通过间接级别使用它们。 然后，您可以出于测试目的在您的间接寻址中替换一个合适的替代方案。

接口是实现这一点的一种方法，但 GoLang 支持函数作为值，因此另一种方法是使用适当的 function 类型的简单变量：

适应你的例子，这可能看起来类似于：

var toMockResponse = ToMockResponse

func (cr crs)TheFunction() error {
   // Some code
   _ := toMockResponse(hello string)
   return nil
}

func ToMockResponse() error {
    return somerror
}

和你的测试：

func TestTheFunction(t *testing.T) {
    cr = Crs{}
    ofn := toMockResponse
    toMockResponse = func(string) error { return nil }
    defer func() { toMockResponse = ofn }()

    err := cr.TheFunction()
    assert.NoError(t, err)
}

我假设您需要导出ToMockResponse ； 故意不导出 function 变量，以便不能在 package 本身之外对其进行操作。

如果被重定向的 function本身没有导出，拒绝使用导出符号与未导出符号的大小写差异来区分它们的能力，那么您可能需要在命名时更有创意一些。 例如，一个名为toMockResponseFn的变量。

脚注

ofn是“原始函数”的助记符。 在更复杂的测试中，您可以通过为您需要的每个创建一个ofn来保留模式（ defer操作 w.r.t 包含 function，而不是范围）：

   {
      ofn := toMockResponse
      toMockResponse = func(string) error { return nil }
      defer func() { toMockResponse = ofn }()
   }
   {
      ofn := now
      now = func() time.Time { return time.Date(..etc..) }
      defer func() { now = ofn }()
   }