本文详解 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 生命周期。这既符合真实浏览器行为，也大幅提升测试健壮性与可维护性。
