golang 单元测试(一)

单元测试函数类型

Test(功能测试)

函数规则:

• 函数名: TestXxxx , 以Test为前缀。Xxxx以大写字母开头
• 参数类型: *testing.T

func TestXxxx(t *testing.T){...}

编写一个简单的例子,假设有下面一个待测函数:

func add(a, b int) int {
    return a + b
}

测试代码如下:

import "testing"

func TestAdd(t *testing.T) {
    cases := []struct {
        first    int
        second   int
        excepted int
    }{
        {1, 2, 3},
        {1, 2, 4},
    }

    for _, c := range cases {
        result := add(c.first, c.second)
        if result != c.excepted {
            t.Fatalf("add function failed, first: %d, second:%d, execpted:%d, result:%d", c.first, c.second, c.excepted, result)
        }
    }
}

执行 go go test -v 结果如下:

=== RUN   TestAdd
--- FAIL: TestAdd (0.00s)
    example_test.go:30: add function failed, first: 1, second:2, execpted:4, result:3
FAIL
exit status 1
FAIL    gotest/gotest   0.006s

第二个测试用例出错,未能通过单元测试,需要检查测试数据或者被测试函数是否符合我们的预期。

从上述过程可以看出, TestXxxx 可以理解为功能性测试,其目的在于测试函数的功能性是否正确, 应当编写尽可能多的测试用例来验证被测函数的正确性。

Benchmark(基准测试)

函数规则:

• 函数名: BenchmarkXxxx, 以Benchmark为前缀。Xxxx以大写字母开头
• 参数类型: *testing.B

func BenchmarkXxx(b *testing.B){...}

性能测试函数如下:

func BenchmarkAdd(b *testing.B) {
    for i := 0; i < b.N; i++ {
        add(1, 2)
    }
}

执行 go test -bench=. -run=^$ ,结果如下:

goos: darwin
goarch: amd64
pkg: gotest/gotest
BenchmarkAdd-8      2000000000           0.32 ns/op
PASS
ok      gotest/gotest   0.667s

测试结果说明:

• -8 表示 8个逻辑cpu个数(下文会解释)
• 2000000000 表示执行了 2000000000 次
• 0.32 ns/op 每次操作耗时
• 0.667s 是总时长

性能测试函数计时器:

当性能测试函数中有一些额外的初始化操作时,可以通过启停计时器来屏蔽这些操作对最终性能测试的影响。简单例子如下:

func BenchmarkAdd(b *testing.B) {
    time.Sleep(10 * time.Second)
    for i := 0; i < b.N; i++ {
        add(1, 2)
    }
}

执行命令 go test -bench=. ,结果如下:

goos: darwin
goarch: amd64
pkg: gotest/gotest
BenchmarkAdd-8             1    10004479538 ns/op
PASS
ok      gotest/gotest   10.011s

单个测试函数耗时为 10004479538 ns,同时我们也可以看到,这个函数执行一次就已经达到了最大上限时间。

加上时间计数器:

func BenchmarkAdd(b *testing.B) {
    b.StopTimer()
    time.Sleep(10 * time.Second)
    b.StartTimer()
    for i := 0; i < b.N; i++ {
        add(1, 2)
    }
}

测试结果如下:

goos: darwin
goarch: amd64
pkg: gotest/gotest
BenchmarkAdd-8      2000000000           0.34 ns/op
PASS
ok      gotest/gotest   60.751s

单次执行耗时为: 0.34 ns 了。

Tips:

对于 b.N 参数而言, test 命令先会尝试设置为1,之后执行函数。如果测试函数没有达到执行上限的话, test 函数会增大,之后再次执行测试函数,如此往复,直到执行时间大于或者等于上限时间为止。 上限时间可以使用 -benchtime 设定

Example(示例测试)

函数规则:

• 函数名: ExampleXxx
• 输出内容至标准输出,参数列表没有限制

示例测试函数如下:

func Println(content string) {
    fmt.Println("The output of\nthis example.")
}
func ExamplePrintln() {
    Println("The output of\nthis example.")
    // Output: The output of
    // this example.
}

go test 命令会检测实际输出与注释中的期望输出是否一致, 一致则测试通过,不一致则测试失败。

go test 常用参数

• -cpu: 设置测试最大 cpu 逻辑数(也就是 GPM 中 P, 最大并行执行的 gorouting 数量, 默认等于 cpu 核数)
• -count: 设置执行测试函数的次数, 默认为 1
• -run: 执行功能测试函数, 支持正则匹配, 可以选择测试函数或者测试文件来仅测试单个函数或者单个文件
• -bench: 执行基准测试函数, 支持正在匹配
• -benchtime: 基准测试最大探索时间上限
• -parallel: 设置同一个被测试代码包中的功能测试函数的最大并发执行数
• -v: 是展示测试过程信息

更多的参数可以参考下文的官方文档

参考

golang 官方文档

极客时间