更新了一版模板

test.go

 package main
 
 import (
-	"fmt"
-	"io"
+	"errors"
+	"go-svc-tpl/api/dto"
+	"go-svc-tpl/internal/controller"
+	"go-svc-tpl/internal/dao/model"
 	"net/http"
+	"net/http/httptest"
+	"strings"
 	"testing"
-	"time"
-)
 
-func TestMyController(t *testing.T) {
-	// 创建一个新的HTTP服务器，用于处理请求
-	mux := http.NewServeMux()
-	mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
-		fmt.Fprintf(w, "Hello, World!")
-	})
-	server := &http.Server{Addr: ":8080", Handler: mux}
+	"github.com/gin-gonic/gin"
+	"github.com/stretchr/testify/assert"
+	"gorm.io/gorm"
+)
 
-	// 启动HTTP服务器
-	go func() {
-		if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
-			t.Errorf("Error starting server: %v", err)
+func TestUserController_Register(t *testing.T) {
+	testCases := []struct {
+		desc     string
+		req      *dto.UserRegisterReq
+		expected error
+	}{
+		{
+			desc: "success",
+			req: &dto.UserRegisterReq{
+				Email:    "test@test.com",
+				Password: "password",
+				Name:     "test",
+			},
+			expected: nil,
+		},
+		{
+			desc: "invalid email",
+			req: &dto.UserRegisterReq{
+				Email:    "test@test",
+				Password: "password",
+				Name:     "test",
+			},
+			expected: errors.New("wrong email pattern"),
+		},
 	}
-	}()
 
-	// 等待服务器启动
-	time.Sleep(1 * time.Second)
+	for _, tC := range testCases {
+		t.Run(tC.desc, func(t *testing.T) {
+			gin.SetMode(gin.TestMode)
+			r := gin.New()
+			r.Use(func(c *gin.Context) {
+				c.Set(controller.USER_ID_KEY, uint(1))
+			})
 
-	// 创建一个HTTP客户端
-	client := &http.Client{
-		Timeout: 5 * time.Second,
+			ctrl := &controller.UserController{
+				db: &gorm.DB{},
 			}
+			_ = model.Migrate(ctrl.db)
+			ctrl.db = model.MockDB()
 
-	// 发送一个GET请求到服务器
-	resp, err := client.Get("http://localhost:8080")
-	if err != nil {
-		t.Errorf("Error sending request: %v", err)
-		return
-	}
-	defer resp.Body.Close()
+			var resp *httptest.ResponseRecorder
+			if data, err := tC.req.ToJSON(); err != nil {
+				t.Fatalf("failed to marshal request to JSON: %v", err)
+			} else {
+				req, _ := http.NewRequest(http.MethodPost, "/register", strings.NewReader(data))
 
-	// 发送一个POST请求到服务器
-	resp, err = client.Post("http://localhost:8080", "application/json", nil)
-	if err != nil {
-		t.Errorf("Error sending request: %v", err)
-		return
+				resp = httptest.NewRecorder()
+				r.POST("/register", func(c *gin.Context) {
+					if err := ctrl.Register(c, tC.req); err != nil {
+						c.AbortWithStatus(http.StatusInternalServerError)
 					}
-	defer resp.Body.Close()
-
-	// 检查响应是否成功
-	if resp.StatusCode != http.StatusOK {
-		t.Errorf("Unexpected status code: %d", resp.StatusCode)
-		return
-	}
-
-	// 读取响应内容
-	body, err := io.ReadAll(resp.Body)
-	if err != nil {
-		t.Errorf("Error reading response body: %v", err)
-		return
+					c.AbortWithStatus(http.StatusOK)
+				})
+				r.ServeHTTP(resp, req)
 			}
 
-	// 检查响应内容是否正确
-	expected := "Hello, World!"
-	if string(body) != expected {
-		t.Errorf("Unexpected response body: %s", string(body))
-		return
+			assert.Equal(t, http.StatusOK, resp.Code)
+			assert.Equal(t, tC.expected, nil)
+		})
 	}
-
-	// 关闭HTTP服务器
-	server.Close()
 }