Add SpeedUpForUT function to sched package (#2484)

* Add SpeedUpForUT function to sched package --------- Signed-off-by: Tengxiao Wang <[email protected]>
libopenstorage · Sep 20, 2024 · 2478442 · 2478442
1 parent 464a4a2
commit 2478442
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/pkg/sched/intervals.go b/pkg/sched/intervals.go
@@ -27,6 +27,8 @@ const (
 var (
 	// speedUp advances the clock faster for tests.
 	speedUp = false
+	// speedUpIntervalForUT overrides the interval of schedules in unit tests; no-op if set to 0
+	speedUpIntervalForUT = time.Duration(0)
 )
 
 // SpeedUp advances the clock faster for tests.
@@ -38,6 +40,10 @@ func inSpeedUp() bool {
 	return speedUp
 }
 
+func SpeedUpForUT(duration time.Duration) {
+	speedUpIntervalForUT = duration
+}
+
 type IntervalSpec struct {
 	Freq    string
 	Period  uint64 `yaml:"period,omitempty"`
@@ -560,6 +566,8 @@ func (p RetainIntervalImpl) nextAfter(t time.Time) time.Time {
 	newTime := p.iv.nextAfter(t)
 	if inSpeedUp() {
 		return t.Add(time.Minute)
+	} else if speedUpIntervalForUT != time.Duration(0) {
+		return t.Add(speedUpIntervalForUT)
 	}
 	return newTime
 }