在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() <pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">// 在另一个 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() <pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">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 } <p>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。