34 lines
822 B
Go
34 lines
822 B
Go
package httpx
|
|
|
|
import (
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func TestResponseHelpers(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
r := gin.New()
|
|
r.GET("/ok", OK)
|
|
r.GET("/fail", func(c *gin.Context) { Fail(c, "bad") })
|
|
r.GET("/auth", func(c *gin.Context) { NoAuth(c, "login") })
|
|
|
|
for _, tc := range []struct {
|
|
path string
|
|
statusCode int
|
|
body string
|
|
}{
|
|
{"/ok", 200, `{"code":0,"data":{},"msg":"操作成功"}`},
|
|
{"/fail", 200, `{"code":7,"data":{},"msg":"bad"}`},
|
|
{"/auth", 401, `{"code":7,"data":null,"msg":"login"}`},
|
|
} {
|
|
req := httptest.NewRequest("GET", tc.path, nil)
|
|
w := httptest.NewRecorder()
|
|
r.ServeHTTP(w, req)
|
|
if w.Code != tc.statusCode || w.Body.String() != tc.body {
|
|
t.Fatalf("%s: status=%d body=%q", tc.path, w.Code, w.Body.String())
|
|
}
|
|
}
|
|
}
|