[feature] migrate to monorepo

backend/internal/handler/auth_handler_test.go

@@ -0,0 +1,276 @@
+package handler
+
+import (
+	"bytes"
+	"encoding/json"
+	"errors"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+	"tss-rocks-be/ent"
+	"tss-rocks-be/internal/config"
+	"tss-rocks-be/internal/service/mock"
+
+	"github.com/gin-gonic/gin"
+	"github.com/stretchr/testify/suite"
+	"go.uber.org/mock/gomock"
+)
+
+type AuthHandlerTestSuite struct {
+	suite.Suite
+	ctrl    *gomock.Controller
+	service *mock.MockService
+	handler *Handler
+	router  *gin.Engine
+}
+
+func (s *AuthHandlerTestSuite) SetupTest() {
+	s.ctrl = gomock.NewController(s.T())
+	s.service = mock.NewMockService(s.ctrl)
+	s.handler = NewHandler(&config.Config{
+		JWT: config.JWTConfig{
+			Secret: "test-secret",
+		},
+	}, s.service)
+	s.router = gin.New()
+}
+
+func (s *AuthHandlerTestSuite) TearDownTest() {
+	s.ctrl.Finish()
+}
+
+func TestAuthHandlerSuite(t *testing.T) {
+	suite.Run(t, new(AuthHandlerTestSuite))
+}
+
+func (s *AuthHandlerTestSuite) TestRegister() {
+	testCases := []struct {
+		name           string
+		request        RegisterRequest
+		setupMock      func()
+		expectedStatus int
+		expectedError  string
+	}{
+		{
+			name: "成功注册",
+			request: RegisterRequest{
+				Email:    "test@example.com",
+				Password: "password123",
+				Role:     "contributor",
+			},
+			setupMock: func() {
+				user := &ent.User{
+					ID:    1,
+					Email: "test@example.com",
+				}
+				s.service.EXPECT().
+					CreateUser(gomock.Any(), "test@example.com", "password123", "contributor").
+					Return(user, nil)
+				s.service.EXPECT().
+					GetUserRoles(gomock.Any(), user.ID).
+					Return([]*ent.Role{{ID: 1, Name: "contributor"}}, nil)
+			},
+			expectedStatus: http.StatusCreated,
+		},
+		{
+			name: "无效的邮箱格式",
+			request: RegisterRequest{
+				Email:    "invalid-email",
+				Password: "password123",
+				Role:     "contributor",
+			},
+			setupMock:      func() {},
+			expectedStatus: http.StatusBadRequest,
+			expectedError:  "Key: 'RegisterRequest.Email' Error:Field validation for 'Email' failed on the 'email' tag",
+		},
+		{
+			name: "密码太短",
+			request: RegisterRequest{
+				Email:    "test@example.com",
+				Password: "short",
+				Role:     "contributor",
+			},
+			setupMock:      func() {},
+			expectedStatus: http.StatusBadRequest,
+			expectedError:  "Key: 'RegisterRequest.Password' Error:Field validation for 'Password' failed on the 'min' tag",
+		},
+		{
+			name: "无效的角色",
+			request: RegisterRequest{
+				Email:    "test@example.com",
+				Password: "password123",
+				Role:     "invalid-role",
+			},
+			setupMock:      func() {},
+			expectedStatus: http.StatusBadRequest,
+			expectedError:  "Key: 'RegisterRequest.Role' Error:Field validation for 'Role' failed on the 'oneof' tag",
+		},
+	}
+
+	for _, tc := range testCases {
+		s.Run(tc.name, func() {
+			// 设置 mock
+			tc.setupMock()
+
+			// 创建请求
+			reqBody, _ := json.Marshal(tc.request)
+			req, _ := http.NewRequest(http.MethodPost, "/register", bytes.NewBuffer(reqBody))
+			req.Header.Set("Content-Type", "application/json")
+			w := httptest.NewRecorder()
+			c, _ := gin.CreateTestContext(w)
+			c.Request = req
+
+			// 执行请求
+			s.handler.Register(c)
+
+			// 验证响应
+			s.Equal(tc.expectedStatus, w.Code)
+			if tc.expectedError != "" {
+				var response map[string]string
+				err := json.Unmarshal(w.Body.Bytes(), &response)
+				s.NoError(err)
+				s.Contains(response["error"], tc.expectedError)
+			} else {
+				var response AuthResponse
+				err := json.Unmarshal(w.Body.Bytes(), &response)
+				s.NoError(err)
+				s.NotEmpty(response.Token)
+			}
+		})
+	}
+}
+
+func (s *AuthHandlerTestSuite) TestLogin() {
+	testCases := []struct {
+		name           string
+		request        LoginRequest
+		setupMock      func()
+		expectedStatus int
+		expectedError  string
+	}{
+		{
+			name: "成功登录",
+			request: LoginRequest{
+				Email:    "test@example.com",
+				Password: "password123",
+			},
+			setupMock: func() {
+				user := &ent.User{
+					ID:    1,
+					Email: "test@example.com",
+				}
+				s.service.EXPECT().
+					GetUserByEmail(gomock.Any(), "test@example.com").
+					Return(user, nil)
+				s.service.EXPECT().
+					ValidatePassword(gomock.Any(), user, "password123").
+					Return(true)
+				s.service.EXPECT().
+					GetUserRoles(gomock.Any(), user.ID).
+					Return([]*ent.Role{{ID: 1, Name: "contributor"}}, nil)
+			},
+			expectedStatus: http.StatusOK,
+		},
+		{
+			name: "无效的邮箱格式",
+			request: LoginRequest{
+				Email:    "invalid-email",
+				Password: "password123",
+			},
+			setupMock:      func() {},
+			expectedStatus: http.StatusBadRequest,
+			expectedError:  "Key: 'LoginRequest.Email' Error:Field validation for 'Email' failed on the 'email' tag",
+		},
+		{
+			name: "用户不存在",
+			request: LoginRequest{
+				Email:    "nonexistent@example.com",
+				Password: "password123",
+			},
+			setupMock: func() {
+				s.service.EXPECT().
+					GetUserByEmail(gomock.Any(), "nonexistent@example.com").
+					Return(nil, errors.New("user not found"))
+			},
+			expectedStatus: http.StatusUnauthorized,
+			expectedError:  "Invalid credentials",
+		},
+		{
+			name: "密码错误",
+			request: LoginRequest{
+				Email:    "test@example.com",
+				Password: "wrong-password",
+			},
+			setupMock: func() {
+				user := &ent.User{
+					ID:    1,
+					Email: "test@example.com",
+				}
+				s.service.EXPECT().
+					GetUserByEmail(gomock.Any(), "test@example.com").
+					Return(user, nil)
+				s.service.EXPECT().
+					ValidatePassword(gomock.Any(), user, "wrong-password").
+					Return(false)
+			},
+			expectedStatus: http.StatusUnauthorized,
+			expectedError:  "Invalid credentials",
+		},
+		{
+			name: "获取用户角色失败",
+			request: LoginRequest{
+				Email:    "test@example.com",
+				Password: "password123",
+			},
+			setupMock: func() {
+				user := &ent.User{
+					ID:    1,
+					Email: "test@example.com",
+				}
+				s.service.EXPECT().
+					GetUserByEmail(gomock.Any(), "test@example.com").
+					Return(user, nil)
+				s.service.EXPECT().
+					ValidatePassword(gomock.Any(), user, "password123").
+					Return(true)
+				s.service.EXPECT().
+					GetUserRoles(gomock.Any(), user.ID).
+					Return(nil, errors.New("failed to get roles"))
+			},
+			expectedStatus: http.StatusInternalServerError,
+			expectedError:  "Failed to get user roles",
+		},
+	}
+
+	for _, tc := range testCases {
+		s.Run(tc.name, func() {
+			// 设置 mock
+			tc.setupMock()
+
+			// 创建请求
+			reqBody, _ := json.Marshal(tc.request)
+			req, _ := http.NewRequest(http.MethodPost, "/login", bytes.NewBuffer(reqBody))
+			req.Header.Set("Content-Type", "application/json")
+			w := httptest.NewRecorder()
+			c, _ := gin.CreateTestContext(w)
+			c.Request = req
+
+			// 执行请求
+			s.handler.Login(c)
+
+			// 验证响应
+			s.Equal(tc.expectedStatus, w.Code)
+			if tc.expectedError != "" {
+				var response map[string]string
+				err := json.Unmarshal(w.Body.Bytes(), &response)
+				s.NoError(err)
+				s.Contains(response["error"], tc.expectedError)
+			} else {
+				var response AuthResponse
+				err := json.Unmarshal(w.Body.Bytes(), &response)
+				s.NoError(err)
+				s.NotEmpty(response.Token)
+			}
+		})
+	}
+}