36 Star 157 Fork 23

小蚂蚁 / gout

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
gout_test.go 1.98 KB
一键复制 编辑 Web IDE 原始数据 按行查看 历史
guonaihong 提交于 2020-12-21 21:15 . 新增NoAutoContentType接口 (#265)
package gout
import (
"fmt"
"github.com/gin-gonic/gin"
"github.com/guonaihong/gout/dataflow"
"github.com/stretchr/testify/assert"
"io/ioutil"
"net/http"
"net/http/httptest"
"sync/atomic"
"testing"
)
func TestDebug(t *testing.T) {
server := func() *gin.Engine {
r := gin.New()
r.GET("/", func(c *gin.Context) {
all, err := ioutil.ReadAll(c.Request.Body)
assert.NoError(t, err)
c.String(200, string(all))
})
return r
}()
ts := httptest.NewServer(http.HandlerFunc(server.ServeHTTP))
test := []func() DebugOpt{
// 没有颜色输出
NoColor,
Trace,
}
s := ""
for k, v := range test {
s = ""
err := GET(ts.URL).
Debug(v()).
SetBody(fmt.Sprintf("%d test debug.", k)).
BindBody(&s).
Do()
assert.NoError(t, err)
assert.Equal(t, fmt.Sprintf("%d test debug.", k), s)
}
}
func TestNew(t *testing.T) {
c := &http.Client{}
tests := []*dataflow.Gout{
New(nil),
New(),
New(c),
}
for _, v := range tests {
assert.NotNil(t, v)
}
}
func setupMethod(total *int32) *gin.Engine {
router := gin.Default()
cb := func(c *gin.Context) {
atomic.AddInt32(total, 1)
}
router.GET("/someGet", cb)
router.POST("/somePost", cb)
router.PUT("/somePut", cb)
router.DELETE("/someDelete", cb)
router.PATCH("/somePatch", cb)
router.HEAD("/someHead", cb)
router.OPTIONS("/someOptions", cb)
return router
}
func TestTopMethod(t *testing.T) {
var total int32
router := setupMethod(&total)
ts := httptest.NewServer(http.HandlerFunc(router.ServeHTTP))
defer ts.Close()
err := GET(ts.URL + "/someGet").Do()
assert.NoError(t, err)
err = POST(ts.URL + "/somePost").Do()
assert.NoError(t, err)
err = PUT(ts.URL + "/somePut").Do()
assert.NoError(t, err)
err = DELETE(ts.URL + "/someDelete").Do()
assert.NoError(t, err)
err = PATCH(ts.URL + "/somePatch").Do()
assert.NoError(t, err)
err = HEAD(ts.URL + "/someHead").Do()
assert.NoError(t, err)
err = OPTIONS(ts.URL + "/someOptions").Do()
assert.NoError(t, err)
assert.Equal(t, int(total), 7)
}
Go
1
https://gitee.com/guonaihong/gout.git
git@gitee.com:guonaihong/gout.git
guonaihong
gout
gout
master

搜索帮助

14c37bed 8189591 565d56ea 8189591