first commit

backend/test/user_handler_test.go

@@ -0,0 +1,177 @@
+package test
+
+import (
+	"bytes"
+	"encoding/json"
+	"errors"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	"github.com/gin-gonic/gin"
+	"github.com/rovina/zsp-backend/internal/dto"
+	"github.com/rovina/zsp-backend/internal/handler"
+)
+
+// mockUserService implements service.UserService for testing
+type mockUserService struct {
+	registerFn   func(req *dto.CreateUserRequest) (*dto.UserResponse, error)
+	loginFn      func(req *dto.LoginRequest) (*dto.UserResponse, string, error)
+	getUserByIDFn func(id uint) (*dto.UserResponse, error)
+}
+
+func (m *mockUserService) Register(req *dto.CreateUserRequest) (*dto.UserResponse, error) {
+	return m.registerFn(req)
+}
+func (m *mockUserService) Login(req *dto.LoginRequest) (*dto.UserResponse, string, error) {
+	return m.loginFn(req)
+}
+func (m *mockUserService) GetUserByID(id uint) (*dto.UserResponse, error) {
+	return m.getUserByIDFn(id)
+}
+
+func setupUserRouter(h *handler.UserHandler) *gin.Engine {
+	gin.SetMode(gin.TestMode)
+	r := gin.New()
+	api := r.Group("/api")
+	{
+		auth := api.Group("/auth")
+		{
+			auth.POST("/register", h.Register)
+			auth.POST("/login", h.Login)
+		}
+		users := api.Group("/users")
+		{
+			users.GET("/:id", h.GetUser)
+		}
+	}
+	return r
+}
+
+func TestUserRegister_Success(t *testing.T) {
+	mock := &mockUserService{
+		registerFn: func(req *dto.CreateUserRequest) (*dto.UserResponse, error) {
+			return &dto.UserResponse{ID: 1, Username: "testuser", Email: "test@example.com"}, nil
+		},
+	}
+	h := handler.NewUserHandler(mock)
+	router := setupUserRouter(h)
+
+	body, _ := json.Marshal(dto.CreateUserRequest{Username: "testuser", Email: "test@example.com", Password: "password123"})
+	w := httptest.NewRecorder()
+	req, _ := http.NewRequest("POST", "/api/auth/register", bytes.NewReader(body))
+	req.Header.Set("Content-Type", "application/json")
+	router.ServeHTTP(w, req)
+
+	if w.Code != http.StatusCreated {
+		t.Errorf("expected status 201, got %d", w.Code)
+	}
+
+	var resp map[string]interface{}
+	json.Unmarshal(w.Body.Bytes(), &resp)
+	if resp["message"] != "User registered successfully" {
+		t.Errorf("unexpected message: %v", resp["message"])
+	}
+}
+
+func TestUserRegister_DuplicateEmail(t *testing.T) {
+	mock := &mockUserService{
+		registerFn: func(req *dto.CreateUserRequest) (*dto.UserResponse, error) {
+			return nil, errors.New("email already exists")
+		},
+	}
+	h := handler.NewUserHandler(mock)
+	router := setupUserRouter(h)
+
+	body, _ := json.Marshal(dto.CreateUserRequest{Username: "testuser", Email: "existing@example.com", Password: "password123"})
+	w := httptest.NewRecorder()
+	req, _ := http.NewRequest("POST", "/api/auth/register", bytes.NewReader(body))
+	req.Header.Set("Content-Type", "application/json")
+	router.ServeHTTP(w, req)
+
+	if w.Code != http.StatusBadRequest {
+		t.Errorf("expected status 400, got %d", w.Code)
+	}
+}
+
+func TestUserLogin_Success(t *testing.T) {
+	mock := &mockUserService{
+		loginFn: func(req *dto.LoginRequest) (*dto.UserResponse, string, error) {
+			return &dto.UserResponse{ID: 1, Username: "testuser", Email: "test@example.com"}, "test-token", nil
+		},
+	}
+	h := handler.NewUserHandler(mock)
+	router := setupUserRouter(h)
+
+	body, _ := json.Marshal(dto.LoginRequest{Email: "test@example.com", Password: "password123"})
+	w := httptest.NewRecorder()
+	req, _ := http.NewRequest("POST", "/api/auth/login", bytes.NewReader(body))
+	req.Header.Set("Content-Type", "application/json")
+	router.ServeHTTP(w, req)
+
+	if w.Code != http.StatusOK {
+		t.Errorf("expected status 200, got %d", w.Code)
+	}
+
+	var resp map[string]interface{}
+	json.Unmarshal(w.Body.Bytes(), &resp)
+	if resp["token"] != "test-token" {
+		t.Errorf("expected token 'test-token', got %v", resp["token"])
+	}
+}
+
+func TestUserLogin_InvalidCredentials(t *testing.T) {
+	mock := &mockUserService{
+		loginFn: func(req *dto.LoginRequest) (*dto.UserResponse, string, error) {
+			return nil, "", errors.New("invalid credentials")
+		},
+	}
+	h := handler.NewUserHandler(mock)
+	router := setupUserRouter(h)
+
+	body, _ := json.Marshal(dto.LoginRequest{Email: "wrong@example.com", Password: "wrong"})
+	w := httptest.NewRecorder()
+	req, _ := http.NewRequest("POST", "/api/auth/login", bytes.NewReader(body))
+	req.Header.Set("Content-Type", "application/json")
+	router.ServeHTTP(w, req)
+
+	if w.Code != http.StatusUnauthorized {
+		t.Errorf("expected status 401, got %d", w.Code)
+	}
+}
+
+func TestUserGet_Success(t *testing.T) {
+	mock := &mockUserService{
+		getUserByIDFn: func(id uint) (*dto.UserResponse, error) {
+			return &dto.UserResponse{ID: 1, Username: "testuser", Email: "test@example.com"}, nil
+		},
+	}
+	h := handler.NewUserHandler(mock)
+	router := setupUserRouter(h)
+
+	w := httptest.NewRecorder()
+	req, _ := http.NewRequest("GET", "/api/users/1", nil)
+	router.ServeHTTP(w, req)
+
+	if w.Code != http.StatusOK {
+		t.Errorf("expected status 200, got %d", w.Code)
+	}
+}
+
+func TestUserGet_NotFound(t *testing.T) {
+	mock := &mockUserService{
+		getUserByIDFn: func(id uint) (*dto.UserResponse, error) {
+			return nil, errors.New("user not found")
+		},
+	}
+	h := handler.NewUserHandler(mock)
+	router := setupUserRouter(h)
+
+	w := httptest.NewRecorder()
+	req, _ := http.NewRequest("GET", "/api/users/999", nil)
+	router.ServeHTTP(w, req)
+
+	if w.Code != http.StatusNotFound {
+		t.Errorf("expected status 404, got %d", w.Code)
+	}
+}