Go 语言测试框架：testing 包基础、表驱动测试、子测试、TestMain、基准测试入门

在 Go 语言生态中，标准库 testing 包是编写单元测试、集成测试与基准测试的基石。本章将系统讲解 testing 包的核心机制，包括基础测试函数、表驱动测试模式、子测试的组织方式、TestMain 的全局控制能力，以及基准测试的编写与解读方法。掌握这些内容，是构建契约压测闭环的第一步。

1. testing 包基础：测试函数规范

Go 的测试函数必须遵循以下约定：

• 文件名以 _test.go 结尾，例如 user_test.go。
• 测试函数名以 Test 开头，参数为 *testing.T。
• 使用 t.Error 或 t.Fatal 报告失败。

// calc.go
package calc

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

// calc_test.go
package calc

import "testing"

func TestAdd(t *testing.T) {
    result := Add(2, 3)
    if result != 5 {
        t.Errorf("Add(2, 3) = %d; want 5", result)
    }
}

运行测试：go test ./... 或 go test -v 查看详细输出。

2. 表驱动测试：数据与逻辑分离

表驱动测试（Table-Driven Tests）是 Go 社区推崇的测试模式，将测试用例定义为结构体切片，使测试逻辑与测试数据解耦，便于扩展和维护。

func TestAddTableDriven(t *testing.T) {
    tests := []struct {
        name string
        a, b int
        want int
    }{
        {"positive", 2, 3, 5},
        {"negative", -1, -2, -3},
        {"zero", 0, 0, 0},
        {"mixed", -5, 8, 3},
    }

    for _, tt := range tests {
        t.Run(tt.name, func(t *testing.T) {
            got := Add(tt.a, tt.b)
            if got != tt.want {
                t.Errorf("Add(%d, %d) = %d; want %d", tt.a, tt.b, got, tt.want)
            }
        })
    }
}

3. 子测试：嵌套组织与并行控制

子测试（Subtests）通过 t.Run 创建，支持嵌套、选择性执行和并行控制。结合表驱动测试，可清晰组织测试层级。

func TestUserValidation(t *testing.T) {
    t.Run("email", func(t *testing.T) {
        t.Run("valid", func(t *testing.T) {
            // 测试有效邮箱
        })
        t.Run("invalid", func(t *testing.T) {
            // 测试无效邮箱
        })
    })

    t.Run("phone", func(t *testing.T) {
        t.Parallel() // 标记为并行
        // 测试手机号
    })
}

选择性执行：go test -run "TestUserValidation/email/valid"。

4. TestMain：测试生命周期管理

TestMain 是测试包的入口函数，允许在测试运行前后执行全局设置与清理，常用于初始化数据库连接、启动 mock 服务、加载配置文件。

func TestMain(m *testing.M) {
    // 前置设置
    setup()

    // 运行所有测试
    code := m.Run()

    // 后置清理
    teardown()

    // 退出
    os.Exit(code)
}

func setup() {
    // 初始化资源
}

func teardown() {
    // 释放资源
}

5. 基准测试入门：性能度量与优化

基准测试（Benchmark）用于测量代码性能，函数名以 Benchmark 开头，参数为 *testing.B。框架自动调整 b.N 以保证稳定测量。

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

运行基准测试：go test -bench=. 或 go test -bench=BenchmarkAdd。

func BenchmarkAddParallel(b *testing.B) {
    b.RunParallel(func(pb *testing.PB) {
        for pb.Next() {
            Add(2, 3)
        }
    })
}

6. 综合示例：契约压测中的 testing 应用

在契约压测闭环中，testing 包用于验证服务接口的行为是否符合契约定义。以下示例展示如何结合表驱动测试与子测试，对 HTTP API 进行契约验证。

func TestUserAPI(t *testing.T) {
    tests := []struct {
        name   string
        method string
        path   string
        body   string
        want   int
    }{
        {"create user", "POST", "/users", `{"name":"alice"}`, 201},
        {"get user", "GET", "/users/1", "", 200},
        {"delete user", "DELETE", "/users/1", "", 204},
    }

    for _, tt := range tests {
        t.Run(tt.name, func(t *testing.T) {
            // 发起 HTTP 请求并验证状态码
            // 此处省略具体实现
        })
    }
}