目前我們的後端服務提供大量的restful api介面，每次上線都需要測試那邊迴歸一遍這些介面，造成人力的浪費。正好藉著這次單元測試和持續整合，我們引入了httptest框架，結合gin來做介面單元測試。

httptest是golang官方提供的一個包，位於/src/net/http/httptest下。

其原理的話我也看了原始碼研究了下，這裡大致說下，它有一個ResponseRecorder結構體，它實現了http.ResponseWriter，同時它自身又包含了http.Response，前者是server端維度下的response，後者是client端維度下的response，也就是說，ResponseRecorder同時實現了server和client的功能。

好接下來看看如何使用httptest

httptest構建

 // Get 根據特定請求uri，發起get請求返回響應
func Get(uri string, router *gin.Engine) []byte {
    // 構造get請求
    req := httptest.NewRequest("GET", uri, nil)
    // 初始化響應
    w := httptest.NewRecorder()

    // 呼叫相應的handler介面
    router.ServeHTTP(w, req)

    // 提取響應
    result := w.Result()
    defer result
 
.Body.Close()

    // 讀取響應body
    body,_ := ioutil.ReadAll(result.Body)
    return body
}

// PostForm 根據特定請求uri和引數param，以表單形式傳遞引數，發起post請求返回響應
func PostForm(uri string, param url.Values, router *gin.Engine) []byte {

    // 構造post請求
    req := httptest.NewRequest("POST", uri, strings.NewReader(param.Encode()))
    req.Header.Set("Content-Type"
 
, "application/x-www-form-urlencoded")

    // 初始化響應
    w := httptest.NewRecorder()

    // 呼叫相應handler介面
    router.ServeHTTP(w, req)

    // 提取響應
    result := w.Result()
    defer result.Body.Close()

    // 讀取響應body
    body, _ := ioutil.ReadAll(result.Body)
    return body
}

在gin中註冊路由，並寫了兩個介面樣例

regAdmin := router.Group("/quote")
regAdmin.GET("/ge", Ge)
regAdmin.POST("/pos", Pos)


type TestForPost struct {
    Strname string  `form:"strname" binding:"required"`
    Number int  `form:"number" binding:"required"`
}


 // Get介面
func Ge(c *gin.Context)  {
    c.String(http.StatusOK, "success")
}


 // Post介面
func Pos(c *gin.Context) {
    req := &TestForPost{}
    if err := c.ShouldBindWith(req, binding.Form); err != nil {
        fmt.Println("bind error",err)
        c.JSON(http.StatusOK, gin.H{
            "errno":"1",
            "errmsg":"引數不匹配",
            "data":"",
        })
        return
    }
    c.JSON(http.StatusOK, gin.H{
        "errno":"0",
        "errmsg":"Null",
        "data":req,
    })
}

寫單元測試

type Response struct {
    Errno string `json:"errno"`
    Errmsg string `json:"errmsg"`
    Data TestForPost `json:"data"`
}

func TestOnGetStringRequest(t *testing.T) {

    // 初始化請求地址
    url:="/quote/ge"
    // 發起Get請求
    body := Get(url, router)

    // 判斷響應是否與預期一致
    if string(body) != "success" {
        t.Errorf("響應字串不符，body:%v\n",string(body))
    }

    convey.Convey("測試GET介面", t, func() {
        convey.So(string(body), convey.ShouldEqual, "success")
    })
}


func TestOnPostRequestForForm(t *testing.T) {
    // 初始化請求地址和請求引數
    uri := "/quote/pos"

    param := url.Values{
        "strname":{"test"},
        "number":{"1"},
    }
    // 發起post請求，以表單形式傳遞引數
    body := PostForm2(uri, param, router)
    //body := PostForm3(uri, param, router)

    // 解析響應，判斷響應是否與預期一致
    response := &Response{}
    if err := json.Unmarshal(body, response); err != nil {
        t.Errorf("解析響應出錯，err:%v\n",err)
    }
    t.Log(response.Data)
    if response.Data.Strname != "test" {
        t.Errorf("響應資料不符，errmsg:%v, data:%v\n",response.Errmsg,response.Data.Strname)
    }
    convey.Convey("測試POST介面", t, func() {
        convey.So(response.Data.Strname, convey.ShouldEqual, "test")
    })
}

完成