编写压力测试

压力测试用来检测函数(方法）的性能，和编写单元功能测试的方法类似,此处不再赘述，但需要注意以下几点：

  • 压力测试用例必须遵循如下格式，其中XXX可以是任意字母数字的组合，但是首字母不能是小写字母
  1. func BenchmarkXXX(b *testing.B) { ... }
  • go test不会默认执行压力测试的函数，如果要执行压力测试需要带上参数-test.bench，语法:-test.bench="test_name_regex",例如go test -test.bench=".*"表示测试全部的压力测试函数
  • 在压力测试用例中,请记得在循环体内使用testing.B.N,以使测试可以正常的运行
  • 文件名也必须以_test.go结尾

下面我们新建一个压力测试文件webbench_test.go，代码如下所示：

  1. package gotest
  3. import (
  4.     "testing"
  5. )
  7. func Benchmark_Division(b *testing.B) {
  8.     for i := 0; i < b.N; i++ { //use b.N for looping 
  9.         Division(4, 5)
  10.     }
  11. }
  13. func Benchmark_TimeConsumingFunction(b *testing.B) {
  14.     b.StopTimer() //调用该函数停止压力测试的时间计数
  16.     //做一些初始化的工作,例如读取文件数据,数据库连接之类的,
  17.     //这样这些时间不影响我们测试函数本身的性能
  19.     b.StartTimer() //重新开始时间
  20.     for i := 0; i < b.N; i++ {
  21.         Division(4, 5)
  22.     }
  23. }

我们执行命令go test webbench_test.go -test.bench=".*"，可以看到如下结果：

  1. Benchmark_Division-4                            500000000          7.76 ns/op         456 B/op          14 allocs/op
  2. Benchmark_TimeConsumingFunction-4            500000000          7.80 ns/op         224 B/op           4 allocs/op
  3. PASS
  4. ok      gotest    9.364s

上面的结果显示我们没有执行任何TestXXX的单元测试函数，显示的结果只执行了压力测试函数，第一条显示了Benchmark_Division执行了500000000次，每次的执行平均时间是7.76纳秒，第二条显示了Benchmark_TimeConsumingFunction执行了500000000，每次的平均执行时间是7.80纳秒。最后一条显示总共的执行时间。

通过上面对单元测试和压力测试的学习，我们可以看到testing包很轻量，编写单元测试和压力测试用例非常简单，配合内置的go test命令就可以非常方便的进行测试，这样在我们每次修改完代码,执行一下go test就可以简单的完成回归测试了。