feat: backend core - auth, user, role, permission, device, webhook, monitoring, cache, repository, service, middleware, API handlers

internal/repository/user_repository_test.go

@@ -0,0 +1,198 @@
+package repository
+
+import (
+	"context"
+	"testing"
+
+	"gorm.io/gorm"
+
+	"github.com/user-management-system/internal/domain"
+)
+
+func setupTestDB(t *testing.T) *gorm.DB {
+	return openTestDB(t)
+}
+
+// TestUserRepository_Create 测试创建用户
+func TestUserRepository_Create(t *testing.T) {
+	db := setupTestDB(t)
+	repo := NewUserRepository(db)
+	ctx := context.Background()
+
+	user := &domain.User{
+		Username: "testuser",
+		Email:    domain.StrPtr("test@example.com"),
+		Phone:    domain.StrPtr("13800138000"),
+		Password: "hashedpassword",
+		Status:   domain.UserStatusActive,
+	}
+
+	if err := repo.Create(ctx, user); err != nil {
+		t.Fatalf("Create() error = %v", err)
+	}
+	if user.ID == 0 {
+		t.Error("创建后用户ID不应为0")
+	}
+}
+
+// TestUserRepository_GetByUsername 测试根据用户名查询
+func TestUserRepository_GetByUsername(t *testing.T) {
+	db := setupTestDB(t)
+	repo := NewUserRepository(db)
+	ctx := context.Background()
+
+	user := &domain.User{
+		Username: "findme",
+		Email:    domain.StrPtr("findme@example.com"),
+		Password: "hash",
+		Status:   domain.UserStatusActive,
+	}
+	repo.Create(ctx, user)
+
+	found, err := repo.GetByUsername(ctx, "findme")
+	if err != nil {
+		t.Fatalf("GetByUsername() error = %v", err)
+	}
+	if found.Username != "findme" {
+		t.Errorf("Username = %v, want findme", found.Username)
+	}
+
+	_, err = repo.GetByUsername(ctx, "notexist")
+	if err == nil {
+		t.Error("查找不存在的用户应返回错误")
+	}
+}
+
+// TestUserRepository_GetByEmail 测试根据邮箱查询
+func TestUserRepository_GetByEmail(t *testing.T) {
+	db := setupTestDB(t)
+	repo := NewUserRepository(db)
+	ctx := context.Background()
+
+	user := &domain.User{
+		Username: "emailuser",
+		Email:    domain.StrPtr("email@example.com"),
+		Password: "hash",
+		Status:   domain.UserStatusActive,
+	}
+	repo.Create(ctx, user)
+
+	found, err := repo.GetByEmail(ctx, "email@example.com")
+	if err != nil {
+		t.Fatalf("GetByEmail() error = %v", err)
+	}
+	if domain.DerefStr(found.Email) != "email@example.com" {
+		t.Errorf("Email = %v, want email@example.com", domain.DerefStr(found.Email))
+	}
+}
+
+// TestUserRepository_Update 测试更新用户
+func TestUserRepository_Update(t *testing.T) {
+	db := setupTestDB(t)
+	repo := NewUserRepository(db)
+	ctx := context.Background()
+
+	user := &domain.User{
+		Username: "updateme",
+		Email:    domain.StrPtr("update@example.com"),
+		Password: "hash",
+		Status:   domain.UserStatusActive,
+	}
+	repo.Create(ctx, user)
+
+	user.Nickname = "已更新"
+	if err := repo.Update(ctx, user); err != nil {
+		t.Fatalf("Update() error = %v", err)
+	}
+
+	found, _ := repo.GetByID(ctx, user.ID)
+	if found.Nickname != "已更新" {
+		t.Errorf("Nickname = %v, want 已更新", found.Nickname)
+	}
+}
+
+// TestUserRepository_Delete 测试删除用户
+func TestUserRepository_Delete(t *testing.T) {
+	db := setupTestDB(t)
+	repo := NewUserRepository(db)
+	ctx := context.Background()
+
+	user := &domain.User{
+		Username: "deleteme",
+		Email:    domain.StrPtr("delete@example.com"),
+		Password: "hash",
+		Status:   domain.UserStatusActive,
+	}
+	repo.Create(ctx, user)
+
+	if err := repo.Delete(ctx, user.ID); err != nil {
+		t.Fatalf("Delete() error = %v", err)
+	}
+
+	_, err := repo.GetByID(ctx, user.ID)
+	if err == nil {
+		t.Error("删除后查询应返回错误")
+	}
+}
+
+// TestUserRepository_ExistsBy 测试存在性检查
+func TestUserRepository_ExistsBy(t *testing.T) {
+	db := setupTestDB(t)
+	repo := NewUserRepository(db)
+	ctx := context.Background()
+
+	user := &domain.User{
+		Username: "existsuser",
+		Email:    domain.StrPtr("exists@example.com"),
+		Phone:    domain.StrPtr("13900139000"),
+		Password: "hash",
+		Status:   domain.UserStatusActive,
+	}
+	repo.Create(ctx, user)
+
+	ok, _ := repo.ExistsByUsername(ctx, "existsuser")
+	if !ok {
+		t.Error("ExistsByUsername 应返回 true")
+	}
+
+	ok, _ = repo.ExistsByEmail(ctx, "exists@example.com")
+	if !ok {
+		t.Error("ExistsByEmail 应返回 true")
+	}
+
+	ok, _ = repo.ExistsByPhone(ctx, "13900139000")
+	if !ok {
+		t.Error("ExistsByPhone 应返回 true")
+	}
+
+	ok, _ = repo.ExistsByUsername(ctx, "notexist")
+	if ok {
+		t.Error("不存在的用户 ExistsByUsername 应返回 false")
+	}
+}
+
+// TestUserRepository_List 测试列表查询
+func TestUserRepository_List(t *testing.T) {
+	db := setupTestDB(t)
+	repo := NewUserRepository(db)
+	ctx := context.Background()
+
+	for i := 0; i < 5; i++ {
+		repo.Create(ctx, &domain.User{
+			Username: "listuser" + string(rune('0'+i)),
+			Password: "hash",
+			Status:   domain.UserStatusActive,
+		})
+	}
+
+	users, total, err := repo.List(ctx, 0, 10)
+	if err != nil {
+		t.Fatalf("List() error = %v", err)
+	}
+	if len(users) != 5 {
+		t.Errorf("len(users) = %d, want 5", len(users))
+	}
+	if total != 5 {
+		t.Errorf("total = %d, want 5", total)
+	}
+}