Go 单元测试怎么替换 HTTP 客户端而不访问网络
来源:17golang原创
时间:2026-09-06 09:32:59 165浏览 收藏
Go 单元测试里,如果被测函数直接使用默认的 http.Client,一次测试就可能真的发出 DNS、TCP 或 TLS 请求。更稳妥的做法是把客户端作为依赖传入,并在测试中替换它的 Transport。由于 http.Client 依赖的是 http.RoundTripper 接口,替身可以直接返回预设的 http.Response,整个测试不需要启动 HTTP 服务,也不会访问外网。
- 替换点是
http.Client.Transport,不是把生产 URL 改成测试地址。 - 自定义
RoundTripper要返回可关闭的Response.Body,并对请求做断言。 - 状态码属于响应结果,拿到响应时不要把非 2xx 自动当成 transport 错误。
为什么替换的是 RoundTripper 而不是把请求 URL 改成本地地址
http.Client 负责更高层的请求行为,真正执行一次 HTTP 事务的是 Transport。官方接口把这个能力抽象成 RoundTripper:给它一个已初始化的请求,返回响应或获取响应时发生的错误。默认实现会按需建立网络连接;测试则可以把同一接口换成内存中的函数。
因此生产代码只需要接收一个客户端,测试代码再构造带替身的客户端。这样请求方法、路径、查询参数和请求头仍然由业务代码决定,测试只控制“传输层返回什么”,不会因为改写 URL 而掩盖调用错误。

http.Client 只依赖 RoundTripper 接口,测试可以在传输边界放入 stubTransport。一个不会访问网络的测试替身怎么写
先给业务函数一个明确的客户端参数。下面的示例只读取响应状态,替身则保存收到的请求,返回一个内存响应:
package report
import (
"fmt"
"io"
"net/http"
)
type roundTripFunc func(*http.Request) (*http.Response, error)
func (f roundTripFunc) RoundTrip(req *http.Request) (*http.Response, error) {
// 由测试决定响应内容,这里不会拨号。
return f(req)
}
func fetchStatus(client *http.Client, endpoint string) (int, error) {
req, err := http.NewRequest(http.MethodGet, endpoint, nil)
if err != nil {
return 0, err
}
resp, err := client.Do(req)
if err != nil {
return 0, err
}
defer resp.Body.Close() // 业务代码负责关闭响应体。
return resp.StatusCode, nil
}
func describeStatus(code int) string {
return fmt.Sprintf("status=%d", code)
}
var _ http.RoundTripper = roundTripFunc(nil)
测试时把 roundTripFunc 放进 http.Client{Transport: ...}。返回响应时要同时提供 StatusCode 和 Body;即使示例只读取状态码,也应返回 io.NopCloser 包装的响应体,避免被测代码一旦开始读取正文就得到空指针。
func TestFetchStatusUsesStub(t *testing.T) {
var gotMethod, gotURL string
client := &http.Client{
Transport: roundTripFunc(func(req *http.Request) (*http.Response, error) {
// 断言请求到达替身,证明测试没有悄悄换掉调用路径。
gotMethod = req.Method
gotURL = req.URL.String()
return &http.Response{
StatusCode: http.StatusAccepted,
Body: io.NopCloser(strings.NewReader("queued")),
Header: make(http.Header),
Request: req,
}, nil
}),
}
code, err := fetchStatus(client, "https://api.example.test/reports")
if err != nil {
t.Fatal(err)
}
if code != http.StatusAccepted {
t.Fatalf("status = %d, want %d", code, http.StatusAccepted)
}
if gotMethod != http.MethodGet || gotURL != "https://api.example.test/reports" {
t.Fatalf("request = %s %s", gotMethod, gotURL)
}
}
这段测试还需要导入 strings 和 testing。示例把请求地址保留为业务真实地址,只是没有让默认 transport 接管它;这能同时验证 URL 拼写和网络隔离。生产代码中若要复用客户端,也应在创建后复用,而不是每个请求重新创建。
请求替身有哪些边界需要主动验证
替身越简单越好,但不能只返回一个 200 就结束。至少把下面几项分开检查:
| 检查对象 | 应验证什么 | 常见遗漏 |
|---|---|---|
| 请求 | 方法、URL、查询参数和关键请求头 | 测试通过但业务请求路径写错 |
| 响应 | 状态码、正文和错误分支 | 只覆盖成功分支,没处理 401/500 |
| 资源 | Response.Body 可读且最终关闭 | 返回 nil Body,或业务忘记 Close |
| 并发 | 多个测试调用时替身状态是否安全 | 共享切片或变量产生数据竞争 |

http.Response.Body 这一可关闭资源。如果要覆盖网络错误,让 RoundTrip 返回一个明确的 error;如果要覆盖服务端错误,则返回带有 StatusCode 的 response 和 nil error。两者含义不同:前者表示没有拿到响应,后者表示拿到了响应,只是业务状态不理想。
常见问题
Transport 不设置会发生什么?
http.Client 会使用默认 transport,测试可能真的访问网络。要隔离网络,显式传入带替身的客户端。
为什么返回 500 时不能同时返回普通 error?
HTTP 500 是已经取得的响应,应该让客户端拿到 response,再由业务层决定是否报错。transport error 留给 DNS、连接失败等“没有取得响应”的情况。
多个测试可以共用一个替身吗?
可以,但替身必须能并发安全地读取和写入状态。更简单的做法是每个测试创建自己的客户端,避免共享请求记录和响应队列。
把 HTTP 依赖注入到业务函数,再在测试中替换 http.RoundTripper,就能同时保留真实请求构造逻辑和稳定的内存响应。真正需要集成验证服务器路由、TLS 或代理行为时,再单独使用测试服务器或更高层测试,不要把这些成本混入每个单元测试。
-
860 收藏
-
843 收藏
-
826 收藏
-
809 收藏
-
792 收藏
-
258 收藏
-
358 收藏
-
438 收藏
-
113 收藏
-
398 收藏
-
278 收藏
-
381 收藏
-
429 收藏
-
128 收藏
-
450 收藏
-
250 收藏
-
402 收藏
-
- 前端进阶之JavaScript设计模式
- 设计模式是开发人员在软件开发过程中面临一般问题时的解决方案,代表了最佳的实践。本课程的主打内容包括JS常见设计模式以及具体应用场景,打造一站式知识长龙服务,适合有JS基础的同学学习。
- 立即学习 543次学习
-
- GO语言核心编程课程
- 本课程采用真实案例,全面具体可落地,从理论到实践,一步一步将GO核心编程技术、编程思想、底层实现融会贯通,使学习者贴近时代脉搏,做IT互联网时代的弄潮儿。
- 立即学习 516次学习
-
- 简单聊聊mysql8与网络通信
- 如有问题加微信:Le-studyg;在课程中,我们将首先介绍MySQL8的新特性,包括性能优化、安全增强、新数据类型等,帮助学生快速熟悉MySQL8的最新功能。接着,我们将深入解析MySQL的网络通信机制,包括协议、连接管理、数据传输等,让
- 立即学习 500次学习
-
- JavaScript正则表达式基础与实战
- 在任何一门编程语言中,正则表达式,都是一项重要的知识,它提供了高效的字符串匹配与捕获机制,可以极大的简化程序设计。
- 立即学习 487次学习
-
- 从零制作响应式网站—Grid布局
- 本系列教程将展示从零制作一个假想的网络科技公司官网,分为导航,轮播,关于我们,成功案例,服务流程,团队介绍,数据部分,公司动态,底部信息等内容区块。网站整体采用CSSGrid布局,支持响应式,有流畅过渡和展现动画。
- 立即学习 485次学习