如何使用Go的httptest模拟多个不同的HTTP响应?
来源:stackoverflow
2024-03-17 13:30:29
0浏览
收藏
在 Go 语言中使用 httptest 模拟 HTTP 响应时,开发者可能需要模拟不同的响应内容。通过在 handlerResponse 函数中使用一个自定义的 responseWriter 结构,可以实现这一目的。这个结构包含一个映射表,其中键为请求次数,值为不同的响应内容。在处理程序函数中,每次收到请求时,responseWriter 会根据请求次数返回相应的响应内容。这样,在进行多次 HTTP 请求时,模拟的服务器可以提供不同的响应数据,满足测试用例的需求。
问题内容
我创建了一些 go 函数,这些函数可以对互联网上的服务进行 http get 调用并解析结果。
我现在正在为这些功能编写测试用例。
在我的测试用例中,我使用 go 包 httptest
来模拟对这些外部服务的调用。下面是我的代码。为了简洁起见,特意删除了错误检查。这是演示。
package main import ( "fmt" "io" "context" "net/http" "net/http/httptest" ) func handlerresponse() http.handler { return http.handlerfunc(func(w http.responsewriter, r *http.request) { w.writeheader(http.statusok) w.write([]byte(`{"a":"b"}`)) }) } func buildmyrequest(ctx context.context, url string) *http.request { request, _ := http.newrequestwithcontext(ctx, "get", url, nil) return request } func myprint(response *http.response) { b := make([]byte, 60000) for { _, err := response.body.read(b) if err == io.eof { break } } fmt.println(string(b)) } func main() { srv := httptest.newserver(handlerresponse()) client := http.client{} myresponse1, _ := client.do(buildmyrequest(context.background(), srv.url)) fmt.println("myresponse1:") myprint(myresponse1) myresponse2, _ := client.do(buildmyrequest(context.background(), srv.url)) fmt.println("myresponse2:") myprint(myresponse2) }
这是它产生的输出:
myResponse1: {"A":"B"} myResponse2: {"A":"B"}
如您所见,我创建了一些虚拟 http 响应数据 {"a":"b"}
,当您向 srv.url
发送 http 请求时,它实际上会命中一个临时 http 服务器,该服务器以虚拟数据。酷!
当您向 srv.url
发送第二个 http 请求时,它再次使用相同的虚拟数据进行响应。但这就是我的问题出现的地方。我希望临时 http 服务器在第二次 {"c":"d"}
和第三次 {"e":"f"}
收到请求时返回一些不同的数据。
如何更改 main()
函数的第一行,以便服务器在后续 http 调用中响应我所需的数据?
正确答案
你可以使用如下的 hack(演示:here)
package main import ( "fmt" "io" "context" "net/http" "net/http/httptest" "sync" ) type responseWriter struct{ resp map[int]string count int lock *sync.Mutex } func NewResponseWriter()*responseWriter{ r := new(responseWriter) r.lock = new(sync.Mutex) r.resp = map[int]string{ 0: `{"E":"F"}`, 1: `{"A":"B"}`, 2: `{"C":"D"}`, } r.count = 0 return r } func (r *responseWriter)GetResp()string{ r.lock.Lock() defer r.lock.Unlock() r.count ++ return r.resp[r.count%3] } func handlerResponse(rr *responseWriter) http.Handler { return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { w.WriteHeader(http.StatusOK) w.Write([]byte(rr.GetResp())) }) } func buildMyRequest(ctx context.Context, url string) *http.Request { request, _ := http.NewRequestWithContext(ctx, "GET", url, nil) return request } func myPrint(response *http.Response) { b := make([]byte, 60000) for { _, err := response.Body.Read(b) if err == io.EOF { break } } fmt.Println(string(b)) } func main() { rr := NewResponseWriter() srv := httptest.NewServer(handlerResponse(rr)) client := http.Client{} myResponse1, err := client.Do(buildMyRequest(context.Background(), srv.URL)) if err != nil{ fmt.Println(err) return } defer myResponse1.Body.Close() fmt.Println("myResponse1:") myPrint(myResponse1) myResponse2, err := client.Do(buildMyRequest(context.Background(), srv.URL)) if err != nil{ fmt.Println(err) return } defer myResponse2.Body.Close() fmt.Println("myResponse2:") myPrint(myResponse2) }
今天带大家了解了的相关知识,希望对你有所帮助;关于Golang的技术知识我们会一点点深入介绍,欢迎大家关注golang学习网公众号,一起学习编程~
版本声明
本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除

- 上一篇
- Win7以太网控制器找不到驱动程序怎么办?

