Go recorder 如何限定测试响应

Go 测试 HTTP Handler 时，httptest.NewRecorder() 适合验证“这个处理函数写出了什么响应”，不负责模拟真实网络连接。最稳的断言入口是 Handler 执行完成后的 recorder.Result()：从返回的 *http.Response 读取最终状态码、响应头和响应体；只有在专门测试刷新行为时才检查 Flushed，不要把 Code 或已弃用的 HeaderMap 当作完整结果。

一、先用 NewRecorder 把 Handler 输出接住

测试入口通常是一个请求对象、一个 recorder 和一次 Handler 调用。这个过程不需要启动端口，适合快速覆盖鉴权失败、参数校验和 JSON 返回等分支。

func TestCreate(t *testing.T) {
    // 请求只用于给 Handler 提供方法、URL 和上下文。
    req := httptest.NewRequest(http.MethodPost, "/items", nil)
    recorder := httptest.NewRecorder()

    handler(recorder, req)

    // Handler 执行完成后，再读取 recorder 中记录的响应。
    resp := recorder.Result()
    defer resp.Body.Close()
}

这里的关键是“直接调用 Handler”。如果测试代码绕过了 Handler，或者在它执行完之前读取 Result，断言就没有明确意义。

二、Result 才是判断最终响应的主入口

Result() 返回一个响应对象，测试可以像读取普通 HTTP 响应一样读取字段。响应体读取完应关闭 Body，避免测试代码养成不清理资源的习惯。

func TestCreateStatus(t *testing.T) {
    // 这个 Handler 显式返回 201，便于观察状态断言边界。
    handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        w.Header().Set("Content-Type", "application/json")
        w.WriteHeader(http.StatusCreated)
        _, _ = w.Write([]byte(`{"ok":true}`))
    })

    recorder := httptest.NewRecorder()
    handler.ServeHTTP(recorder, httptest.NewRequest(http.MethodPost, "/items", nil))
    resp := recorder.Result()
    defer resp.Body.Close()

    // Result 给出最终状态和第一次写入时的响应头快照。
    if resp.StatusCode != http.StatusCreated {
        t.Fatalf("status = %d, want %d", resp.StatusCode, http.StatusCreated)
    }
    if got := resp.Header.Get("Content-Type"); got != "application/json" {
        t.Fatalf("content type = %q", got)
    }
}

官方实现说明，Handler 从未调用 WriteHeader 或 Write 时，recorder 的 Code 可能仍为 0；此时用 Result 读取隐式的 200 更符合用户实际收到的响应。

三、Code、HeaderMap 和 Flushed 各自只解决一个问题

普通 JSON 或 HTML Handler 只需断言 Result。只有测试 SSE、分块输出或显式刷新时，才额外检查 recorder.Flushed。如果测试目标是响应头，优先读取 resp.Header，因为它代表 Result 生成的响应视图。

四、什么时候应该换成 httptest.NewServer

ResponseRecorder 的边界在 Handler 内部：它不会替你验证 TCP 连接、客户端 Cookie 保存、重定向跟随、真实 URL 解析或中间件在服务器上的组合行为。遇到这些目标，改用 httptest.NewServer(handler)，让 http.Client 发起真实的本地 HTTP 请求。

可以把两层测试分开：ResponseRecorder 覆盖大量分支和错误返回，NewServer 只保留少量关键链路。这样失败时能快速判断是 Handler 输出错了，还是服务器级行为改变了，测试也不会因为每个断言都启动完整链路而变慢。

常见问题

为什么 recorder.Code 是 0，但接口看起来应该返回 200？

Handler 可能没有显式写响应。调用完成后的 Result 会补出隐式成功状态，普通断言不要直接依赖 Code。

Result 返回的 Body 需要关闭吗？

需要。把读取后的 Body 用 defer 关闭，保持测试代码与真实 HTTP 客户端一样的资源习惯。

ResponseRecorder 能测试 Flush 吗？

可以检查 Flushed 是否为 true，但这只能说明 Handler 调用了 Flush，不等于真实网络环境中的流式传输时序已经被完整验证。