pig/pig-farm-controller
1
0
Fork 0
Code Issues 11 Pull Requests Actions Packages Projects Releases Wiki Activity

提取修改猪群数量逻辑

Browse Source
This commit is contained in:
huang
2025-10-06 15:08:32 +08:00
parent 59b6977367
commit 91e18c432c
8 changed files with 66 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
internal/infra/repository/plan_repository_test.go
View File

@@ -870,7 +870,7 @@ func TestPlanRepository_Create(t *testing.T) {
type testCase struct {
name string
setupDB func(db *gorm.DB) // 准备数据库的初始状态
inputPlan *models.Plan // 传入 Create 方法的计划对象
inputPlan *models.Plan // 传入 CreateTx 方法的计划对象
expectedError error // 期望的错误类型
verifyDB func(t *testing.T, db *gorm.DB, createdPlan *models.Plan) // 验证数据库状态
}
@@ -1040,7 +1040,7 @@ func TestPlanRepository_Create(t *testing.T) {
{Name: "Task 2", ExecutionOrder: 1}, // 重复的顺序
},
},
expectedError: fmt.Errorf("任务执行顺序重复: %d", 1), // 假设 Create 方法会返回此错误
expectedError: fmt.Errorf("任务执行顺序重复: %d", 1), // 假设 CreateTx 方法会返回此错误
verifyDB: func(t *testing.T, db *gorm.DB, createdPlan *models.Plan) {
var count int64
db.Model(&models.Plan{}).Where("name = ?", "重复任务顺序计划").Count(&count)
@@ -1061,7 +1061,7 @@ func TestPlanRepository_Create(t *testing.T) {
{ChildPlanID: 2, ExecutionOrder: 1}, // 重复的顺序
},
},
expectedError: fmt.Errorf("子计划执行顺序重复: %d", 1), // 假设 Create 方法会返回此错误
expectedError: fmt.Errorf("子计划执行顺序重复: %d", 1), // 假设 CreateTx 方法会返回此错误
verifyDB: func(t *testing.T, db *gorm.DB, createdPlan *models.Plan) {
var count int64
db.Model(&models.Plan{}).Where("name = ?", "重复子计划顺序计划").Count(&count)
@@ -1078,7 +1078,7 @@ func TestPlanRepository_Create(t *testing.T) {
// 准备数据库状态
tc.setupDB(db)
// 执行 Create 操作
// 执行 CreateTx 操作
err := repo.CreatePlan(tc.inputPlan)
// 断言错误