Go 中如何测试带 Cookie 的 HTTP 请求

本文详解 Go 单元测试中处理 Cookie 的常见误区，重点解决 `Response.Location()` 报错问题，并提供基于 `http.Client.Jar` 的自动化 Cookie 管理方案，附可运行示例代码。

在 Go 的 HTTP 测试中，初学者常误用 resp.Location() 来获取响应来源 URL，从而导致测试失败——正如错误信息所示：http: no Location header in response。这是因为 Location 响应头仅存在于 3xx 重定向响应中，而普通成功响应（如 200 OK）默认不包含该头字段。直接调用 rs.Location() 会返回 nil 和错误，进而中断测试流程。

正确的做法是：若需获取本次响应所对应的请求地址（即 Cookie 的作用域依据），应使用 rs.Request.URL：

// ✅ 正确：获取发起该响应的原始请求 URL（用于 Cookie 作用域判断）
originURL := rs.Request.URL // 或 rs.Request.URL.String()

// ❌ 错误：Location 仅对重定向有效，普通响应会 panic
// url, err := rs.Location() // → http: no Location header in response

但更关键的是：测试 Cookie 逻辑时，通常无需手动提取并设置 Cookie。Go 标准库已内置完善的 Cookie 管理机制——只需为 http.Client 配置 Jar 字段，即可自动处理 Cookie 的接收、存储与发送：

import (
    "net/http"
    "net/http/cookiejar"
    "net/http/httptest"
    "testing"

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

func TestAuthorizationReader(t *testing.T) {
    // 模拟服务端：返回带 Set-Cookie 的响应
    server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        http.SetCookie(w, &http.Cookie{
            Name:  "auth_token",
            Value: "abc123",
            Path:  "/",
            HttpOnly: true,
        })
        w.WriteHeader(http.StatusOK)
        w.Write([]byte("OK"))
    }))
    defer server.Close()

    // ✅ 启用自动 Cookie 管理
    jar, _ := cookiejar.New(nil)
    client := &http.Client{
        Jar: jar,
    }

    // 第一次请求：服务端设 Cookie
    resp1, err := client.Get(server.URL)
    assert.NoError(t, err)
    assert.Equal(t, http.StatusOK, resp1.StatusCode)
    resp1.Body.Close()

    // 第二次请求：Client 自动携带 Cookie
    resp2, err := client.Get(server.URL)
    assert.NoError(t, err)
    assert.Equal(t, http.StatusOK, resp2.StatusCode)

    // 验证请求头是否含 Cookie（可选断言）
    req2 := resp2.Request
    assert.Contains(t, req2.Header.Get("Cookie"), "auth_token=abc123")
    resp2.Body.Close()
}

⚠️ 注意事项：

• cookiejar.New(nil) 接收一个 *cookiejar.Options，传 nil 表示使用默认策略（支持子域名、路径匹配等）；
• 手动调用 c.Jar.SetCookies(url, cookies) 仅在特殊场景（如模拟跨域注入）才需使用，常规测试应依赖自动管理；
• httptest.NewServer 创建的是真实 HTTP 服务，务必 defer server.Close() 防止端口泄漏；
• 若被测中间件（如 Use(testReader())）依赖特定路由或上下文，请确保 httptest.NewServer 的 handler 正确复现其行为。

总结：测试 Cookie 的核心不是“手动搬运”，而是“构造符合规范的交互链路”——启用 Client.Jar，让标准库按 RFC 6265 自动处理 Cookie 生命周期。这既符合真实浏览器行为，也大幅提升测试健壮性与可维护性。