pig-farm-controller/internal/infra/models/user_test.go

// 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, "空密码不应被哈希")
	})
}