当前位置:首页 > 文章列表 > Golang > Go教程 > Go 单元测试怎么替换 HTTP 客户端而不访问网络

Go 单元测试怎么替换 HTTP 客户端而不访问网络

来源:17golang原创 2026-09-06 09:32:59 0浏览 收藏

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 而掩盖调用错误。

Go 单元测试中业务函数、http.Client、http.RoundTripper、stubTransport 与 http.Response 的调用边界关系图
图1: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: ...}。返回响应时要同时提供 StatusCodeBody;即使示例只读取状态码,也应返回 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)
    }
}

这段测试还需要导入 stringstesting。示例把请求地址保留为业务真实地址,只是没有让默认 transport 接管它;这能同时验证 URL 拼写和网络隔离。生产代码中若要复用客户端,也应在创建后复用,而不是每个请求重新创建。

请求替身有哪些边界需要主动验证

替身越简单越好,但不能只返回一个 200 就结束。至少把下面几项分开检查:

检查对象应验证什么常见遗漏
请求方法、URL、查询参数和关键请求头测试通过但业务请求路径写错
响应状态码、正文和错误分支只覆盖成功分支,没处理 401/500
资源Response.Body 可读且最终关闭返回 nil Body,或业务忘记 Close
并发多个测试调用时替身状态是否安全共享切片或变量产生数据竞争
Go HTTP 单元测试中 Test 函数、请求构造、传输替身、响应体与请求响应断言的边界关系图
图2:测试既要检查请求是否正确,也要处理 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 或代理行为时,再单独使用测试服务器或更高层测试,不要把这些成本混入每个单元测试。

版本声明
本文转载于:17golang原创 如有侵犯,请联系study_golang@163.com删除
CSS Flex 子元素撑破容器怎么处理CSS Flex 子元素撑破容器怎么处理
上一篇
CSS Flex 子元素撑破容器怎么处理
venera漫画为什么搜不到作品?漫画源依赖、网络排查与下载边界说明
下一篇
venera漫画为什么搜不到作品?漫画源依赖、网络排查与下载边界说明
查看更多
最新文章
查看更多
课程推荐
  • 前端进阶之JavaScript设计模式
    前端进阶之JavaScript设计模式
    设计模式是开发人员在软件开发过程中面临一般问题时的解决方案,代表了最佳的实践。本课程的主打内容包括JS常见设计模式以及具体应用场景,打造一站式知识长龙服务,适合有JS基础的同学学习。
    543次学习
  • GO语言核心编程课程
    GO语言核心编程课程
    本课程采用真实案例,全面具体可落地,从理论到实践,一步一步将GO核心编程技术、编程思想、底层实现融会贯通,使学习者贴近时代脉搏,做IT互联网时代的弄潮儿。
    516次学习
  • 简单聊聊mysql8与网络通信
    简单聊聊mysql8与网络通信
    如有问题加微信:Le-studyg;在课程中,我们将首先介绍MySQL8的新特性,包括性能优化、安全增强、新数据类型等,帮助学生快速熟悉MySQL8的最新功能。接着,我们将深入解析MySQL的网络通信机制,包括协议、连接管理、数据传输等,让
    500次学习
  • JavaScript正则表达式基础与实战
    JavaScript正则表达式基础与实战
    在任何一门编程语言中,正则表达式,都是一项重要的知识,它提供了高效的字符串匹配与捕获机制,可以极大的简化程序设计。
    487次学习
  • 从零制作响应式网站—Grid布局
    从零制作响应式网站—Grid布局
    本系列教程将展示从零制作一个假想的网络科技公司官网,分为导航,轮播,关于我们,成功案例,服务流程,团队介绍,数据部分,公司动态,底部信息等内容区块。网站整体采用CSSGrid布局,支持响应式,有流畅过渡和展现动画。
    485次学习
查看更多
AI推荐
  • SuperCLUE中文大模型评测基准:功能、能力维度与应用指南
    SuperCLUE
    SuperCLUE是权威的中文大语言模型综合评测基准,涵盖语言理解、知识应用、AI Agent智能体及安全性等12项核心能力。通过多轮对话与客观测试,定期发布榜单与技术报告,为模型研发、优化及行业选型提供科学依据。
    162次使用
  • C-Eval中文评测基准:大语言模型多学科能力评估指南
    C-Eval
    深入了解C-Eval中文评估套件,涵盖52个学科与4级难度。本文详解其功能特点、Zero-shot/Few-shot使用方法及代码示例,助您全面评测LLM中文理解与泛化能力。
    88次使用
  • LangGPT提示词框架:结构化Prompt设计方法与开源工具指南
    LangGPT
    LangGPT是一种受编程语言启发的结构化提示词设计工具,提供双层框架、模块化模板及变量功能,帮助用户高效编写高质量Prompt。该项目已在GitHub免费开源,适用于内容创作、编程辅助等多场景。
    7次使用
  • ClickPrompt:AI提示词生成与优化工具,支持Stable Diffusion、ChatGPT及代码辅助
    ClickPrompt
    ClickPrompt是一款专为AI提示词编写者设计的开源在线工具,支持Stable Diffusion绘图、ChatGPT对话及GitHub Copilot代码辅助。提供Prompt自动生成、一键运行、社区分享及可视化优化功能,帮助用户高效获取精准AI输出。
    49次使用
  • PromptHero官网:AI提示词搜索、优化与学习平台,支持Midjourney/Stable Diffusion
    PromptHero
    PromptHero是专业的AI提示词搜索引擎与优化平台,支持Stable Diffusion、Midjourney等主流模型。提供海量提示词库、分类搜索、在线课程及社区互动,助力用户高效生成高质量AI图像与文本。
    32次使用
微信登录更方便
  • 密码登录
  • 注册账号
登录即同意 用户协议隐私政策
返回登录
  • 重置密码