在Go单元测试中使用context主要用于测试函数对超时、取消和值传递的响应。1. 使用context.Background()作为根上下文测试正常流程；2. 通过context.WithCancel创建可取消上下文，验证函数能否正确处理取消信号；3. 利用context.WithTimeout确保函数在超时后及时退出；4. 用context.WithValue传递请求级数据并测试其正确读取。关键在于根据被测函数需求构造合适上下文类型，确保覆盖各种上下文状态下的行为，同时注意并发安全与资源释放，如及时调用cancel。
在Go语言的单元测试中使用context，主要是为了模拟超时控制、传递请求范围的数据，或测试函数对上下文取消的响应行为。虽然测试本身不需要context，但当你测试的函数依赖context.Context时，就需要在测试中正确构造和使用它。
使用默认的 context.Background()
大多数情况下，你可以用 context.Background() 作为根上下文传入被测函数。它是一个安全的起点，适合大多数场景。
例如：假设你有一个函数需要从数据库获取用户信息，并接受一个 context：
func GetUser(ctx context.Context, userID string) (*User, error) {
// 模拟带上下文的数据库调用
select {
case <-ctx.Done():
return nil, ctx.Err()
default:
// 正常逻辑
return &User{ID: userID, Name: "Alice"}, nil
}
}
对应的测试可以这样写：
立即学习“go语言免费学习笔记（深入）”；
func TestGetUser_Success(t *testing.T) {
ctx := context.Background()
user, err := GetUser(ctx, "123")
if err != nil {
t.Fatalf("expected no error, got %v", err)
}
if user.ID != "123" {
t.Errorf("expected ID 123, got %s", user.ID)
}
}
测试上下文取消（Cancel）
验证你的函数是否能正确响应上下文取消，是使用 context 测试的重要部分。
你可以手动创建可取消的 context，并在适当时间触发取消。
func TestGetUser_WithContextCancel(t *testing.T) {
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
// 在另一个 goroutine 中取消 context
go func() {
time.Sleep(10 * time.Millisecond)
cancel()
}()
_, err := GetUser(ctx)
// 预期因取消而返回错误
if err != context.Canceled {
t.Errorf("expected context.Canceled, got %v", err)
}
}
注意：这类测试涉及并发，建议设置超时防止死锁，比如使用 t.Run 配合子测试的超时控制。
测试上下文超时
使用 context.WithTimeout 可以测试函数在限定时间内是否正常退出。
func TestGetUser_WithTimeout(t *testing.T) {
ctx, cancel := context.WithTimeout(context.Background(), 50*time.Millisecond)
defer cancel()
start := time.Now()
_, err := GetUser(ctx)
elapsed := time.Since(start)
if err != context.DeadlineExceeded {
t.Errorf("expected deadline exceeded, got %v", err)
}
if elapsed > 100*time.Millisecond {
t.Errorf("function took too long: %v", elapsed)
}
}
向 Context 传递值进行测试
如果你的函数从 context 中读取数据（如请求ID、认证信息），你可以在测试中用 context.WithValue 构造上下文。
func GetRequestUser(ctx context.Context) string {
user := ctx.Value("user").(string)
return user
}
func TestGetRequestUser(t *testing.T) {
ctx := context.WithValue(context.Background(), "user", "alice")
user := GetRequestUser(ctx)
if user != "alice" {
t.Errorf("expected alice, got %s", user)
}
}
提示：尽量避免滥用 context 传值，仅用于请求范围的元数据。测试中也不应依赖复杂结构，保持清晰。
基本上就这些。在单元测试中使用 context 的关键是根据被测函数的行为选择合适的 context 类型：Background、带取消、带超时或带值。核心目标是验证函数在各种 context 状态下的正确响应。不复杂但容易忽略细节，比如忘记调用 cancel 或未处理 Done channel。
