Golang 函数：如何在测试中模拟上下文取消-Golang-PHP中文网

Golang 函数：如何在测试中模拟上下文取消

王林

发布： 2024-10-06 15:39:02

原创

1123人浏览过

golang 函数：如何在测试中模拟上下文取消

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

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

使用 context.WithCancel

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

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

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

登录后复制

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

面试猫

AI面试助手，在线面试神器，助你轻松拿Offer

查看详情

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中文网其它相关文章！