如何测试Go语言中返回闭包的函数？求惯用法示例

嘿，针对你这段返回闭包的Go代码，我来分享几个Go社区里测试这类函数的常用思路，都是实践中比较惯用的方式：

1. 验证闭包对外部依赖的捕获与调用 #

闭包的核心特性之一是会捕获外部作用域的变量（比如你代码里的o.repo），所以测试的第一个关键点就是验证这个依赖是否被正确捕获，并且闭包是否按照预期调用了它的方法。

具体来说，你可以用mock对象来替代真实的OrderRepoInterface实现：

• 构造一个mock的repo，预设它的save方法的调用预期（比如调用次数、传入的参数、返回的错误）
• 创建EventHandler实例并传入这个mock repo
• 调用OnOrderWasCreated()拿到闭包
• 执行闭包，然后断言mock repo的save方法是否符合预设的调用预期，比如：
  • 是否被调用了一次
  • 传入的Message的version是否被闭包设置为1
  • 传入的payload是否和测试用例一致

2. 测试闭包的业务逻辑与错误传递 #

除了依赖调用，还要验证闭包自身的业务逻辑：

• 测试正常场景：当mock repo的save方法返回nil时，闭包是否也返回nil
• 测试错误场景：当mock repo返回特定错误（比如sql.ErrNoRows）时，闭包是否正确传递这个错误，没有额外修改或吞掉错误
• 验证闭包对输入参数的修改：比如你代码里把msg.version设为1，要测试这个修改是否确实生效，即使传入的原始Message版本不是1

3. 隔离不同状态的依赖进行测试 #

因为闭包的行为依赖于EventHandler的状态（也就是repo），所以可以构造不同状态的依赖来测试闭包的表现：

• 比如构造一个模拟“保存失败”的repo，测试闭包是否正确返回失败错误
• 构造一个模拟“保存成功”的repo，测试闭包是否正常完成流程
• 甚至可以测试当repo为nil时（虽然你的业务代码里可能不会出现，但测试边界情况也有意义），闭包执行时是否会触发预期的panic或错误

简单思路示例（非完整可运行代码） #

// 定义mock的OrderRepo
type MockOrderRepo struct {
    mock.Mock
}

func (m *MockOrderRepo) save(msg Message) error {
    args := m.Called(msg)
    return args.Error(0)
}

func TestOnOrderWasCreated_Closure(t *testing.T) {
    // 初始化mock repo和handler
    mockRepo := new(MockOrderRepo)
    handler := &EventHandler{repo: mockRepo}
    
    // 获取闭包
    vHandler := handler.OnOrderWasCreated()
    testMsg := &Message{payload: "test-order"}
    
    // 预设预期：save会被调用一次，参数是version=1的Message，返回nil
    mockRepo.On("save", Message{version: 1, payload: "test-order"}).Return(nil)
    
    // 执行闭包
    err := vHandler.function(testMsg)
    
    // 断言结果
    assert.NoError(t, err)
    mockRepo.AssertExpectations(t)
    assert.Equal(t, 1, testMsg.version) // 验证version被正确修改
}

总的来说，这类测试的核心就是用mock隔离真实依赖，聚焦验证闭包的行为是否符合预期——包括对外部依赖的调用、自身的参数修改、错误传递这几个核心点。

内容的提问来源于stack exchange，提问作者Edwin O.