Go 中非同步函數的單元測試方法

在 Go 中，非同步函數可以透過並發測試進行單元測試，以模擬並發執行並測試非同步函數的行為。步驟如下：建立一個超時上下文。建立一個通道來接收結果。呼叫非同步函數並將結果寫入通道。從通道中讀取結果並檢查預期值。使用 select 語句處理逾時或接收到結果。

Go 中非同步函數的單元測試方法

在Go 中，非同步函數（又稱協程）可以透過並發測試進行單元測試。並發測試允許我們模擬並發執行，以測試非同步函數的行為。

實戰案例

假設我們有一個名為greetAsync() 的非同步函數，它接收一個名稱並傳回一個帶有問候訊息的chan string。以下是如何使用並發測試對該函數進行單元測試：

package async

import (
    "context"
    "testing"
    "time"
)

func TestGreetAsync(t *testing.T) {
    tests := []struct {
        name string
        expected string
    }{
       {"Alice", "Hello Alice!"},
       {"Bob", "Hello Bob!"},
    }

    for _, test := range tests {
        t.Run(test.name, func(t *testing.T) {
            // 创建一个超时上下文
            ctx, cancel := context.WithTimeout(context.Background(), 5 * time.Second)
            defer cancel()
            
            // 创建一个通道来接收结果
            ch := make(chan string, 1)
            
            // 调用 greetAsync() 并将结果写入通道
            go greetAsync(ctx, test.name, ch)
            
            // 从通道中读取结果并检查预期值
            select {
            case r := <-ch:
                if r != test.expected {
                    t.Errorf("expected %q, got %q", test.expected, r)
                }
            case <-ctx.Done():
                t.Errorf("timeout waiting for response")
            }
        })
    }
}

func greetAsync(ctx context.Context, name string, ch chan string) {
    select {
    case <-ctx.Done():
        return // 上下文已超时，返回
    default: 
        // 上下文仍在有效期内，发送问候消息
        ch <- "Hello " + name + "!"
    }
}

在這個範例中，我們設定了一個超時上下文，使用select 從通道中讀取結果或超時，並在兩種情況下進行斷言以驗證預期行為。

以上是Go 中非同步函數的單元測試方法的詳細內容。更多資訊請關注PHP中文網其他相關文章！