Go程序例子(71)：单元测试

// 为了演示，这段代码放在 main 包中，但它可以是任何包。测试代码通常与它测试的代码位于同一个包中。
package main

import (
    "fmt"
    "testing"
)
// 我们将测试这个简单的整数最小值实现。通常情况下，我们测试的代码会放在名为 intutils.go 的源文件中，而它的测试文件则会命名为 intutils_test.go。
func IntMin(a, b int) int {
    if a < b {
        return a
    }
    return b
}
// 通过编写一个以 Test 开头的函数来创建测试。
func TestIntMinBasic(t *testing.T) {
    ans := IntMin(2, -2)
    if ans != -2 {
				// t.Error* 会报告测试失败但继续执行测试，而 t.Fatal* 会报告测试失败并立即停止测试。
        t.Errorf("IntMin(2, -2) = %d; want -2", ans)
    }
}
// 编写测试可能会很重复，因此惯用的方式是使用表驱动风格，即将测试输入和预期输出列在一个表中，然后通过一个循环遍历它们并执行测试逻辑。
func TestIntMinTableDriven(t *testing.T) {
    var tests = []struct {
        a, b int
        want int
    }{
        {0, 1, 0},
        {1, 0, 0},
        {2, -2, -2},
        {0, -1, -1},
        {-1, 0, -1},
    }
		// t.Run 允许运行“子测试”，每个表条目对应一个子测试。在执行 go test -v 时，这些子测试会单独显示。
    for _, tt := range tests {

        testname := fmt.Sprintf("%d,%d", tt.a, tt.b)
        t.Run(testname, func(t *testing.T) {
            ans := IntMin(tt.a, tt.b)
            if ans != tt.want {
                t.Errorf("got %d, want %d", ans, tt.want)
            }
        })
    }
}
// 基准测试通常放在 _test.go 文件中，并以 Benchmark 开头命名。测试运行器会多次执行每个基准函数，每次运行时增加 b.N，直到收集到精确的测量结果。
func BenchmarkIntMin(b *testing.B) {
		// 通常情况下，基准测试会在循环中运行我们要测试的函数 b.N 次。
    for i := 0; i < b.N; i++ {
        IntMin(1, 2)
    }
}

➜ go test -v   
=== RUN   TestIntMinBasic
--- PASS: TestIntMinBasic (0.00s)
=== RUN   TestIntMinTableDriven
=== RUN   TestIntMinTableDriven/0,1
=== RUN   TestIntMinTableDriven/1,0
=== RUN   TestIntMinTableDriven/2,-2
=== RUN   TestIntMinTableDriven/0,-1
=== RUN   TestIntMinTableDriven/-1,0
--- PASS: TestIntMinTableDriven (0.00s)
    --- PASS: TestIntMinTableDriven/0,1 (0.00s)
    --- PASS: TestIntMinTableDriven/1,0 (0.00s)
    --- PASS: TestIntMinTableDriven/2,-2 (0.00s)
    --- PASS: TestIntMinTableDriven/0,-1 (0.00s)
    --- PASS: TestIntMinTableDriven/-1,0 (0.00s)
PASS
ok      testing-and-benchmarking        0.196s