测试多线程函数需解决执行顺序不可控、资源竞争和测试提前结束问题，应使用sync.WaitGroup或channel确保所有goroutine完成。
测试多线程函数在 Golang 中是一个常见但容易出错的任务。由于并发执行的不确定性，直接用常规方式测试可能会导致结果不稳定或出现竞态条件。要正确测试多线程函数，关键在于控制并发行为、合理使用同步机制，并借助 Go 自带的竞态检测工具。
理解并发测试的核心挑战
多线程函数通常涉及 goroutine、channel、sync 包等并发原语。测试这类函数时，主要面临以下几个问题：
- 执行顺序不可控：goroutine 的调度由 runtime 决定，无法保证执行顺序。
- 资源竞争：多个 goroutine 同时访问共享变量可能导致数据不一致。
- 测试提前结束：主 goroutine（测试函数）可能在子 goroutine 完成前就退出，导致漏测。
因此，测试必须确保所有并发操作完成后再进行断言。
使用 sync.WaitGroup 等待 goroutine 结束
最常见的做法是使用 sync.WaitGroup 来协调多个 goroutine 的完成状态。它能有效防止测试函数过早返回。
立即学习“go语言免费学习笔记（深入）”；
示例：测试一个并行处理任务的函数
func ProcessTasks(tasks []int, workerCount int) int {
var sum int64
var mu sync.Mutex
var wg sync.WaitGroup
for i := 0; i < workerCount; i++ {
wg.Add(1)
go func() {
defer wg.Done()
for _, task := range tasks {
mu.Lock()
sum += int64(task)
mu.Unlock()
}
}()
}
wg.Wait()
return int(sum)}
func TestProcessTasks(t testing.T) {
tasks := []int{1, 2, 3}
result := ProcessTasks(tasks, 3)
expected := 18 // 每个任务被3个worker各处理一次：(1+2+3)3 = 18
if result != expected {
t.Errorf("期望 %d，实际 %d", expected, result)
}
}
这里通过 WaitGroup 确保所有 worker 执行完毕后才返回结果，测试可以稳定获取最终值。
利用 channel 进行通信和同步
另一种方式是使用 channel 接收各个 goroutine 的结果，避免共享变量带来的复杂性。
示例：用 channel 收集结果
func SumInParallel(nums []int) int {
ch := make(chan int, len(nums))
for _, n := range nums {
go func(val int) {
ch <- val * 2
}(n)
}
total := 0
for i := 0; i < len(nums); i++ {
total += <-ch
}
return total}
func TestSumInParallel(t testing.T) {
nums := []int{1, 2, 3}
got := SumInParallel(nums)
want := 12 // (12 + 22 + 32) = 12
if got != want {
t.Errorf("期望 %d，实际 %d", want, got)
}
}
这种方式天然避免了锁的使用，逻辑更清晰，也更容易测试。
启用竞态检测（Race Detector）
Go 提供了强大的竞态检测工具。在测试时加上 -race 标志，可以自动发现数据竞争问题。
运行命令：
go test -race -v ./...如果存在未加锁的共享变量读写，会直接报出 race condition 错误。这是保障并发安全的重要手段。
基本上就这些。只要确保 goroutine 能被正确等待、共享资源有适当保护，并开启竞态检测，Golang 中的多线程函数测试就能做到可靠且可维护。
