大佬教程收集整理的这篇文章主要介绍了Golang单元测试之httptest使用,大佬教程大佬觉得挺不错的,现在分享给大家,也给大家做个参考。
现在有一个需求那就是,我们需要使用Golang的net/Http包中的http.Get(url)方法去向服务器端请求数据,但是负责服务端的同事并没有将接口实现(可能是同事太忙,把妹,喝酒,扯淡, XO等等)以至于你只知道返回数据的json格式,然而无法请求到真实的数据,但是你的工作进度并不能因为同事而耽误,需要测试你的代码的正确性,那么怎么办?办法就是通过单元测试中的httptest,实现http server,并设置好返回值,那么http.Get(url)的请求就会直接打到单元测试的http server上,同时得到你设置好的返回值,你就可以继续去处理数据,测试你的代码逻辑了。
一、导入
设置一个场景:
需要向服务器端去获取住址为"shanghai"的所有人的信息,请求函数为:
func GeTinfo(api String) ([]Person,error)这个方法包括:
1.向服务端发送get请求获取数据
2.将数据序列化person的信息
3.如果出现错误,返回error
需要测试的内容:
1.需要测试返回值是我们设置的值:
var personResponse = []Person{ { Name : "wahaha",Address : "shanghai",Age : 20,},{ Name : "lebaishi",Age : 10,}2.需要测试GeTinfo方法是发送了正确的http请求,及method是否正确,路径是否正确,请求参数是否正确。
3.返回状态码是否正确。
二、Golang tesTing基础
Go提供了一个tesTing包来写单元测试。假设我们有个文件叫person.go,那么我们的测试文件就需要明明为person_test.go:
package person import ( "tesTing" ) func TestPublishUnreachable(t *tesTing.T) { api := "http://localhost:8090" _,err := GeTinfo(api) if err != nil { t.Errorf("GeTinfo() return an error") } }注:
>测试函数以Test*开头。
>测试函数将*tesTing.T作为参数,可以在失败的情况下使用Errorf()方法。
>在包内使用go test来运行单元测试。
这个单元测试将会失败,因为我们还没有实现GeTinfo()方法,下面是person.go文件:
package person import ( "net/http" "fmt" ) const ( ADDRESS = "shanghai" ) type Person struct { Name String `json:"name"` Address String `json:"address"` AgE int `json:"age"` } func GeTinfo(api String) ([]Person,error) { url := fmt.Sprintf("%s/person?addr=%s",api,ADDRESS) resp,err := http.Get(url) if err != nil { return []Person{},err } if resp.StatusCode != http.StatusOK { return []Person{},fmt.Errorf("get info didn’t respond 200 OK: %s",resp.Status) } return nil,nil }当然运行go test也会返回一个错误,因为请求地址的问题,请求的并不是一个实际上的http server,那么自然也不会有正常的返回。
三、Golang httptest
上面一个例子很有用,但是如何去发送一个真正的http request而不去真正的启动一个http server(亦或者请求任意的server)?答案是使用Go 的httptest包,这个包可以非常简单的创建一个测试的http server,那么下面我们将展示一下完整的代码,并解释一下整体的测试流程:
person.go:
package person import ( "net/http" "fmt" "io/IoUtil" "encoding/json" "github.com/astaxie/beego/logs" ) const ( ADDRESS = "shanghai" ) type Person struct { Name String `json:"name"` Address String `json:"address"` AgE int `json:"age"` } func GeTinfo(api String) ([]Person,err := http.Get(url) defer resp.body.Close() if err != nil { return []Person{},resp.Status) } bodyBytes,_ := IoUtil.ReadAll(resp.body) personList := make([]Person,0) err = json.Unmarshal(bodyBytes,&personList) if err != nil { logs.Error("decode data fail") return []Person{},fmt.Errorf("decode data fail") } return personList,nil }
person_test.go:
package person import ( "tesTing" "net/http/httptest" "net/http" "fmt" "encoding/json" ) var personResponse = []Person{ { Name : "wahaha",} var personResponseBytes,_ = json.Marshal(personResponsE) func TestPublishWrongResponseStatus(t *tesTing.T) { ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter,r *http.request) { w.WriteHeader(http.StatusOK) w.Write(personResponseBytes) if r.Method != "GET"{ t.Errorf("Expected 'GET' request,got '%s'",r.Method) } if r.URl.EscapedPath() != "/person" { t.Errorf("Expected request to '/person',r.URl.EscapedPath()) } r.ParseForm() topic := r.Form.Get("addr") if topic != "shanghai" { t.Errorf("Expected request to have 'addr=shanghai',got: '%s'",topiC) } })) defer ts.Close() api := tS.URL fmt.Println("url:",api) resp,_ := GeTinfo(api) fmt.Println("reps:",resp) }解释一下:
>我们通过httptest.NewServer创建了一个测试的http server
>读请求设置通过变量r *http.request,写变量(也就是返回值)通过w http.ResponseWriter
>通过tS.URL来获取请求的URL(一般都是<http://ip:port>)
>通过r.Method来获取请求的方法,来测试判断我们的请求方法是否正确
>获取请求路径:r.URl.EscapedPath(),本例中的请求路径就是"/person"
>获取请求参数:r.ParseForm,r.Form.Get("addr")
>设置返回的状态码:w.WriteHeader(http.StatusOK)
>设置返回的内容(这就是我们想要的结果):w.Write(personResponseBytes),注意w.Write()接收的参数是[]byte,因此需要将object对象列表通过json.Marshal(personResponsE)转换成字节。
综上,我们可以通过不发送httptest来模拟出httpserver和返回值来进行自己代码的测试了。
Author:忆之独秀
Email:leaguenew@qq.com
注明出处:http://code.js-code.com/article/p-pioxewyr-bpo.html
以上是大佬教程为你收集整理的Golang单元测试之httptest使用全部内容,希望文章能够帮你解决Golang单元测试之httptest使用所遇到的程序开发问题。
如果觉得大佬教程网站内容还不错,欢迎将大佬教程推荐给程序员好友。
本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。
如您有任何意见或建议可联系处理。小编QQ:384754419,请注明来意。