- 下一篇
- Meta新增两大万卡集群,投入近50000块英伟达H100 GPU
查看更多
最新文章
-
- Golang · Go问答 | 1年前 |
- 在读取缓冲通道中的内容之前退出
- 139浏览 收藏
-
- Golang · Go问答 | 1年前 |
- 戈兰岛的全球 GOPRIVATE 设置
- 204浏览 收藏
-
- Golang · Go问答 | 1年前 |
- 如何将结构作为参数传递给 xml-rpc
- 325浏览 收藏
-
- Golang · Go问答 | 1年前 |
- 如何用golang获得小数点以下两位长度?
- 478浏览 收藏
-
- Golang · Go问答 | 1年前 |
- 如何通过 client-go 和 golang 检索 Kubernetes 指标
- 486浏览 收藏
-
- Golang · Go问答 | 1年前 |
- 将多个“参数”映射到单个可变参数的习惯用法
- 439浏览 收藏
-
- Golang · Go问答 | 1年前 |
- 将 HTTP 响应正文写入文件后出现 EOF 错误
- 357浏览 收藏
-
- Golang · Go问答 | 1年前 |
- 结构中映射的匿名列表的“复合文字中缺少类型”
- 352浏览 收藏
-
- Golang · Go问答 | 1年前 |
- NATS Jetstream 的性能
- 101浏览 收藏
-
- Golang · Go问答 | 1年前 |
- 如何将复杂的字符串输入转换为mapstring?
- 440浏览 收藏
-
- Golang · Go问答 | 1年前 |
- 相当于GoLang中Java将Object作为方法参数传递
- 212浏览 收藏
-
- Golang · Go问答 | 1年前 |
- 如何确保所有 goroutine 在没有 time.Sleep 的情况下终止?
- 143浏览 收藏
查看更多
课程推荐
-
- 前端进阶之JavaScript设计模式
- 设计模式是开发人员在软件开发过程中面临一般问题时的解决方案,代表了最佳的实践。本课程的主打内容包括JS常见设计模式以及具体应用场景,打造一站式知识长龙服务,适合有JS基础的同学学习。
- 542次学习
-
- GO语言核心编程课程
- 本课程采用真实案例,全面具体可落地,从理论到实践,一步一步将GO核心编程技术、编程思想、底层实现融会贯通,使学习者贴近时代脉搏,做IT互联网时代的弄潮儿。
- 509次学习
-
- 简单聊聊mysql8与网络通信
- 如有问题加微信:Le-studyg;在课程中,我们将首先介绍MySQL8的新特性,包括性能优化、安全增强、新数据类型等,帮助学生快速熟悉MySQL8的最新功能。接着,我们将深入解析MySQL的网络通信机制,包括协议、连接管理、数据传输等,让
- 497次学习
-
- JavaScript正则表达式基础与实战
- 在任何一门编程语言中,正则表达式,都是一项重要的知识,它提供了高效的字符串匹配与捕获机制,可以极大的简化程序设计。
- 487次学习
-
- 从零制作响应式网站—Grid布局
- 本系列教程将展示从零制作一个假想的网络科技公司官网,分为导航,轮播,关于我们,成功案例,服务流程,团队介绍,数据部分,公司动态,底部信息等内容区块。网站整体采用CSSGrid布局,支持响应式,有流畅过渡和展现动画。
- 484次学习
查看更多
AI推荐
-
- 边界AI平台
- 探索AI边界平台,领先的智能AI对话、写作与画图生成工具。高效便捷,满足多样化需求。立即体验!
- 386次使用
-
- 免费AI认证证书
- 科大讯飞AI大学堂推出免费大模型工程师认证,助力您掌握AI技能,提升职场竞争力。体系化学习,实战项目,权威认证,助您成为企业级大模型应用人才。
- 397次使用
-
- 茅茅虫AIGC检测
- 茅茅虫AIGC检测,湖南茅茅虫科技有限公司倾力打造,运用NLP技术精准识别AI生成文本,提供论文、专著等学术文本的AIGC检测服务。支持多种格式,生成可视化报告,保障您的学术诚信和内容质量。
- 537次使用
-
- 赛林匹克平台(Challympics)
- 探索赛林匹克平台Challympics,一个聚焦人工智能、算力算法、量子计算等前沿技术的赛事聚合平台。连接产学研用,助力科技创新与产业升级。
- 634次使用
-
- 笔格AIPPT
- SEO 笔格AIPPT是135编辑器推出的AI智能PPT制作平台,依托DeepSeek大模型,实现智能大纲生成、一键PPT生成、AI文字优化、图像生成等功能。免费试用,提升PPT制作效率,适用于商务演示、教育培训等多种场景。
- 541次使用
查看更多
相关文章
-
- GoLand调式动态执行代码
- 2023-01-13 502浏览
-
- 用Nginx反向代理部署go写的网站。
- 2023-01-17 502浏览
-
- Golang取得代码运行时间的问题
- 2023-02-24 501浏览
-
- 请问 go 代码如何实现在代码改动后不需要Ctrl+c,然后重新 go run *.go 文件?
- 2023-01-08 501浏览
-
- 如何从同一个 io.Reader 读取多次
- 2023-04-11 501浏览