From 8acefb2dd3f2311b29e992e61ade596bb924c93e Mon Sep 17 00:00:00 2001 From: huang <1724659546@qq.com> Date: Fri, 12 Sep 2025 16:36:44 +0800 Subject: [PATCH] =?UTF-8?q?DeviceRepository=E5=8D=95=E6=B5=8B=E5=AE=9E?= =?UTF-8?q?=E7=8E=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../repository/device_repository_test.go | 107 ++++++++++++++++++ internal/infra/repository/main_test.go | 38 +++++++ .../infra/repository/user_repository_test.go | 27 +---- 3 files changed, 151 insertions(+), 21 deletions(-) create mode 100644 internal/infra/repository/device_repository_test.go create mode 100644 internal/infra/repository/main_test.go diff --git a/internal/infra/repository/device_repository_test.go b/internal/infra/repository/device_repository_test.go new file mode 100644 index 0000000..034dbe3 --- /dev/null +++ b/internal/infra/repository/device_repository_test.go @@ -0,0 +1,107 @@ +package repository_test + +import ( + "encoding/json" + "strconv" + "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 TestGormDeviceRepository(t *testing.T) { + db := setupTestDB(t) + repo := repository.NewGormDeviceRepository(db) + + // --- 准备测试数据 --- + loraProps, _ := json.Marshal(models.LoraProperties{LoraAddress: "0xABCD"}) + busProps, _ := json.Marshal(models.BusProperties{BusID: 1, BusAddress: 10}) + + areaController := &models.Device{ + Name: "1号猪舍主控", + Type: models.DeviceTypeAreaController, + Location: "1号猪舍", + Properties: loraProps, + } + + t.Run("创建 - 成功创建区域主控", func(t *testing.T) { + err := repo.Create(areaController) + assert.NoError(t, err) + assert.NotZero(t, areaController.ID, "创建后应获得一个非零ID") + assert.Nil(t, areaController.ParentID, "区域主控的 ParentID 应为 nil") + }) + + var createdDevice *models.Device + t.Run("通过ID查找 - 成功找到已创建的设备", func(t *testing.T) { + var err error + createdDevice, err = repo.FindByID(areaController.ID) + assert.NoError(t, err) + assert.NotNil(t, createdDevice) + assert.Equal(t, areaController.Name, createdDevice.Name) + }) + + t.Run("通过字符串ID查找 - 使用有效字符串ID找到设备", func(t *testing.T) { + foundDevice, err := repo.FindByIDString(strconv.FormatUint(uint64(areaController.ID), 10)) + assert.NoError(t, err) + assert.NotNil(t, foundDevice) + assert.Equal(t, areaController.ID, foundDevice.ID) + }) + + t.Run("通过字符串ID查找 - 使用无效字符串ID", func(t *testing.T) { + _, err := repo.FindByIDString("invalid-id") + assert.Error(t, err, "使用无效ID字符串应返回错误") + }) + + // 创建一个子设备 + childDevice := &models.Device{ + Name: "1号猪舍温度传感器", + Type: models.DeviceTypeDevice, + SubType: models.SubTypeSensorTemp, + ParentID: &areaController.ID, + Location: "1号猪舍东侧", + Properties: busProps, + } + + t.Run("创建 - 成功创建子设备", func(t *testing.T) { + err := repo.Create(childDevice) + assert.NoError(t, err) + assert.NotZero(t, childDevice.ID) + assert.NotNil(t, childDevice.ParentID) + assert.Equal(t, areaController.ID, *childDevice.ParentID) + }) + + t.Run("通过父ID列出 - 找到子设备", func(t *testing.T) { + children, err := repo.ListByParentID(&areaController.ID) + assert.NoError(t, err) + assert.Len(t, children, 1, "应找到一个子设备") + assert.Equal(t, childDevice.ID, children[0].ID) + }) + + t.Run("通过父ID列出 - 找到顶层设备", func(t *testing.T) { + parents, err := repo.ListByParentID(nil) + assert.NoError(t, err) + assert.Len(t, parents, 1, "应找到一个顶层设备") + assert.Equal(t, areaController.ID, parents[0].ID) + }) + + t.Run("更新 - 成功更新设备信息", func(t *testing.T) { + childDevice.Location = "1号猪舍西侧" + err := repo.Update(childDevice) + assert.NoError(t, err) + + updatedDevice, _ := repo.FindByID(childDevice.ID) + assert.Equal(t, "1号猪舍西侧", updatedDevice.Location) + }) + + t.Run("删除 - 成功删除设备", func(t *testing.T) { + err := repo.Delete(childDevice.ID) + assert.NoError(t, err) + + // 验证设备已被软删除 + _, err = repo.FindByID(childDevice.ID) + assert.Error(t, err, "删除后应无法找到设备") + assert.ErrorIs(t, err, gorm.ErrRecordNotFound, "错误类型应为 RecordNotFound") + }) +} diff --git a/internal/infra/repository/main_test.go b/internal/infra/repository/main_test.go new file mode 100644 index 0000000..1ec9752 --- /dev/null +++ b/internal/infra/repository/main_test.go @@ -0,0 +1,38 @@ +package repository_test + +import ( + "os" + "testing" + + "git.huangwc.com/pig/pig-farm-controller/internal/infra/models" + "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, "连接内存数据库不应出错") + + // 自动迁移所有需要的表结构 + err = db.AutoMigrate(&models.User{}, &models.Device{}) + assert.NoError(t, err, "数据库迁移不应出错") + + return db +} + +// TestMain 是一个特殊的函数，它会在包内的所有测试运行之前被调用。 +// 我们可以在这里进行一些全局的设置和清理工作。 +func TestMain(m *testing.M) { + // 在所有测试运行前可以执行一些设置代码 + + // 运行包中的所有测试 + code := m.Run() + + // 在所有测试运行后可以执行一些清理代码 + + // 退出测试 + os.Exit(code) +} diff --git a/internal/infra/repository/user_repository_test.go b/internal/infra/repository/user_repository_test.go index e9c2fcf..b90af54 100644 --- a/internal/infra/repository/user_repository_test.go +++ b/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")