优化结构

pkg/httpx/response_test.go

@@ -0,0 +1,33 @@
+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())
+		}
+	}
+}