如何使用Golang实现单元与性能测试结合_Golang testing综合测试示例

go test 默认不运行基准测试,需用 -bench 参数;可同时执行单元测试和基准测试,如 go test -run=TestAdd -bench=BenchmarkAdd -benchmem。

如何用 go test 同时跑单元测试和基准测试

默认情况下 go test 不会执行 Benchmark* 函数,必须显式加 -bench 参数。想“一次命令兼顾两者”,得组合使用 -run-bench,但要注意它们的匹配逻辑互不干扰:

  • -run 只控制 Test* 的执行(支持正则,如 -run=^TestAdd$
  • -bench 只控制 Benchmark* 的执行(也支持正则,如 -bench=^BenchmarkAdd$
  • 二者可共存:go test -run=TestAdd -bench=BenchmarkAdd -benchmem
  • 若只写 -bench=.,它会运行所有基准测试,不管 -run 是否匹配到测试函数

Benchmark 中调用 testing.B 的常见误用

基准测试不是把逻辑塞进 b.N 循环就完事——循环体里不能含初始化、I/O、随机数等干扰项,否则结果失真。典型错误包括:

  • for i := 0; i 内部调用 rand.Intn()time.Now()
  • 每次循环都新建大结构体或分配 slice,未复用
  • 忘记调用 b.ReportAllocs() 就断言内存表现
  • b.StopTimer() / b.StartTimer() 位置不对,漏掉关键路径计时

正确做法是把预热、准备、清理拆开:

func BenchmarkParseJSON(b *testing.B) {
    data := []byte(`{"name":"foo","age":42}`)
    var v map[string]interface{}
    
    b.ResetTimer() // 确保只测核心解析
    for i := 0; i < b.N; i++ {
        json.Unmarshal(data, &v)
    }
}

如何为同一函数写单元测试与性能测试并共享逻辑

避免重复实现,建议把被测逻辑封装成导出函数或闭包,单元测试和基准测试都调用它。不要在 Test* 里复制 Benchmark* 的循环逻辑。

  • 把核心逻辑抽成独立函数,例如 CalculateSum(nums []int) int
  • 单元测试验证边界值:TestCalculateSum(t *testing.T) 调用它并比对结果
  • 基准测试专注吞吐:BenchmarkCalculateSum(b *testing.B) 在循环中调用它
  • 若需模拟耗时操作(如加锁、channel 通信),务必在基准测试中用 b.ReportMetric() 显式标注单位,例如 b.ReportMetric(float64(costMs), "ms/op")

为什么 go test -bench=. -benchmem 结果里 B/op 有时为 0

B/op 表示每次操作平均分配的字节数,为 0 通常意味着:编译器做了逃逸分析优化,把本该堆分配的对象转为栈分配;或者你压根没触发内存分配(比如纯计算、复用已有变量)。

  • 检查是否用了 make([]int, 0, N) 预分配容量,避免扩容导致额外分配
  • 确认没有隐式字符串转 []byte 或反之(string(b) / []byte(s) 都分配)
  • go build -gcflags="-m" your_file.go 查看逃逸分析输出
  • -benchmem 必须和 -bench 一起用才生效,单独用无效

真正难的是让性能测试反映真实负载——比如加锁逻辑在单 goroutine 下快如闪电,一上多协程就暴露竞争,这种场景得靠 runtime.GOMAXPROCS 和手动启多个 goroutine 模拟,而不是只依赖默认的单线程 b.N 循环。