如何使用Golang测试HTTP客户端请求

使用 httptest 创建模拟服务器可避免真实网络调用，通过预设响应验证客户端请求构造与处理。示例中启动测试服务器并校验请求路径、返回状态码及 body，支持检查头信息如 Authorization、POST 体内容，并可模拟 404、500、超时等场景，确保客户端错误处理正确。测试后需关闭服务器资源，推荐封装为辅助函数复用。

测试 HTTP 客户端请求在 Golang 中是常见需求，关键在于避免真实网络调用，同时验证请求构造和响应处理的正确性。核心思路是使用 httptest 包创建模拟服务器，替换客户端的请求目标。

使用 httptest 创建模拟服务器

通过 net/http/httptest 启动一个本地测试服务器，它能接收请求并返回预设响应。这样客户端代码无需修改，仍调用 http.Do 或 http.Get，但目标地址指向测试服务器。

基本步骤：

• 调用 httptest.NewServer 创建服务器，传入自定义的 http.HandlerFunc
• 在 handler 中验证请求方法、路径、头、体等是否符合预期
• 写入预设的响应状态码和 body
• 测试结束后调用 server.Close() 释放资源

func TestHTTPClient(t *testing.T) {
    server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        if r.URL.Path != "/api/data" {
            t.Errorf("期望路径 /api/data，实际: %s", r.URL.Path)
        }
        w.WriteHeader(http.StatusOK)
        w.Write([]byte(`{"status": "ok"}`))
    }))
    defer server.Close()

    // 假设 client 是你的 HTTP 客户端，将 endpoint 指向 server.URL
    resp, err := client.Get(server.URL + "/api/data")
    if err != nil {
        t.Fatal(err)
    }
    defer resp.Body.Close()

    // 验证响应
    if resp.StatusCode != http.StatusOK {
        t.Errorf("期望状态 200，实际: %d", resp.StatusCode)
    }
}

验证请求头和请求体

测试中常需确认客户端是否正确设置了认证头、Content-Type 等。也可以检查 POST 请求的 body 内容。

立即学习“go语言免费学习笔记（深入）”；

白瓜面试

白瓜面试 - AI面试助手,辅助笔试面试神器

40

查看详情

在测试 handler 中读取并解析请求内容：

• 使用 r.Header.Get("Authorization") 检查头信息
• 对 POST 请求，用 ioutil.ReadAll(r.Body) 读取 body 并做比对
• 可结合 json.Unmarshal 验证 JSON 结构

auth := r.Header.Get("Authorization")
if auth != "Bearer token123" {
    t.Errorf("缺少或错误的 Authorization 头")
}

模拟不同响应场景

真实环境中，服务可能返回 404、500、超时等。测试应覆盖这些情况以验证客户端的错误处理逻辑。

做法是在不同测试用例中让 mock server 返回不同状态：

• 返回 404：设置 w.WriteHeader(http.StatusNotFound)
• 返回 500：设置 w.WriteHeader(http.StatusInternalServerError)
• 模拟超时：在 handler 中延迟写入，或关闭连接
• 返回无效 JSON：写入格式错误的字符串，测试解析失败处理

以上就是如何使用Golang测试HTTP客户端请求的详细内容，更多请关注php中文网其它相关文章！