Go httptest.NewRequest设置请求体与ContentLength的测试方法

测试 Go 的 HTTP Handler 时，httptest.NewRequest 适合构造“服务端收到的入站请求”。要让请求体和长度字段保持一致，优先传入 *bytes.Buffer、*bytes.Reader 或 *strings.Reader；这三类 Reader 的当前剩余长度会被写入 Request.ContentLength。普通的 io.Reader 无法预先知道长度时，ContentLength 会是 -1。

官方文档：https://pkg.go.dev/net/http/httptest

先确认 NewRequest 构造的是入站请求

httptest.NewRequest 返回的是交给 http.Handler 的请求对象，目标地址可以写成路径，也可以写绝对 URL。空方法会按 GET 处理；使用 https:// 目标时，测试请求还会带一个非空的 TLS 状态。它的用途是模拟服务端看到的请求，而不是拿去调用 http.Client.Do。

最容易混淆的是 Body 与 ContentLength 的关系：Body 是可读取的数据流，ContentLength 是构造请求时记录的长度提示。两者应当一起检查，但不能把 Body 读取后的剩余长度当成请求原始长度。

用可确定长度的 Reader 设置请求体

JSON 测试通常从 bytes.NewBufferString 开始。调用 NewRequest 时，Buffer 的当前长度会被记录到请求中；后续 Handler 从 req.Body 读走数据，不会让 req.ContentLength 跟着变成零。

package handler_test

import (
    "bytes"
    "io"
    "net/http"
    "net/http/httptest"
    "testing"
)

func TestCreateRequestBody(t *testing.T) {
    payload := `{"name":"demo"}`
    body := bytes.NewBufferString(payload)
    req := httptest.NewRequest(http.MethodPost, "/items", body)

    // ContentLength 记录构造时的字节数，Body 仍由 Handler 负责读取。
    if req.ContentLength != int64(len(payload)) {
        t.Fatalf("unexpected ContentLength: got %d", req.ContentLength)
    }
    got, err := io.ReadAll(req.Body)
    if err != nil {
        t.Fatalf("read request body: %v", err)
    }
    // 读取结果应与长度提示对应，避免测试体和断言各写一份数据。
    if string(got) != payload {
        t.Fatalf("unexpected body: got %q", got)
    }
}

如果测试要复用同一份内容，可以提前保留字符串或字节切片，再分别创建 Reader；不要在第一次读取后直接复用已经耗尽的 Buffer。strings.NewReader 和 bytes.NewReader 也会按“当前剩余长度”设置字段，Reader 在创建前若已读取过一部分，长度断言就应当反映剩余部分。

把 ContentLength 断言放进 Handler 回归测试

生产 Handler 通常更关心能否读到完整 JSON，而不是单独依赖长度字段。测试可以用一个小型 Handler 同时记录长度提示和解析结果，覆盖“长度正确但 Body 为空”“Body 有内容但长度未知”等边界。

func TestHandlerSeesRequestMetadata(t *testing.T) {
    body := strings.NewReader(`{"enabled":true}`)
    req := httptest.NewRequest(http.MethodPost, "/flags", body)
    rec := httptest.NewRecorder()

    // 读取一次请求体并检查长度提示，模拟 Handler 的真实入口。
    raw, err := io.ReadAll(req.Body)
    if err != nil {
        t.Fatalf("read body: %v", err)
    }
    if req.ContentLength != int64(len(raw)) {
        t.Fatalf("length mismatch: declared=%d actual=%d", req.ContentLength, len(raw))
    }
    rec.Header().Set("Content-Type", "application/json")
    rec.WriteHeader(http.StatusCreated)
    // Recorder 只保存响应，不改变请求体的长度语义。
    if rec.Code != http.StatusCreated {
        t.Fatalf("unexpected status: %d", rec.Code)
    }
}

上面的测试片段重点是断言位置：先读取 Body，再把读取结果和原始长度对比；如果 Handler 会根据 ContentLength 拒绝超大请求，还应加入普通 Reader 的未知长度用例，明确业务是允许流式读取还是必须先做上限控制。

用四组边界用例收紧测试语义

不要为了让断言通过而随意手动改写 ContentLength。只有当测试目标本身就是“客户端声明了错误长度”时，才应该显式设置错误值，并同时说明 Handler 预期返回什么状态码。正常请求体测试应让 NewRequest 根据 Reader 建立一致语义。

常见问题

为什么传入自定义 io.Reader 后 ContentLength 是 -1？

因为通用 Reader 没有可安全获取的总长度，httptest.NewRequest 会把长度标为未知。测试应验证 Handler 的流式处理或大小限制逻辑，而不是强行期待具体字节数。

读取 Body 后还能用 ContentLength 判断已读长度吗？

不能。ContentLength 记录的是请求构造时的长度提示；已读长度应根据读取到的字节数单独统计。

httptest.NewRequest 和 http.NewRequest 应该怎么选？

测试服务端 Handler 选 httptest.NewRequest；测试客户端发起请求、重定向或传输层行为时，才使用 http.NewRequest。