如何测试函数是否在goroutine中被调用并验证参数正确性？

你说得完全对！问题根源就是goroutine的异步执行特性——测试函数MyTest在调用service.MyFunc()后会快速结束，而goroutine里的MyMockFunc(1)还没来得及触发mock的断言，测试就已经退出了，所以断言失败。

下面给你几个不需要修改MyMockFunc参数的解决方案：

方案1：利用Mock库的等待功能（推荐） #

如果你用的是testify/mock这类常用的Go Mock库，它自带了WaitForExpectations方法，可以让测试等待所有mock断言被触发，完全不需要改动原函数：

import (
    "testing"
    "time"
    "github.com/stretchr/testify/mock"
)

func MyTest(t *testing.T) {
    // 设置mock期望：调用MyMockFunc并传入参数1
    service.EXPECT().MyMockFunc(1)
    
    // 触发业务逻辑，里面会启动goroutine调用MyMockFunc
    service.MyFunc()
    
    // 等待最多1秒，直到所有mock期望都被满足
    if err := service.Mock.WaitForExpectations(t, time.Second); err != nil {
        t.Fatalf("mock断言未满足：%v", err)
    }
}

这个方法最简洁，WaitForExpectations会自动等待异步调用的mock函数执行完成，超时后如果断言还没触发就会报错。

方案2：用sync.WaitGroup配合Mock的回调 #

如果不想依赖第三方库的高级功能，也可以用sync.WaitGroup，通过Mock的回调函数标记任务完成，不需要修改MyMockFunc的参数：

import (
    "testing"
    "sync"
)

func MyTest(t *testing.T) {
    var wg sync.WaitGroup
    wg.Add(1) // 因为要等待1次goroutine调用
    
    // 设置mock期望，同时添加Do回调：当MyMockFunc被调用时，标记WaitGroup完成
    service.EXPECT().MyMockFunc(1).Do(func(_ int) {
        wg.Done()
    })
    
    service.MyFunc()
    wg.Wait() // 阻塞直到goroutine里的mock函数执行完毕
}

这里的Do回调是Mock库提供的钩子，会在MyMockFunc被调用时执行，我们用它来通知WaitGroup任务完成，测试就会等goroutine执行完再结束。

方案3：手动轮询检查Mock状态 #

如果连Mock库都不想用，自己实现Mock结构体的话，可以通过加锁和轮询的方式等待异步调用完成：

import (
    "testing"
    "sync"
    "time"
)

// 自定义Mock结构体
type MockService struct {
    mu    sync.Mutex
    called bool
    recvArg int
}

// 原签名的MyMockFunc，不需要改参数
func (m *MockService) MyMockFunc(arg int) {
    m.mu.Lock()
    defer m.mu.Unlock()
    m.called = true
    m.recvArg = arg
}

func MyTest(t *testing.T) {
    mockSvc := &MockService{}
    // 替换全局service为自定义mock
    service = mockSvc
    
    service.MyFunc()
    
    // 轮询检查，最多等待1秒
    deadline := time.Now().Add(time.Second)
    for time.Now().Before(deadline) {
        mockSvc.mu.Lock()
        called := mockSvc.called
        arg := mockSvc.recvArg
        mockSvc.mu.Unlock()
        
        if called && arg == 1 {
            return // 断言通过，测试结束
        }
        time.Sleep(10 * time.Millisecond) // 短时间休眠，避免占用过多CPU
    }
    
    t.Fatal("MyMockFunc未在规定时间内被调用，或参数不正确")
}

这个方法完全不依赖第三方库，但代码相对繁琐，适合轻量级场景。

所有方案都不需要修改MyMockFunc的参数，完美符合你的需求。

内容的提问来源于stack exchange，提问作者yndolok