使用WaitGroup和channel控制并发测试，避免竞态条件。通过传入*sync.WaitGroup同步协程完成，用channel替代time.Sleep控制执行时机，结合select与超时确保测试可靠。
go语言的并发编程能力是其核心优势之一，但这也给单元测试带来了挑战。如何对包含
goroutine
、
channel
和
sync
机制的函数进行可靠的测试？关键在于控制并发行为、避免竞态条件，并确保测试可重复、不依赖时序。以下是几个实用技巧。
使用WaitGroup同步协程完成
当被测函数启动多个
goroutine
并期望它们全部完成时，使用
sync.WaitGroup
是最直接的方式。在测试中传入
*sync.WaitGroup
，或通过接口抽象等待逻辑，便于控制执行流程。
示例：
func ProcessTasks(tasks []string, wg *sync.WaitGroup) { for _, task := range tasks { wg.Add(1) go func(t string) { defer wg.Done() // 模拟处理 fmt.Println("Processed:", t) }(task) } }
测试中可以这样验证：
立即学习“go语言免费学习笔记（深入）”；
func TestProcessTasks(t *testing.T) { var wg sync.WaitGroup tasks := []string{"a", "b", "c"} <pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">ProcessTasks(tasks, &wg) wg.Wait() // 等待所有协程结束 // 此处可断言预期结果（如共享变量状态）
}
用Channel控制执行时机
对于依赖消息传递的并发函数，可通过注入
chan
来观察或控制行为。避免在测试中使用
time.Sleep
这类不可靠方式等待。
建议做法：
- 将输入/输出通道作为参数传入函数，测试时用缓冲通道替代
- 使用
select
配合
time.After
设置超时，防止测试永久阻塞
- 验证数据是否按预期发送到通道
示例：
func Monitor(stopCh <-chan struct{}, resultCh chan<- int) { count := 0 ticker := time.NewTicker(10 * time.Millisecond) defer ticker.Stop() <pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">for { select { case <-ticker.C: count++ case <-stopCh: resultCh <- count return } }
}
测试：
func TestMonitor_StopsGracefully(t *testing.T) { stopCh := make(chan struct{}) resultCh := make(chan int, 1) <pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">go Monitor(stopCh, resultCh) time.Sleep(50 * time.Millisecond) close(stopCh) select { case count := <-resultCh: if count == 0 { t.Fatal("expected non-zero count") } case <-time.After(100 * time.Millisecond): t.Fatal("timeout waiting for result") }
}
利用testify/mock模拟并发依赖
当并发函数依赖外部服务或复杂接口时，使用testify/mock创建可控的模拟对象。这样可以隔离并发逻辑，专注于测试目标函数的行为。
例如，一个并发调用API的worker：
type APIClient interface { Fetch(id string) (Data, error) }
在测试中mock该接口，并设定返回值与延迟，验证并发请求是否正确处理错误、超时或重试。
启用-race检测竞态条件
Go内置的竞态检测器是并发测试的重要工具。运行测试时加上
-race
标志：
go test -race ./...
它能自动发现大多数读写冲突。虽然会降低性能，但在CI中定期运行能有效捕捉潜在问题。
基本上就这些。关键是让并发逻辑可观察、可控制，避免依赖时间顺序，同时善用工具保障安全性。测试不是要复制真实并发环境，而是验证关键路径的正确性与鲁棒性。
© 版权声明
文章版权归作者所有，未经允许请勿转载。
THE END
暂无评论内容