使用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.Mutexcounter := 0var wg sync.WaitGroupfor 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 能有效防御挂起问题。
基本上就这些。只要保证协程能被正确等待、避免竞态、设置合理超时,多协程函数的测试就能稳定可靠。不复杂但容易忽略细节。
以上就是如何用Golang测试多协程函数_Golang 多协程函数测试实践的详细内容,更多请关注创想鸟其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1425960.html
微信扫一扫
支付宝扫一扫