DeviceRepository单测实现

internal/infra/repository/user_repository_test.go

@@ -7,24 +7,9 @@ import (
 	"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/driver/sqlite"
 	"gorm.io/gorm"
 )
 
-// setupTestDB 是一个辅助函数，用于为每个测试创建一个
-// 干净的、内存中的 SQLite 数据库实例。
-func setupTestDB(t *testing.T) *gorm.DB {
-	// "file::memory:?cache=shared" 是 GORM 连接内存 SQLite 的标准方式
-	db, err := gorm.Open(sqlite.Open("file::memory:?cache=shared"), &gorm.Config{})
-	assert.NoError(t, err, "连接内存数据库不应出错")
-
-	// 自动迁移 User 表结构
-	err = db.AutoMigrate(&models.User{})
-	assert.NoError(t, err, "数据库迁移不应出错")
-
-	return db
-}
-
 func TestGormUserRepository(t *testing.T) {
 	db := setupTestDB(t)
 	repo := repository.NewGormUserRepository(db)
@@ -35,7 +20,7 @@ func TestGormUserRepository(t *testing.T) {
 		Password: plainPassword, // 我们提供的是明文密码
 	}
 
-	t.Run("Create - 成功创建并验证密码哈希", func(t *testing.T) {
+	t.Run("创建 - 成功创建并验证密码哈希", func(t *testing.T) {
 		err := repo.Create(userToCreate)
 		assert.NoError(t, err)
 
@@ -53,7 +38,7 @@ func TestGormUserRepository(t *testing.T) {
 		assert.True(t, savedUser.CheckPassword(plainPassword), "存储的密码哈希应该能与原明文匹配")
 	})
 
-	t.Run("Create - 用户名冲突", func(t *testing.T) {
+	t.Run("创建 - 用户名冲突", func(t *testing.T) {
 		// 尝试创建一个同名用户
 		duplicateUser := &models.User{Username: "testuser", Password: "anypassword"}
 		err := repo.Create(duplicateUser)
@@ -64,7 +49,7 @@ func TestGormUserRepository(t *testing.T) {
 		assert.Contains(t, err.Error(), "UNIQUE constraint failed: users.username", "错误信息应包含唯一键冲突")
 	})
 
-	t.Run("FindByUsername - 找到用户", func(t *testing.T) {
+	t.Run("按用户名查找 - 找到用户", func(t *testing.T) {
 		foundUser, err := repo.FindByUsername("testuser")
 		assert.NoError(t, err)
 		assert.NotNil(t, foundUser)
@@ -72,20 +57,20 @@ func TestGormUserRepository(t *testing.T) {
 		assert.Equal(t, "testuser", foundUser.Username)
 	})
 
-	t.Run("FindByUsername - 未找到用户", func(t *testing.T) {
+	t.Run("按用户名查找 - 未找到用户", func(t *testing.T) {
 		_, err := repo.FindByUsername("nonexistent")
 		assert.Error(t, err, "查找不存在的用户应该返回错误")
 		assert.ErrorIs(t, err, gorm.ErrRecordNotFound, "错误类型应为 gorm.ErrRecordNotFound")
 	})
 
-	t.Run("FindByID - 找到用户", func(t *testing.T) {
+	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("FindByID - 未找到用户", func(t *testing.T) {
+	t.Run("按ID查找 - 未找到用户", func(t *testing.T) {
 		_, err := repo.FindByID(99999)
 		assert.Error(t, err, "查找不存在的ID应该返回错误")
 		assert.ErrorIs(t, err, gorm.ErrRecordNotFound, "错误类型应为 gorm.ErrRecordNotFound")