//?gin.go
package?httptest_demo
import?(
?"fmt"
?"net/http"
?"github.com/gin-gonic/gin"
)
//?Param?請(qǐng)求參數(shù)
type?Param?struct?{
?Name?string?`json:"name"`
}
//?helloHandler?/hello請(qǐng)求處理函數(shù)
func?helloHandler(c?*gin.Context)?{
?var?p?Param
?if?err?:=?c.ShouldBindJSON(&p);?err?!=?nil?{
??c.JSON(http.StatusOK,?gin.H{
???"msg":?"we?need?a?name",
??})
??return
?}
?c.JSON(http.StatusOK,?gin.H{
??"msg":?fmt.Sprintf("hello?%s",?p.Name),
?})
}
//?SetupRouter?路由
func?SetupRouter()?*gin.Engine?{
?router?:=?gin.Default()
?router.POST("/hello",?helloHandler)
?return?router
}

//?gin_test.go
package?httptest_demo
import?(
?"encoding/json"
?"net/http"
?"net/http/httptest"
?"strings"
?"testing"
?"github.com/stretchr/testify/assert"
)
func?Test_helloHandler(t?*testing.T)?{
?//?定義兩個(gè)測(cè)試用例
?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一個(gè)HTTP請(qǐng)求
???req?:=?httptest.NewRequest(
????"POST",??????????????????????//?請(qǐng)求方法
????"/hello",????????????????????//?請(qǐng)求URL
????strings.NewReader(tt.param),?//?請(qǐng)求參數(shù)
???)
???//?mock一個(gè)響應(yīng)記錄器
???w?:=?httptest.NewRecorder()
???//?讓server端處理mock請(qǐng)求并記錄返回的響應(yīng)內(nèi)容
???r.ServeHTTP(w,?req)
???//?校驗(yàn)狀態(tài)碼是否符合預(yù)期
???assert.Equal(t,?http.StatusOK,?w.Code)
???//?解析并檢驗(yàn)響應(yīng)內(nèi)容是否復(fù)合預(yù)期
???var?resp?map[string]string
???err?:=?json.Unmarshal([]byte(w.Body.String()),?&resp)
???assert.Nil(t,?err)
???assert.Equal(t,?tt.expect,?resp["msg"])
??})
?}
}

//?api.go
//?ReqParam?API請(qǐng)求參數(shù)
type?ReqParam?struct?{
?X?int?`json:"x"`
}
//?Result?API返回結(jié)果
type?Result?struct?{
?Value?int?`json:"value"`
}
func?GetResultByAPI(x,?y?int)?int?{
?p?:=?&ReqParam{X:?x}
?b,?_?:=?json.Marshal(p)
?//?調(diào)用其他服務(wù)的API
?resp,?err?:=?http.Post(
??"http://your-api.com/post",
??"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
?}
?//?這里是對(duì)API返回的數(shù)據(jù)做一些邏輯處理
?return?ret.Value?+?y
}

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

//?api_test.go
package?gock_demo
import?(
?"testing"
?"github.com/stretchr/testify/assert"
?"gopkg.in/h2non/gock.v1"
)
func?TestGetResultByAPI(t?*testing.T)?{
?defer?gock.Off()?//?測(cè)試執(zhí)行后刷新掛起的mock
?//?mock?請(qǐng)求外部api時(shí)傳參x=1返回100
?gock.New("http://your-api.com").
??Post("/post").
??MatchType("json").
??JSON(map[string]int{"x":?1}).
??Reply(200).
??JSON(map[string]int{"value":?100})
?//?調(diào)用我們的業(yè)務(wù)函數(shù)
?res?:=?GetResultByAPI(1,?1)
?//?校驗(yàn)返回結(jié)果是否符合預(yù)期
?assert.Equal(t,?res,?101)
?//?mock?請(qǐng)求外部api時(shí)傳參x=2返回200
?gock.New("http://your-api.com").
??Post("/post").
??MatchType("json").
??JSON(map[string]int{"x":?2}).
??Reply(200).
??JSON(map[string]int{"value":?200})
?//?調(diào)用我們的業(yè)務(wù)函數(shù)
?res?=?GetResultByAPI(2,?2)
?//?校驗(yàn)返回結(jié)果是否符合預(yù)期
?assert.Equal(t,?res,?202)
?assert.True(t,?gock.IsDone())?//?斷言mock被觸發(fā)
}