4. 测试框架选型

在 Go Web 契约压测中，测试框架的选型直接决定了脚本的可维护性、断言能力与压测效率。本章从 Go 原生测试工具链出发，逐步深入到第三方库的对比，帮助你在不同场景下做出合理选择。

4.1 Go 测试工具链概览

Go 语言自 1.0 版本起就内置了完整的测试基础设施，无需额外安装即可编写单元测试、基准测试和示例测试。下表梳理了 Go 原生工具链的核心组件及其在契约压测中的角色：

4.2 testing 包基础：测试函数与子测试

testing 包是所有 Go 测试的基石。在契约压测中，我们主要使用以下模式：

• Test 函数：签名 func TestXxx(t *testing.T)，用于验证单个契约场景。
• 子测试：通过 t.Run(name, func(t *testing.T)) 组织多组输入，实现契约矩阵的并行验证。
• Cleanup 注册：t.Cleanup(func()) 确保测试资源（如测试服务器）正确释放。
• 错误报告：t.Error / t.Fatal / t.Log 控制测试流程与输出。

func TestUserContract(t *testing.T) {
    // 启动测试服务器
    ts := httptest.NewServer(handler)
    t.Cleanup(ts.Close)

    // 子测试：正常创建用户
    t.Run("create_user_201", func(t *testing.T) {
        resp, err := http.Post(ts.URL+"/users", "application/json",
            strings.NewReader(`{"name":"alice"}`))
        if err != nil {
            t.Fatal(err)
        }
        if resp.StatusCode != http.StatusCreated {
            t.Errorf("expected 201, got %d", resp.StatusCode)
        }
    })

    // 子测试：重复创建应返回 409
    t.Run("create_user_duplicate_409", func(t *testing.T) {
        // ... 类似逻辑
    })
}

4.3 httptest 包介绍：本地模拟 HTTP 交互

net/http/httptest 是契约压测的核心工具，它提供了两种关键能力：

• httptest.NewServer：基于真实的 net.Listener 启动一个本地 HTTP 服务器，支持 TLS、自定义 Handler 和中间件。
• httptest.NewRecorder：实现 http.ResponseWriter，用于在单元测试中捕获 Handler 的响应，无需启动网络端口。

在契约压测中，NewServer 用于启动被测服务的完整实例，NewRecorder 则用于快速验证 Handler 级别的契约行为。下图展示了两种模式在测试流程中的位置：

4.4 第三方库对比：testify、gock、vegeta

当原生工具无法满足复杂断言、外部依赖模拟或高并发压测时，第三方库成为必要补充。下表从契约压测的四个关键维度进行对比：

4.4.1 testify：增强断言与测试套件

testify 是 Go 社区最流行的测试增强库，其 assert 包提供了比原生 t.Error 更丰富的比较函数：

import "github.com/stretchr/testify/assert"

func TestContractWithTestify(t *testing.T) {
    resp := doCreateUser(t, `{"name":"bob"}`)

    // 链式断言，失败时输出友好差异
    assert.Equal(t, 201, resp.StatusCode)
    assert.Contains(t, resp.Header.Get("Content-Type"), "application/json")

    var body map[string]interface{}
    json.NewDecoder(resp.Body).Decode(&body)
    assert.Equal(t, "bob", body["name"])
    assert.NotEmpty(t, body["id"])
}

4.4.2 gock：拦截外部 HTTP 依赖

在契约压测中，被测服务可能依赖第三方 API。gock 可以在不启动真实外部服务的情况下，模拟其契约响应：

import "github.com/h2non/gock"

func TestExternalContract(t *testing.T) {
    defer gock.Off() // 确保测试结束后恢复

    // 模拟外部支付服务的契约
    gock.New("https://payment.example.com").
        Post("/charge").
        MatchHeader("Authorization", "Bearer token123").
        JSON(map[string]string{"amount": "100"}).
        Reply(200).
        JSON(map[string]string{"status": "success"})

    // 执行被测逻辑，此时 HTTP 调用会被 gock 拦截
    result := processPayment("token123", 100)
    assert.Equal(t, "success", result.Status)
}

4.4.3 vegeta：契约压测的负载生成器

vegeta 专为 HTTP 压测设计，支持恒定速率和递增速率两种攻击模式。在契约压测中，我们用它来验证服务在负载下的契约行为：

import "github.com/tsenart/vegeta/v12/lib"

func TestContractUnderLoad(t *testing.T) {
    rate := vegeta.Rate{Freq: 100, Per: time.Second} // 100 QPS
    duration := 10 * time.Second

    targeter := vegeta.NewStaticTargeter(vegeta.Target{
        Method: "POST",
        URL:    "http://localhost:8080/api/orders",
        Body:   []byte(`{"product":"book","qty":1}`),
        Header: http.Header{"Content-Type": []string{"application/json"}},
    })

    attacker := vegeta.NewAttacker()
    var metrics vegeta.Metrics
    for res := range attacker.Attack(targeter, rate, duration, "contract-test") {
        metrics.Add(res)
        // 在压测过程中实时验证契约
        if res.Code != 201 {
            t.Errorf("contract violation: expected 201, got %d", res.Code)
        }
    }
    metrics.Close()

    // 输出压测指标
    t.Logf("99th percentile latency: %v", metrics.Latencies.P99)
    t.Logf("success rate: %.2f%%", metrics.Success*100)
}

4.5 选型决策流程

综合以上分析，在 Go Web 契约压测中，推荐按以下流程选择测试框架：