golang单元测试：使用httptest和gock工具进行网络测试

Go单测从零到溜系列1—网络测试

这是Go语言单元测试从零到溜系列教程的第1篇，介绍了如何使用server编写单元测试。

假设我们的业务逻辑是搭建一个server端，对外提供HTTP服务。我们编写了一个helloHandler函数，用来处理用户请求。

// gin.gopackage ( "fmt" "net/ "github.com/gin-gonic/gin")// Param 请求参数type Param struct { Name string `json:"name"`}// helloHandler /hello请求处理函数func helloHandler(c *gin.Context) { var p Param if err := c.ShouldBindJSON(&p); err != nil { c.JSON(gin.H{ "msg": "we need a name", }) return } c.JSON(gin.H{ "msg": fmt.Sprintf("hello %s", p.Name), })}// SetupRouter 路由func SetupRouter() *gin.Engine { router := gin.Default() router.POST("/hello", helloHandler) return router}

现在我们需要为helloHandler函数编写单元测试，这种情况下我们就可以使用gin_test.gopackage ( "encoding/json" "net/ "net/ "strings" "testing" "github.com/stretchr/testify/assert")func Test_helloHandler(t *testing.T) { // 定义两个测试用例 tests := []struct { name string param string expect string }{ {"base case", `{"name": "liwenzhou"}`, "hello liwenzhou"}, {"bad case", "", "we need a name"}, } r := SetupRouter() for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { // mock一个HTTP请求 req := "POST", // 请求方法 "/hello", // 请求URL strings.NewReader(tt.param), // 请求参数 ) // mock一个响应记录器 w := // 让server端处理mock请求并记录返回的响应内容 r.ServeHTTP(w, req) // 校验状态码是否符合预期 assert.Equal(t, w.Code) // 解析并检验响应内容是否复合预期 var resp map[string]string err := json.Unmarshal([]byte(w.Body.String()), &resp) assert.Nil(t, err) assert.Equal(t, tt.expect, resp["msg"]) }) }}

执行单元测试，查看测试结果。

❯ go test -v=== RUN Test_helloHandler[GIN-debug] [WARNING] Creating an Engine instance with the Logger and Recovery middleware already attached.[GIN-debug] [WARNING] Running in "debug" mode. Switch to "release" mode in production. - using env: export GIN_MODE=release - using code: gin.SetMode(gin.ReleaseMode)[GIN-debug] POST /hello --> golang-unit-test-demo/(3 handlers)=== RUN Test_helloHandler/base_case[GIN] 2021/09/14 - 22:00:04 | 200 | 164.839µs | 192.0.2.1 | POST "/hello"=== RUN Test_helloHandler/bad_case[GIN] 2021/09/14 - 22:00:04 | 200 | 23.723µs | 192.0.2.1 | POST "/hello"--- PASS: Test_helloHandler (0.00s) --- PASS: Test_helloHandler/base_case (0.00s) --- PASS: Test_helloHandler/bad_case (0.00s)PASSok golang-unit-test-demo/ 0.055s

通过这个示例我们就掌握了如何使用Server服务中为请求处理函数编写单元测试了。

gock

上面的示例介绍了如何在HTTP Server服务类场景下为请求处理函数编写单元测试，那么如果我们是在代码中请求外部API的场景（比如通过API调用其他服务获取返回值）又该怎么编写单元测试呢？

例如，我们有以下业务逻辑代码，依赖外部API：api.go// ReqParam API请求参数type ReqParam struct { X int `json:"x"`}// Result API返回结果type Result struct { Value int `json:"value"`}func GetResultByAPI(x, y int) int { p := &ReqParam{X: x} b, _ := json.Marshal(p) // 调用其他服务的API resp, err := " "application/json", bytes.NewBuffer(b), ) if err != nil { return -1 } body, _ := ioutil.ReadAll(resp.Body) var ret Result if err := json.Unmarshal(body, &ret); err != nil { return -1 } // 这里是对API返回的数据做一些逻辑处理 return ret.Value + y}

在对类似上述这类业务代码编写单元测试的时候，如果不想在测试过程中真正去发送请求或者依赖的外部接口还没有开发完成时，我们可以在单元测试中对依赖的API进行mock。

这里推荐使用gock这个库。

安装

go get -u gopkg.in/h2non/gock.v1

使用示例

使用gock对外部API进行mock，即mock指定参数返回约定好的响应内容。下面的代码中mock了两组数据，组成了两个测试用例。

// api_test.gopackage gock_demoimport ( "testing" "github.com/stretchr/testify/assert" "gopkg.in/h2non/gock.v1")func TestGetResultByAPI(t *testing.T) { defer gock.Off() // 测试执行后刷新挂起的mock // mock 请求外部api时传参x=1返回100 gock.New(" Post("/post"). MatchType("json"). JSON(map[string]int{"x": 1}). Reply(200). JSON(map[string]int{"value": 100}) // 调用我们的业务函数 res := GetResultByAPI(1, 1) // 校验返回结果是否符合预期 assert.Equal(t, res, 101) // mock 请求外部api时传参x=2返回200 gock.New(" Post("/post"). MatchType("json"). JSON(map[string]int{"x": 2}). Reply(200). JSON(map[string]int{"value": 200}) // 调用我们的业务函数 res = GetResultByAPI(2, 2) // 校验返回结果是否符合预期 assert.Equal(t, res, 202) assert.True(t, gock.IsDone()) // 断言mock被触发}

执行上面写好的单元测试，看一下测试结果。

❯ go test -v=== RUN TestGetResultByAPI--- PASS: TestGetResultByAPI (0.00s)PASSok golang-unit-test-demo/gock_demo 0.054s

测试结果和预期的完全一致。

在这个示例中，为了让大家能够清晰的了解gock的使用，我特意没有使用表格驱动测试。给大家留一个小作业：自己动手把这个单元测试改写成表格驱动测试的风格，就当做是对最近两篇教程的复习和测验。

总结

在日常工作开发中为代码编写单元测试时如何处理外部依赖是最常见的问题，本文介绍了如何使用在下一篇中，我们将更进一步，详细介绍针对依赖MySQL和Redis的场景如何编写单元测试。

参考链接：

Go单测从零到溜系列1—网络测试 | 李文周的博客李文周的Blog Go语言测试单元测试 go test 执行测试，网络测试mock,如何编写单元测试,单元测试怎么写,go测试https://liwenzhou.com/posts/Go/golang-unit-test-1/