Golang并发函数单元测试实践技巧

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)
    }
}

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() // 等待所有协程结束

// 此处可断言预期结果（如共享变量状态）

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")
}

type APIClient interface {
    Fetch(id string) (Data, error)
}

go test -race ./...