3. Go 测试进阶：Benchmark、Example、TestMain 与覆盖率

单元测试与表驱动测试解决了功能正确性的问题，但在生产级项目中，我们还需要回答三个更深入的问题：代码跑得快不快？（性能）、用法对不对？（文档与可读性）、测全了没有？（覆盖率）。本章将系统讲解 Go 测试体系中这四个进阶工具，帮助你构建更健壮的冒烟回归测试套件。

3.1 Benchmark 基准测试：量化性能，防止回归

基准测试用于测量代码的执行耗时、内存分配次数和分配字节数。在冒烟回归流程中，基准测试可以捕获因代码变更导致的性能退化。

3.1.1 基本规则与命名

• 文件名以 _test.go 结尾。
• 函数签名：func BenchmarkXxx(b *testing.B)。
• 参数 b.N 由框架自动调整，保证测试有足够的运行时间以获取稳定数据。

3.1.2 一个完整的基准测试示例

// string_concat_test.go
package benchmark

import (
    "strings"
    "testing"
)

// 待测函数：使用 + 拼接
func ConcatPlus(parts []string) string {
    result := ""
    for _, p := range parts {
        result += p
    }
    return result
}

// 待测函数：使用 strings.Builder
func ConcatBuilder(parts []string) string {
    var b strings.Builder
    for _, p := range parts {
        b.WriteString(p)
    }
    return b.String()
}

func BenchmarkConcatPlus(b *testing.B) {
    parts := []string{"Go", "语言", "冒烟", "回归", "测试"}
    b.ResetTimer() // 重置计时器，排除准备数据的时间
    for i := 0; i < b.N; i++ {
        ConcatPlus(parts)
    }
}

func BenchmarkConcatBuilder(b *testing.B) {
    parts := []string{"Go", "语言", "冒烟", "回归", "测试"}
    b.ResetTimer()
    for i := 0; i < b.N; i++ {
        ConcatBuilder(parts)
    }
}

3.1.3 运行与解读

go test -bench=. -benchmem

3.2 Example 示例测试：可执行文档

Example 函数既是测试，也是文档。Go 的 go doc 和 godoc 会将其输出直接渲染到文档中，同时 go test 会验证示例的输出是否与注释匹配。

3.2.1 命名与结构

• 函数名：Example、ExampleXxx、ExampleXxx_suffix。
• 在函数体末尾使用 // Output: 注释声明期望输出。
• 也可以使用 // Unordered output: 忽略行顺序。

3.2.2 示例代码

package example_test

import (
    "fmt"
    "strings"
)

func ExampleJoin() {
    result := strings.Join([]string{"Go", "测试", "进阶"}, " ")
    fmt.Println(result)
    // Output: Go 测试 进阶
}

func ExampleJoin_unordered() {
    words := []string{"apple", "banana", "cherry"}
    for _, w := range words {
        fmt.Println(w)
    }
    // Unordered output:
    // banana
    // cherry
    // apple
}

3.2.3 在冒烟回归中的价值

3.3 TestMain：自定义测试入口

当测试套件需要全局的初始化与清理（如启动数据库容器、加载配置文件、建立连接池）时，TestMain 提供了统一的入口点。

3.3.1 函数签名与流程

func TestMain(m *testing.M) {
    // 1. 全局初始化
    setup()

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

    // 3. 全局清理
    teardown()

    // 4. 退出
    os.Exit(code)
}

3.3.2 冒烟回归中的典型用法

package smoke

import (
    "os"
    "testing"
)

var dbConn *DB

func setup() {
    // 启动测试容器或连接测试数据库
    dbConn = connectToTestDB()
}

func teardown() {
    dbConn.Close()
}

func TestMain(m *testing.M) {
    setup()
    code := m.Run()
    teardown()
    os.Exit(code)
}

func TestQueryUser(t *testing.T) {
    user, err := dbConn.QueryUser(1)
    if err != nil {
        t.Fatalf("QueryUser failed: %v", err)
    }
    if user.Name != "Alice" {
        t.Errorf("expected Alice, got %s", user.Name)
    }
}

3.3.3 与冒烟回归的集成

• 环境准备：在 setup() 中启动 Docker 容器或 Mock 服务。
• 资源隔离：每个测试包可以有自己的 TestMain，互不干扰。
• 失败快速：如果 setup() 失败，直接 os.Exit(1)，避免运行无效测试。

3.4 测试覆盖率：量化测试的广度

覆盖率是冒烟回归质量门禁的重要指标。Go 内置了覆盖率工具，无需第三方库。

3.4.1 基本命令

# 运行测试并生成覆盖率数据
go test -coverprofile=coverage.out

# 查看覆盖率百分比
go tool cover -func=coverage.out

# 生成 HTML 可视化报告
go tool cover -html=coverage.out -o coverage.html

3.4.2 覆盖率类型与解读

3.4.3 在冒烟回归中设置门禁

# 在 CI 脚本中检查覆盖率是否达标
go test -coverprofile=coverage.out ./...
COVERAGE=$(go tool cover -func=coverage.out | grep total | awk '{print $3}' | sed 's/%//')
if (( $(echo "$COVERAGE < 80.0" | bc -l) )); then
    echo "❌ 覆盖率 $COVERAGE% 低于门禁 80%"
    exit 1
fi
echo "✅ 覆盖率 $COVERAGE% 通过"

3.5 本章知识体系总览

3.6 冒烟回归中的组合策略

在实际项目中，这四个工具不是孤立使用的。以下表格展示了它们在冒烟回归流水线中的典型角色：