使用sync.WaitGroup确保协程完成,t.Parallel()并行测试,-race检测竞态,context超时控制,保证多协程测试稳定。

测试多协程函数在 Golang 中是一个常见但容易出错的场景。由于并发执行的不确定性,直接测试可能会导致竞态、超时或结果不可靠。关键在于控制协程生命周期、合理使用同步机制,并借助 Go 的测试工具确保稳定性。
使用 sync.WaitGroup 等待协程完成
在测试中启动多个协程时,必须确保所有协程执行完毕后再进行断言,否则测试可能在协程运行前就结束了。
注意:不能依赖 time.Sleep 做等待,这不可靠且不精确。推荐使用 sync.WaitGroup 来协调协程结束:
- 在主 goroutine 中创建 WaitGroup 并传入每个协程
- 每个协程执行完任务后调用
Done() - 主协程调用
Wait()阻塞直到所有任务完成
示例代码:
立即学习“go语言免费学习笔记(深入)”;
func TestConcurrentFunction(t *testing.T) {
var wg sync.WaitGroup
results := make([]int, 10)
for i := 0; i < 10; i++ {
wg.Add(1)
go func(i int) {
defer wg.Done()
results[i] = doWork(i) // 模拟耗时操作
}(i)
}
wg.Wait() // 确保所有协程完成
for i, r := range results {
if r != expectedValue(i) {
t.Errorf("work at %d got %d, want %d", i, r, expectedValue(i))
}
}}
利用 t.Parallel() 并行运行测试用例
当多个测试用例彼此独立时,可以标记为并行执行,提升整体测试速度。
调用 t.Parallel() 可将当前测试放入并行队列:
- 多个测试用例可同时运行,由 Go 运行时调度
- 适合用于压力测试或多组并发逻辑验证
- 避免共享可变状态,防止测试间干扰
示例:
func TestConcurrentAdd(t *testing.T) {
t.Parallel()
var mu sync.Mutex
counter := 0
var wg sync.WaitGroup
for i := 0; i < 100; i++ {
wg.Add(1)
go func() {
defer wg.Done()
mu.Lock()
counter++
mu.Unlock()
}()
}
wg.Wait()
if counter != 100 {
t.Fatalf("expected 100, got %d", counter)
}}
检测数据竞争(Race Condition)
Go 自带竞态检测工具,可在测试时启用 -race 标志来发现潜在问题。
运行命令:
go test -race -v ./...
该功能会监控内存访问,一旦发现多个 goroutine 同时读写同一变量且无同步保护,就会报错。
- 即使测试通过,也应定期使用
-race检查 - 特别关注未加锁的全局变量、共享切片或 map
- 竞态检测会显著降低性能,仅用于开发和 CI 阶段
设置超时防止测试卡死
协程可能因死锁、阻塞 channel 或无限循环导致测试永不结束。
使用 t.Timeout() 设置最大执行时间:
func TestConcurrentWithTimeout(t *testing.T) {
ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
defer cancel()
done := make(chan bool)
go func() {
slowConcurrentProcess() // 可能卡住的操作
done <- true
}()
select {
case <-done:
// 正常完成
case <-ctx.Done():
t.Fatal("test timed out")
}}
结合 context 和 select 能有效防御挂起问题。
基本上就这些。只要保证协程能被正确等待、避免竞态、设置合理超时,多协程函数的测试就能稳定可靠。不复杂但容易忽略细节。










