stack/ee/webhooks/pkg/backoff/exponential_test.go

package backoff

import (
	"fmt"
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
)

func TestExponential_Nominal(t *testing.T) {
	policy := NewExponential(time.Minute, time.Hour, 24*time.Hour)
	wantDurations := []time.Duration{
		time.Minute,
		2 * time.Minute,
		4 * time.Minute,
		8 * time.Minute,
		16 * time.Minute,
		32 * time.Minute,
		time.Hour,
		time.Hour,
		time.Hour,
	}
	for attemptNumber, wantDelay := range wantDurations {
		t.Run(fmt.Sprint("attempt #", attemptNumber), func(t *testing.T) {
			t.Parallel()
			gotDelay, err := policy.GetRetryDelay(attemptNumber)

			assert.NoError(t, err)
			assert.Equal(t, wantDelay, gotDelay)
		})
	}
}

func TestExponential_Limit(t *testing.T) {
	// Attempt:             0    1    2
	// Delay:               1m   2m   X
	// sinceFirstAttempt:   1m   3m   X
	policy := NewExponential(time.Minute, 5*time.Minute, 3*time.Minute)

	delay, err := policy.GetRetryDelay(1)
	assert.NoError(t, err)
	assert.Equal(t, 2*time.Minute, delay)

	_, err = policy.GetRetryDelay(2)
	assert.ErrorIs(t, err, ErrMaxAttemptsReached)
}