优化结构

internal/worker/task_runtime_test.go

@@ -75,7 +75,8 @@ func TestReloadPreservesExistingScheduleOnPreparationFailure(t *testing.T) {
 	repo := &workerTaskRepo{}
 	scheduler := newTestTaskScheduler(repo)
 	old := &taskbiz.TimedTask{ID: 1, Name: "old", Spec: "0 0 * * *", Enabled: true}
-	if err := scheduler.Schedule(old); err != nil {
+	repo.items = []*taskbiz.TimedTask{old}
+	if err := scheduler.ScheduleID(context.Background(), old.ID); err != nil {
 		t.Fatalf("schedule old task: %v", err)
 	}
 
@@ -116,16 +117,17 @@ func assertOnlyScheduledTask(t *testing.T, scheduler *TaskScheduler, id uint) {
 	}
 }
 
-func TestConcurrentScheduleKeepsSingleEntry(t *testing.T) {
-	scheduler := newTestTaskScheduler(&workerTaskRepo{})
+func TestConcurrentScheduleIDKeepsSingleEntry(t *testing.T) {
+	repo := &workerTaskRepo{items: []*taskbiz.TimedTask{{ID: 7, Name: "same", Spec: "0 0 * * *", Enabled: true}}}
+	scheduler := newTestTaskScheduler(repo)
 	const workers = 32
 	var wg sync.WaitGroup
 	for i := 0; i < workers; i++ {
 		wg.Add(1)
 		go func() {
 			defer wg.Done()
-			if err := scheduler.Schedule(&taskbiz.TimedTask{ID: 7, Name: "same", Spec: "0 0 * * *", Enabled: true}); err != nil {
-				t.Errorf("Schedule() error = %v", err)
+			if err := scheduler.ScheduleID(context.Background(), 7); err != nil {
+				t.Errorf("ScheduleID() error = %v", err)
 			}
 		}()
 	}