common controller tester

xcommon/controller.go

 package xcommon
 
 import (
+	"context"
 	"net/http"
+	"testing"
+	"time"
 
 	"github.com/gin-gonic/gin"
+	"gitlab.wanzhuangkj.com/tush/xpkg/gin/response"
+	"gitlab.wanzhuangkj.com/tush/xpkg/httpcli"
+	"gitlab.wanzhuangkj.com/tush/xpkg/utils/mapUtils"
 	"gitlab.wanzhuangkj.com/tush/xpkg/xcommon/validator"
 )
 
@@ -47,13 +53,56 @@ func (e *Controller) Bind(c *gin.Context, req interface{}) error {
 	return nil
 }
 
-//
-//func (x *Controller) GetUser(ctx context.Context) *biz.User {
-//	val := ctx.Value(ctxUtils.KeyUser)
-//	if val != nil {
-//		if u, ok := val.(*biz.User); ok {
-//			return u
-//		}
-//	}
-//	return nil
-//}
+type ControllerTester struct {
+	Headers map[string]string
+	Addr    string
+	client  *httpcli.Client
+}
+
+func NewControllerTester(host string, header map[string]string) *ControllerTester {
+	c := ControllerTester{
+		Headers: header,
+		Addr:    host,
+		client: httpcli.NewClient(&http.Client{
+			Transport: &http.Transport{
+				MaxIdleConns:        100,
+				MaxIdleConnsPerHost: 100,
+				IdleConnTimeout:     90 * time.Second,
+			},
+			Timeout: 30 * time.Second,
+		}),
+	}
+	return &c
+}
+
+func (x *ControllerTester) Before(t *testing.T) {
+	// if Headers["Authorization"] != "" {
+	// 	return
+	// }
+	// loginReply, err := Login()
+	// if err != nil {
+	// 	t.Error(err)
+	// }
+	// UserID = loginReply.UserID
+	// Authorization = "Bearer " + loginReply.Token
+	// Headers["Authorization"] = Authorization
+}
+
+func (x *ControllerTester) After(t *testing.T) {
+
+}
+
+func (x *ControllerTester) CommonHttp(method, URL string, in any, reply *response.Result) error {
+	if method == http.MethodDelete || method == http.MethodGet {
+		in = mapUtils.StructToMapWithJsonTag(in)
+	}
+	if err := x.client.NewRequest().
+		SetHeaders(x.Headers).
+		SetURL(URL).SetContentType("application/json").
+		Do(context.Background(), method, in).
+		BindJSON(&reply).
+		Err(); err != nil {
+		return err
+	}
+	return nil
+}