在 Go 测试中模拟上下文取消

上下文取消是 Go 中一个强大的工具，用于控制协程的生命周期。在测试中模拟上下文取消对于验证处理取消的情况是否正确非常重要。

使用 context.WithCancel

模拟上下文取消的最简单方法是使用 [context.WithCancel](https://pkg.go.dev/context#WithCancel) 函数。它会创建一个新的上下文，其中包含一个取消函数:

立即学习“go语言免费学习笔记（深入）”；

ctx, cancel := context.WithCancel(context.Background())

cancel 函数可用于取消上下文:

cancel()

实战案例

考虑以下函数，该函数从信道读取值，直到上下文被取消:

func ReadValues(ctx context.Context, ch <-chan int) <-chan int {
    out := make(chan int)
    go func() {
        for {
            select {
            case <-ctx.Done():
                close(out)
                return
            case v := <-ch:
                out <- v
            }
        }
    }()
    return out
}

测试模拟取消

为了测试此函数，可以在测试中模拟上下文取消:

func TestReadValues(t *testing.T) {
    ctx, cancel := context.WithCancel(context.Background())
    values := []int{1, 2, 3}
    ch := make(chan int, len(values))
    for _, v := range values {
        ch <- v
    }

    out := ReadValues(ctx, ch)

    // 接收通道中的值，直到上下文被取消
    received := []int{}
    for v := range out {
        received = append(received, v)
    }

    // 检查收到的值
    if !reflect.DeepEqual(received, values) {
        t.Errorf("Received values do not match expected values")
    }

    // 取消上下文
    cancel()
}

这将测试 ReadValues 函数在上下文被取消时处理值的行为。

以上就是Golang 函数：如何在测试中模拟上下文取消的详细内容，更多请关注php中文网其它相关文章！