// Package repository_test 包含对 repository 包的集成测试 package repository_test import ( "testing" "git.huangwc.com/pig/pig-farm-controller/internal/infra/models" "git.huangwc.com/pig/pig-farm-controller/internal/infra/repository" "github.com/stretchr/testify/assert" "gorm.io/gorm" ) func TestGormUserRepository(t *testing.T) { db := setupTestDB(t) repo := repository.NewGormUserRepository(db) plainPassword := "my-secret-password" userToCreate := &models.User{ Username: "testuser", Password: plainPassword, // 我们提供的是明文密码 } t.Run("创建 - 成功创建并验证密码哈希", func(t *testing.T) { err := repo.Create(userToCreate) assert.NoError(t, err) // 验证用户已被创建 assert.NotZero(t, userToCreate.ID) // 从数据库中直接取回记录,以验证 BeforeSave 钩子是否生效 var savedUser models.User db.First(&savedUser, userToCreate.ID) // 验证密码字段存储的不是明文 assert.NotEqual(t, plainPassword, savedUser.Password, "数据库中存储的密码不应是明文") // 验证存储的哈希是正确的 assert.True(t, savedUser.CheckPassword(plainPassword), "存储的密码哈希应该能与原明文匹配") }) t.Run("创建 - 用户名冲突", func(t *testing.T) { // 尝试创建一个同名用户 duplicateUser := &models.User{Username: "testuser", Password: "anypassword"} err := repo.Create(duplicateUser) // 我们期望一个错误,因为用户名是唯一的 assert.Error(t, err, "创建同名用户应该返回错误") // 更精确地,可以检查是否是唯一键冲突错误 assert.Contains(t, err.Error(), "UNIQUE constraint failed: users.username", "错误信息应包含唯一键冲突") }) t.Run("按用户名查找 - 找到用户", func(t *testing.T) { foundUser, err := repo.FindByUsername("testuser") assert.NoError(t, err) assert.NotNil(t, foundUser) assert.Equal(t, userToCreate.ID, foundUser.ID) assert.Equal(t, "testuser", foundUser.Username) }) t.Run("按用户名查找 - 未找到用户", func(t *testing.T) { _, err := repo.FindByUsername("nonexistent") assert.Error(t, err, "查找不存在的用户应该返回错误") assert.ErrorIs(t, err, gorm.ErrRecordNotFound, "错误类型应为 gorm.ErrRecordNotFound") }) t.Run("按ID查找 - 找到用户", func(t *testing.T) { foundUser, err := repo.FindByID(userToCreate.ID) assert.NoError(t, err) assert.NotNil(t, foundUser) assert.Equal(t, userToCreate.ID, foundUser.ID) }) t.Run("按ID查找 - 未找到用户", func(t *testing.T) { _, err := repo.FindByID(99999) assert.Error(t, err, "查找不存在的ID应该返回错误") assert.ErrorIs(t, err, gorm.ErrRecordNotFound, "错误类型应为 gorm.ErrRecordNotFound") }) }