测试带 context.Context 的 Go 函数需主动构造可控上下文(如 WithCancel/WithTimeout),手动触发取消或超时,精确断言 errors.Is(err, context.Canceled) 等,覆盖依赖响应与 value 边界场景。

测试带 context.Context 的 Go 函数,关键在于控制上下文生命周期、模拟取消行为、验证超时与传播逻辑,而不是简单传入 context.Background() 就完事。
用 context.WithCancel 或 context.WithTimeout 构造可控上下文
直接用 context.Background() 无法触发取消或超时,测不出函数对上下文的响应。应主动构造可取消/可超时的上下文,在测试中手动调用 cancel() 或等待超时发生。
- 测试取消场景:用
ctx, cancel := context.WithCancel(context.Background()),启动 goroutine 调用被测函数后立即cancel(),检查是否提前返回、是否释放资源 - 测试超时场景:用
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Millisecond),确保函数在超时后返回context.DeadlineExceeded错误 - 记得 defer
cancel()避免 goroutine 泄漏(尤其在非取消路径下)
断言错误是否为 context 相关错误
Go 标准库约定:上下文取消或超时应返回 context.Canceled 或 context.DeadlineExceeded。不要只检查 err != nil,要精确比对错误类型。
- 用
errors.Is(err, context.Canceled)判断是否因取消退出(兼容包装后的错误) - 同理用
errors.Is(err, context.DeadlineExceeded)检查超时路径 - 避免用
err == context.Canceled—— 实际错误常被fmt.Errorf或其他函数包装过
模拟下游依赖的 context 行为(如 HTTP client、DB 查询)
被测函数若调用 http.Client.Do 或 database/sql.QueryContext,它们本身会响应 context。测试时无需 mock 这些调用,但需确保传入的 context 确实能影响它们。
立即学习“go语言免费学习笔记(深入)”;
- 写一个真实的小 HTTP server,用
httptest.NewServer启动,handler 中 sleep 超过测试 timeout,验证 client 是否提前断开 - 对数据库操作,可用
sqlmock拦截查询并延迟响应,再配合WithTimeout触发 cancel - 如果依赖不可控(如第三方 SDK 不响应 context),应在测试中明确标注“跳过 context 测试”并加注释说明原因
注意 context.Value 的测试边界
如果函数依赖 ctx.Value(key),测试时需显式注入值,并覆盖空值、非法值、nil 值等边界情况。
- 用
context.WithValue(parent, key, value)构造带值上下文,key 建议定义为私有 unexported 类型防止冲突 - 必须测试
ctx = context.Background()(无值)路径,确认函数有合理默认行为或明确报错 - 避免在测试里用字符串或 int 当 key —— 容易和其它测试干扰,也难追溯来源
基本上就这些。context 测试不复杂但容易忽略主动控制和精确断言,抓住“谁取消、何时取消、如何响应”这条线,就能覆盖大部分场景。










