Go 中异步函数的单元测试方法

在 go 中，异步函数可以通过并发测试进行单元测试，以模拟并发执行并测试异步函数的行为。步骤如下：创建一个超时上下文。创建一个通道来接收结果。调用异步函数并将结果写入通道。从通道中读取结果并检查预期值。使用 select 语句处理超时或接收到结果。

Go 中异步函数的单元测试方法

在 Go 中，异步函数（又称协程）可以通过并发测试进行单元测试。并发测试允许我们模拟并发执行，以测试异步函数的行为。

实战案例

刺鸟创客

一款专业高效稳定的AI内容创作平台

下载

假设我们有一个名为 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 从通道中读取结果或超时，并在两种情况下进行断言以验证预期行为。