1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48
|
package backoff
import (
"math/rand"
"testing"
"time"
"github.com/stretchr/testify/assert"
)
func TestNewExponentialIntervalWithDefaultOptions(t *testing.T) {
p := NewExponentialInterval()
assert.Equal(t, defaultMaxInterval, p.maxInterval)
assert.Equal(t, defaultMinInterval, p.minInterval)
assert.Equal(t, defaultMultiplier, p.multiplier)
assert.Equal(t, &nopJitter{}, p.jitter)
}
func TestNewExponentialIntervalWithCustomOptions(t *testing.T) {
jitter := 0.99
maxInterval := 24 * time.Hour
minInterval := time.Nanosecond
multiplier := float64(99999)
rng := rand.New(rand.NewSource(time.Now().UnixNano()))
p := NewExponentialInterval(
WithJitterFactor(jitter),
WithMaxInterval(maxInterval),
WithMinInterval(minInterval),
WithMultiplier(multiplier),
WithRNG(rng),
)
assert.Equal(t, maxInterval, time.Duration(p.maxInterval))
assert.Equal(t, minInterval, time.Duration(p.minInterval))
assert.Equal(t, multiplier, p.multiplier)
assert.Equal(t, newRandomJitter(jitter, rng), p.jitter)
}
func TestNewExponentialIntervalWithOnlyJitterOptions(t *testing.T) {
jitter := 0.99
p := NewExponentialInterval(
WithJitterFactor(jitter),
)
generatedRandomJitter := p.jitter.(*randomJitter)
assert.Equal(t, newRandomJitter(jitter, generatedRandomJitter.rng), p.jitter)
}
|