Golang单元测试之httptest使用

概述现在有一个需求那就是，我们需要使用Golang的net/http包中的http.Get(url)方法去向服务器端请求数据，但是负责服务端的同事并没有将接口实现（可能是同事太忙，把妹，喝酒，扯淡， XO等等）以至于你只知道返回数据的json格式，然而无法请求到真实的数据，但是你的工作进度并不能因为同事而耽误，需要测试你的代码的正确性，那么怎么办？办法就是通过单元测试中的httptest，实现http

现在有一个需求那就是，我们需要使用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 personimport (	"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 personimport (	"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 personimport (	"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 personimport (	"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：[email protected]

注明出处：http://www.jb51.cc/article/p-pioxewyr-bpo.html

总结

以上是内存溢出为你收集整理的Golang单元测试之httptest使用全部内容，希望文章能够帮你解决Golang单元测试之httptest使用所遇到的程序开发问题。

如果觉得内存溢出网站内容还不错，欢迎将内存溢出网站推荐给程序员好友。