// Package models_test 包含对 models 包的单元测试 package models_test import ( "testing" "git.huangwc.com/pig/pig-farm-controller/internal/infra/models" "github.com/stretchr/testify/assert" "golang.org/x/crypto/bcrypt" ) func TestUser_CheckPassword(t *testing.T) { plainPassword := "my-secret-password" // 1. 生成一个密码哈希用于测试 hashedPassword, err := bcrypt.GenerateFromPassword([]byte(plainPassword), bcrypt.DefaultCost) assert.NoError(t, err, "生成密码哈希不应出错") user := &models.User{ Password: string(hashedPassword), } t.Run("密码正确", func(t *testing.T) { // 2. 使用正确的明文密码进行校验 match := user.CheckPassword(plainPassword) assert.True(t, match, "正确的密码应该校验通过") }) t.Run("密码错误", func(t *testing.T) { // 3. 使用错误的明文密码进行校验 match := user.CheckPassword("wrong-password") assert.False(t, match, "错误的密码应该校验失败") }) t.Run("空密码", func(t *testing.T) { // 4. 使用空字符串作为密码进行校验 match := user.CheckPassword("") assert.False(t, match, "空密码应该校验失败") }) } func TestUser_BeforeCreate(t *testing.T) { t.Run("密码应被成功哈希", func(t *testing.T) { plainPassword := "securepassword123" user := &models.User{ Username: "testuser", Password: plainPassword, } // 模拟 GORM 钩子调用 err := user.BeforeCreate(nil) // GORM 钩子通常接收 *gorm.DB,这里我们传入 nil,因为 BeforeCreate 不依赖 DB assert.NoError(t, err, "BeforeCreate 不应返回错误") // 验证密码是否已被哈希(不再是明文) assert.NotEqual(t, plainPassword, user.Password, "密码应已被哈希") // 验证哈希后的密码是否能被正确校验 assert.True(t, user.CheckPassword(plainPassword), "哈希后的密码应能通过校验") }) t.Run("空密码不应被哈希", func(t *testing.T) { plainPassword := "" user := &models.User{ Username: "empty_pass_user", Password: plainPassword, } // 模拟 GORM 钩子调用 err := user.BeforeCreate(nil) assert.NoError(t, err, "BeforeCreate 不应返回错误") // 验证密码仍然是空字符串 assert.Equal(t, plainPassword, user.Password, "空密码不应被哈希") }) }