用go testing 作单元测试压力测试

这里是个简单的例子，用go testing 作单元测试压力测试，压力测试可用作性能测试，对比不同的实现方式的性能。

测试的包文件夹 mypack 共有三个文件 :

• mypack.go 单元功能文件
• mypack_test.go 单元测试文件
• mypack_benchmark_test.go 压力测试文件

mypack.go 文件的内容：

package mypack

import (
    "errors"
)

func Division(a, b float64) (float64, error) {
    if b == 0 {
        return 0, errors.New("除数不能为0")
    }

    return a / b, nil
}

mypack_test.go 的文件内容：

package mypack

import (
    "testing"
)

// 输入的参数列表
var fibTests = []struct {
    a        float64 // input
    b        float64 // input
    expected float64 // expected result
}{
    {4, 2, 2},
    {6, 3, 2},
    {100, 10, 10},
    {8, 2, 4},
    {15, 5, 3},
}

func Test_Division_1(t *testing.T) {
    for _, fib := range fibTests {
        if i, e := Division(fib.a, fib.b); i != fib.expected || e != nil { //try a unit test on function
            t.Error("除法函数测试没通过") // 如果不是如预期的那么就报错
        } else {
            t.Log("第一个测试通过了") //记录一些你期望记录的信息
        }
    }
}

func Test_Division_2(t *testing.T) {
    if _, e := Division(6, 0); e == nil { //try a unit test on function
        t.Error("Division did not work as expected.") // 如果不是如预期的那么就报错
    } else {
        t.Log("one test passed.", e) //记录一些你期望记录的信息
    }
}

mypack_benchmark_test.go 的文件内容：

package mypack

import (
    "testing"
)

func Benchmark_Division(b *testing.B) {
    for i := 0; i < b.N; i++ { //use b.N for looping 
        Division(4, 5)
    }
}

func Benchmark_TimeConsumingFunction(b *testing.B) {
    b.StopTimer() //调用该函数停止压力测试的时间计数

    //做一些初始化的工作,例如读取文件数据,数据库连接之类的,
    //这样这些时间不影响我们测试函数本身的性能

    b.StartTimer() //重新开始时间
    for i := 0; i < b.N; i++ {
        Division(4, 5)
    }
}

切换到这个文件夹下实行：

go test -v -bench=".*"

输出：

=== RUN Test_Division_1
--- PASS: Test_Division_1 (0.00s)
    mypack_test.go:25: 第一个测试通过了
    mypack_test.go:25: 第一个测试通过了
    mypack_test.go:25: 第一个测试通过了
    mypack_test.go:25: 第一个测试通过了
    mypack_test.go:25: 第一个测试通过了
=== RUN Test_Division_2
--- PASS: Test_Division_2 (0.00s)
    mypack_test.go:34: one test passed. 除数不能为0
PASS
Benchmark_Division  200000000            6.96 ns/op
Benchmark_TimeConsumingFunction 200000000            6.94 ns/op
ok      _/home/ubuntu/tmp/mypack    4.203s