如何在 Golang 單元測試中處理並發？-Golang-PHP中文網

如何在 Golang 單元測試中處理並發？

WBOY

發布： 2024-06-02 18:14:01

原創

455 人瀏覽過

在 Go 單元測試中優雅地處理並發性需要以下步驟：使用 goroutines 啟用並發執行。使用 channels 在 goroutines 之間進行通訊。使用 sync.WaitGroup 同步 goroutines，確保在斷言結果之前所有 goroutines 完成。考慮競爭條件、執行順序和隔離性，以確保測試的健全性。

如何在 Golang 单元测试中处理并发？

如何在Go 單元測試中優雅地處理並發：基於實戰的指南

在並發環境中編寫可靠的單元測試對於建立健壯的應用程式至關重要。然而，在 Go 中處理並發測試可能具有挑戰性。本文將引導你逐步了解如何優雅地在 Go 單元測試中處理並發，並透過一個實戰案例進行說明。

取得並發性

goroutines：並發性的函數，並行執行。
channels：用於在 goroutines 之間通訊。
sync.WaitGroup：用於等待一組 goroutines 完成。

同步測試

當有多個 goroutines 在運行時，重要的是要確保測試的執行順序和資料一致性。

goroutine 等待：使用 sync.WaitGroup 來等待所有 goroutines 完成後再斷言結果。
channel 通訊：使用 channels 在 goroutines 之間傳遞資料和控制執行流程。

實戰案例

考慮一個簡單的函數SumInts，它傳回一組整數的總和：

func SumInts(nums []int) int {
    sum := 0
    for _, num := range nums {
        sum += num
    }
    return sum
}

登入後複製

我們可以使用以下單元測試來測試此函數的並發行為：

import (
    "sync"
    "testing"
)

func TestSumIntsConcurrent(t *testing.T) {
    wg := sync.WaitGroup{}
    ch := make(chan int)

    for i := 0; i < 100; i++ {
        wg.Add(1)
        go func(i int) {
            result := SumInts([]int{1, 2, 3, 4, 5})
            ch <- result
            wg.Done()
        }(i)
    }

    wg.Wait()
    close(ch)

    var sum int
    for result := range ch {
        sum += result
    }

    if sum != 100 * 15 {
        t.Errorf("Expected sum to be %d, got %d", 100 * 15, sum)
    }
}

登入後複製

在這個測試中：