Function TestTaskBackoffReset

pkg/component/tasks_internal_test.go:40–89

(t *testing.T)

}

func TestTaskBackoffReset(t *testing.T) {
	if enabled := os.Getenv("TEST_TIMING"); enabled == "" {
		t.Skip("Timing sensitive tests are disabled")
	}

	a := assertions.New(t)
	ctx := test.Context()

	c, err := New(test.GetLogger(t), &Config{})
	a.So(err, should.BeNil)

	var (
		calls        uint
		lastCallTime time.Time
		wg           sync.WaitGroup
	)
	wg.Add(4)
	c.RegisterTask(&task.Config{
		Context: ctx,
		ID:      "failing_but_recovering",
		Func: func(_ context.Context) error {
			defer wg.Done()
			calls++
			backoff := time.Since(lastCallTime)
			switch calls {
			case 1:
				// Nothing to expect in the initial call
			case 2:
				// Expect the first backoff interval
				a.So(backoff, should.BeBetweenOrEqual, timeMult(test.Delay, 0.9), timeMult(test.Delay, 1.1))
			case 3:
				// Same jitter requirement, but now act as if we are 'working'
				a.So(backoff, should.BeBetweenOrEqual, timeMult(2*test.Delay, 0.9), timeMult(2*test.Delay, 1.1))
				time.Sleep(3 * test.Delay)
			case 4:
				// Expect the backoff to have been reset
				a.So(backoff, should.BeBetweenOrEqual, timeMult(test.Delay, 0.9), timeMult(test.Delay, 1.1))
			}
			lastCallTime = time.Now()
			return nil
		},
		Restart: task.RestartAlways,
		Backoff: TestTaskBackoffConfig,
	})

	// Wait for all invocations.
	test.Must[any](nil, c.Start())
	defer c.Close()
	wg.Wait()
}

func timeMult(t time.Duration, c float64) time.Duration {
	ft := float64(t)