lightning: fix test lightning_checkpoint_chunks

pingcap · Mar 28, 2022 · 9fedebf · 9fedebf
1 parent 05b9157
commit 9fedebf
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/br/pkg/lightning/lightning.go b/br/pkg/lightning/lightning.go
@@ -267,6 +267,9 @@ func (l *Lightning) RunOnceWithOptions(taskCtx context.Context, taskCfg *config.
 	}
 
 	taskCfg.TaskID = time.Now().UnixNano()
+	failpoint.Inject("SetTaskID", func(val failpoint.Value) {
+		taskCfg.TaskID = int64(val.(int))
+	})
 
 	return l.run(taskCtx, taskCfg, o)
 }