补充测试用例
This commit is contained in:
@@ -11,97 +11,252 @@ import (
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
func TestGormDeviceRepository(t *testing.T) {
|
||||
// createTestDevice 辅助函数，用于创建测试设备
|
||||
func createTestDevice(t *testing.T, db *gorm.DB, name string, deviceType models.DeviceType, parentID *uint) *models.Device {
|
||||
device := &models.Device{
|
||||
Name: name,
|
||||
Type: deviceType,
|
||||
ParentID: parentID,
|
||||
// 其他字段可以根据需要添加
|
||||
}
|
||||
err := db.Create(device).Error
|
||||
assert.NoError(t, err)
|
||||
return device
|
||||
}
|
||||
|
||||
func TestRepoCreate(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)
|
||||
t.Run("成功创建区域主控", func(t *testing.T) {
|
||||
device := &models.Device{
|
||||
Name: "主控A",
|
||||
Type: models.DeviceTypeAreaController,
|
||||
Location: "猪舍1",
|
||||
Properties: loraProps,
|
||||
}
|
||||
err := repo.Create(device)
|
||||
assert.NoError(t, err)
|
||||
assert.NotZero(t, areaController.ID, "创建后应获得一个非零ID")
|
||||
assert.Nil(t, areaController.ParentID, "区域主控的 ParentID 应为 nil")
|
||||
assert.NotZero(t, device.ID, "创建后应获得一个非零ID")
|
||||
assert.Nil(t, device.ParentID, "区域主控的 ParentID 应为 nil")
|
||||
})
|
||||
|
||||
var createdDevice *models.Device
|
||||
t.Run("通过ID查找 - 成功找到已创建的设备", func(t *testing.T) {
|
||||
var err error
|
||||
createdDevice, err = repo.FindByID(areaController.ID)
|
||||
t.Run("成功创建子设备", func(t *testing.T) {
|
||||
parent := createTestDevice(t, db, "父设备", models.DeviceTypeAreaController, nil)
|
||||
child := &models.Device{
|
||||
Name: "子设备A",
|
||||
Type: models.DeviceTypeDevice,
|
||||
ParentID: &parent.ID,
|
||||
}
|
||||
err := repo.Create(child)
|
||||
assert.NoError(t, err)
|
||||
assert.NotNil(t, createdDevice)
|
||||
assert.Equal(t, areaController.Name, createdDevice.Name)
|
||||
assert.NotZero(t, child.ID)
|
||||
assert.NotNil(t, child.ParentID)
|
||||
assert.Equal(t, parent.ID, *child.ParentID)
|
||||
})
|
||||
}
|
||||
|
||||
t.Run("通过字符串ID查找 - 使用有效字符串ID找到设备", func(t *testing.T) {
|
||||
foundDevice, err := repo.FindByIDString(strconv.FormatUint(uint64(areaController.ID), 10))
|
||||
func TestRepoFindByID(t *testing.T) {
|
||||
db := setupTestDB(t)
|
||||
repo := repository.NewGormDeviceRepository(db)
|
||||
|
||||
device := createTestDevice(t, db, "测试设备", models.DeviceTypeAreaController, nil)
|
||||
|
||||
t.Run("成功通过ID查找", func(t *testing.T) {
|
||||
foundDevice, err := repo.FindByID(device.ID)
|
||||
assert.NoError(t, err)
|
||||
assert.NotNil(t, foundDevice)
|
||||
assert.Equal(t, areaController.ID, foundDevice.ID)
|
||||
assert.Equal(t, device.ID, foundDevice.ID)
|
||||
assert.Equal(t, device.Name, foundDevice.Name)
|
||||
})
|
||||
|
||||
t.Run("通过字符串ID查找 - 使用无效字符串ID", func(t *testing.T) {
|
||||
t.Run("查找不存在的ID", func(t *testing.T) {
|
||||
_, err := repo.FindByID(9999) // 不存在的ID
|
||||
assert.Error(t, err)
|
||||
assert.ErrorIs(t, err, gorm.ErrRecordNotFound)
|
||||
})
|
||||
|
||||
t.Run("数据库查询失败", func(t *testing.T) {
|
||||
// 模拟数据库连接关闭，强制查询失败
|
||||
sqlDB, _ := db.DB()
|
||||
sqlDB.Close()
|
||||
|
||||
_, err := repo.FindByID(device.ID)
|
||||
assert.Error(t, err)
|
||||
assert.Contains(t, err.Error(), "database is closed")
|
||||
})
|
||||
}
|
||||
|
||||
func TestRepoFindByIDString(t *testing.T) {
|
||||
db := setupTestDB(t)
|
||||
repo := repository.NewGormDeviceRepository(db)
|
||||
|
||||
device := createTestDevice(t, db, "测试设备", models.DeviceTypeAreaController, nil)
|
||||
|
||||
t.Run("成功通过字符串ID查找", func(t *testing.T) {
|
||||
idStr := strconv.FormatUint(uint64(device.ID), 10)
|
||||
foundDevice, err := repo.FindByIDString(idStr)
|
||||
assert.NoError(t, err)
|
||||
assert.NotNil(t, foundDevice)
|
||||
assert.Equal(t, device.ID, foundDevice.ID)
|
||||
})
|
||||
|
||||
t.Run("无效的字符串ID格式", func(t *testing.T) {
|
||||
_, err := repo.FindByIDString("invalid-id")
|
||||
assert.Error(t, err, "使用无效ID字符串应返回错误")
|
||||
assert.Error(t, err)
|
||||
assert.Contains(t, err.Error(), "无效的设备ID格式")
|
||||
})
|
||||
|
||||
// 创建一个子设备
|
||||
childDevice := &models.Device{
|
||||
Name: "1号猪舍温度传感器",
|
||||
Type: models.DeviceTypeDevice,
|
||||
SubType: models.SubTypeSensorTemp,
|
||||
ParentID: &areaController.ID,
|
||||
Location: "1号猪舍东侧",
|
||||
Properties: busProps,
|
||||
}
|
||||
t.Run("查找不存在的字符串ID", func(t *testing.T) {
|
||||
idStr := strconv.FormatUint(uint64(9999), 10) // 不存在的ID
|
||||
_, err := repo.FindByIDString(idStr)
|
||||
assert.Error(t, err)
|
||||
assert.ErrorIs(t, err, gorm.ErrRecordNotFound)
|
||||
})
|
||||
|
||||
t.Run("创建 - 成功创建子设备", func(t *testing.T) {
|
||||
err := repo.Create(childDevice)
|
||||
t.Run("数据库查询失败", func(t *testing.T) {
|
||||
sqlDB, _ := db.DB()
|
||||
sqlDB.Close()
|
||||
|
||||
idStr := strconv.FormatUint(uint64(device.ID), 10)
|
||||
_, err := repo.FindByIDString(idStr)
|
||||
assert.Error(t, err)
|
||||
assert.Contains(t, err.Error(), "database is closed")
|
||||
})
|
||||
}
|
||||
|
||||
func TestRepoListAll(t *testing.T) {
|
||||
db := setupTestDB(t)
|
||||
repo := repository.NewGormDeviceRepository(db)
|
||||
|
||||
t.Run("成功获取空列表", func(t *testing.T) {
|
||||
devices, err := repo.ListAll()
|
||||
assert.NoError(t, err)
|
||||
assert.NotZero(t, childDevice.ID)
|
||||
assert.NotNil(t, childDevice.ParentID)
|
||||
assert.Equal(t, areaController.ID, *childDevice.ParentID)
|
||||
assert.Empty(t, devices)
|
||||
})
|
||||
|
||||
t.Run("通过父ID列出 - 找到子设备", func(t *testing.T) {
|
||||
children, err := repo.ListByParentID(&areaController.ID)
|
||||
t.Run("成功获取包含设备的列表", func(t *testing.T) {
|
||||
createTestDevice(t, db, "设备1", models.DeviceTypeAreaController, nil)
|
||||
createTestDevice(t, db, "设备2", models.DeviceTypeDevice, nil)
|
||||
|
||||
devices, err := repo.ListAll()
|
||||
assert.NoError(t, err)
|
||||
assert.Len(t, children, 1, "应找到一个子设备")
|
||||
assert.Equal(t, childDevice.ID, children[0].ID)
|
||||
assert.Len(t, devices, 2)
|
||||
assert.Equal(t, "设备1", devices[0].Name)
|
||||
assert.Equal(t, "设备2", devices[1].Name)
|
||||
})
|
||||
|
||||
t.Run("通过父ID列出 - 找到顶层设备", func(t *testing.T) {
|
||||
t.Run("数据库查询失败", func(t *testing.T) {
|
||||
sqlDB, _ := db.DB()
|
||||
sqlDB.Close()
|
||||
|
||||
_, err := repo.ListAll()
|
||||
assert.Error(t, err)
|
||||
assert.Contains(t, err.Error(), "database is closed")
|
||||
})
|
||||
}
|
||||
|
||||
func TestRepoListByParentID(t *testing.T) {
|
||||
db := setupTestDB(t)
|
||||
repo := repository.NewGormDeviceRepository(db)
|
||||
|
||||
parent1 := createTestDevice(t, db, "父设备1", models.DeviceTypeAreaController, nil)
|
||||
parent2 := createTestDevice(t, db, "父设备2", models.DeviceTypeAreaController, nil)
|
||||
child1_1 := createTestDevice(t, db, "子设备1-1", models.DeviceTypeDevice, &parent1.ID)
|
||||
child1_2 := createTestDevice(t, db, "子设备1-2", models.DeviceTypeDevice, &parent1.ID)
|
||||
_ = createTestDevice(t, db, "子设备2-1", models.DeviceTypeDevice, &parent2.ID)
|
||||
|
||||
t.Run("成功通过父ID查找子设备", func(t *testing.T) {
|
||||
children, err := repo.ListByParentID(&parent1.ID)
|
||||
assert.NoError(t, err)
|
||||
assert.Len(t, children, 2)
|
||||
assert.Contains(t, []uint{child1_1.ID, child1_2.ID}, children[0].ID)
|
||||
assert.Contains(t, []uint{child1_1.ID, child1_2.ID}, children[1].ID)
|
||||
})
|
||||
|
||||
t.Run("成功通过nil父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)
|
||||
assert.Len(t, parents, 2)
|
||||
assert.Contains(t, []uint{parent1.ID, parent2.ID}, parents[0].ID)
|
||||
assert.Contains(t, []uint{parent1.ID, parent2.ID}, parents[1].ID)
|
||||
})
|
||||
|
||||
t.Run("更新 - 成功更新设备信息", func(t *testing.T) {
|
||||
childDevice.Location = "1号猪舍西侧"
|
||||
err := repo.Update(childDevice)
|
||||
t.Run("查找不存在的父ID", func(t *testing.T) {
|
||||
nonExistentParentID := uint(9999)
|
||||
children, err := repo.ListByParentID(&nonExistentParentID)
|
||||
assert.NoError(t, err) // GORM 在未找到时返回空列表而不是错误
|
||||
assert.Empty(t, children)
|
||||
})
|
||||
|
||||
t.Run("数据库查询失败", func(t *testing.T) {
|
||||
sqlDB, _ := db.DB()
|
||||
sqlDB.Close()
|
||||
|
||||
_, err := repo.ListByParentID(&parent1.ID)
|
||||
assert.Error(t, err)
|
||||
assert.Contains(t, err.Error(), "database is closed")
|
||||
})
|
||||
}
|
||||
|
||||
func TestRepoUpdate(t *testing.T) {
|
||||
db := setupTestDB(t)
|
||||
repo := repository.NewGormDeviceRepository(db)
|
||||
|
||||
device := createTestDevice(t, db, "原始设备", models.DeviceTypeAreaController, nil)
|
||||
|
||||
t.Run("成功更新设备信息", func(t *testing.T) {
|
||||
device.Name = "更新后的设备"
|
||||
device.Location = "新地点"
|
||||
err := repo.Update(device)
|
||||
assert.NoError(t, err)
|
||||
|
||||
updatedDevice, _ := repo.FindByID(childDevice.ID)
|
||||
assert.Equal(t, "1号猪舍西侧", updatedDevice.Location)
|
||||
updatedDevice, err := repo.FindByID(device.ID)
|
||||
assert.NoError(t, err)
|
||||
assert.Equal(t, "更新后的设备", updatedDevice.Name)
|
||||
assert.Equal(t, "新地点", updatedDevice.Location)
|
||||
})
|
||||
|
||||
t.Run("删除 - 成功删除设备", func(t *testing.T) {
|
||||
err := repo.Delete(childDevice.ID)
|
||||
t.Run("数据库更新失败", func(t *testing.T) {
|
||||
sqlDB, _ := db.DB()
|
||||
sqlDB.Close()
|
||||
|
||||
device.Name = "更新失败的设备"
|
||||
err := repo.Update(device)
|
||||
assert.Error(t, err)
|
||||
assert.Contains(t, err.Error(), "database is closed")
|
||||
})
|
||||
}
|
||||
|
||||
func TestRepoDelete(t *testing.T) {
|
||||
db := setupTestDB(t)
|
||||
repo := repository.NewGormDeviceRepository(db)
|
||||
|
||||
device := createTestDevice(t, db, "待删除设备", models.DeviceTypeAreaController, nil)
|
||||
|
||||
t.Run("成功删除设备", func(t *testing.T) {
|
||||
err := repo.Delete(device.ID)
|
||||
assert.NoError(t, err)
|
||||
|
||||
// 验证设备已被软删除
|
||||
_, err = repo.FindByID(childDevice.ID)
|
||||
_, err = repo.FindByID(device.ID)
|
||||
assert.Error(t, err, "删除后应无法找到设备")
|
||||
assert.ErrorIs(t, err, gorm.ErrRecordNotFound, "错误类型应为 RecordNotFound")
|
||||
})
|
||||
|
||||
t.Run("删除不存在的设备", func(t *testing.T) {
|
||||
err := repo.Delete(9999) // 不存在的ID
|
||||
assert.NoError(t, err) // GORM 的 Delete 方法在删除不存在的记录时不会报错
|
||||
})
|
||||
|
||||
t.Run("数据库删除失败", func(t *testing.T) {
|
||||
sqlDB, _ := db.DB()
|
||||
sqlDB.Close()
|
||||
|
||||
err := repo.Delete(device.ID)
|
||||
assert.Error(t, err)
|
||||
assert.Contains(t, err.Error(), "database is closed")
|
||||
})
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user