115 lines
3.3 KiB
Go
115 lines
3.3 KiB
Go
package origin
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"golang.org/x/net/context"
|
|
)
|
|
|
|
func immediateTimeAfter(time.Duration) <-chan time.Time {
|
|
c := make(chan time.Time, 1)
|
|
c <- time.Now()
|
|
return c
|
|
}
|
|
|
|
func TestBackoffRetries(t *testing.T) {
|
|
// make backoff return immediately
|
|
timeAfter = immediateTimeAfter
|
|
ctx := context.Background()
|
|
backoff := BackoffHandler{MaxRetries: 3}
|
|
if !backoff.Backoff(ctx) {
|
|
t.Fatalf("backoff failed immediately")
|
|
}
|
|
if !backoff.Backoff(ctx) {
|
|
t.Fatalf("backoff failed after 1 retry")
|
|
}
|
|
if !backoff.Backoff(ctx) {
|
|
t.Fatalf("backoff failed after 2 retry")
|
|
}
|
|
if backoff.Backoff(ctx) {
|
|
t.Fatalf("backoff allowed after 3 (max) retries")
|
|
}
|
|
}
|
|
|
|
func TestBackoffCancel(t *testing.T) {
|
|
// prevent backoff from returning normally
|
|
timeAfter = func(time.Duration) <-chan time.Time { return make(chan time.Time) }
|
|
ctx, cancelFunc := context.WithCancel(context.Background())
|
|
backoff := BackoffHandler{MaxRetries: 3}
|
|
cancelFunc()
|
|
if backoff.Backoff(ctx) {
|
|
t.Fatalf("backoff allowed after cancel")
|
|
}
|
|
if _, ok := backoff.GetBackoffDuration(ctx); ok {
|
|
t.Fatalf("backoff allowed after cancel")
|
|
}
|
|
}
|
|
|
|
func TestBackoffGracePeriod(t *testing.T) {
|
|
currentTime := time.Now()
|
|
// make timeNow return whatever we like
|
|
timeNow = func() time.Time { return currentTime }
|
|
// make backoff return immediately
|
|
timeAfter = immediateTimeAfter
|
|
ctx := context.Background()
|
|
backoff := BackoffHandler{MaxRetries: 1}
|
|
if !backoff.Backoff(ctx) {
|
|
t.Fatalf("backoff failed immediately")
|
|
}
|
|
// the next call to Backoff would fail unless it's after the grace period
|
|
backoff.SetGracePeriod()
|
|
// advance time to after the grace period (~4 seconds) and see what happens
|
|
currentTime = currentTime.Add(time.Second * 5)
|
|
if !backoff.Backoff(ctx) {
|
|
t.Fatalf("backoff failed after the grace period expired")
|
|
}
|
|
// confirm we ignore grace period after backoff
|
|
if backoff.Backoff(ctx) {
|
|
t.Fatalf("backoff allowed after 1 (max) retry")
|
|
}
|
|
}
|
|
|
|
func TestGetBackoffDurationRetries(t *testing.T) {
|
|
// make backoff return immediately
|
|
timeAfter = immediateTimeAfter
|
|
ctx := context.Background()
|
|
backoff := BackoffHandler{MaxRetries: 3}
|
|
if _, ok := backoff.GetBackoffDuration(ctx); !ok {
|
|
t.Fatalf("backoff failed immediately")
|
|
}
|
|
backoff.Backoff(ctx) // noop
|
|
if _, ok := backoff.GetBackoffDuration(ctx); !ok {
|
|
t.Fatalf("backoff failed after 1 retry")
|
|
}
|
|
backoff.Backoff(ctx) // noop
|
|
if _, ok := backoff.GetBackoffDuration(ctx); !ok {
|
|
t.Fatalf("backoff failed after 2 retry")
|
|
}
|
|
backoff.Backoff(ctx) // noop
|
|
if _, ok := backoff.GetBackoffDuration(ctx); ok {
|
|
t.Fatalf("backoff allowed after 3 (max) retries")
|
|
}
|
|
if backoff.Backoff(ctx) {
|
|
t.Fatalf("backoff allowed after 3 (max) retries")
|
|
}
|
|
}
|
|
|
|
func TestGetBackoffDuration(t *testing.T) {
|
|
// make backoff return immediately
|
|
timeAfter = immediateTimeAfter
|
|
ctx := context.Background()
|
|
backoff := BackoffHandler{MaxRetries: 3}
|
|
if duration, ok := backoff.GetBackoffDuration(ctx); !ok || duration != time.Second {
|
|
t.Fatalf("backoff didn't return 1 second on first retry")
|
|
}
|
|
backoff.Backoff(ctx) // noop
|
|
if duration, ok := backoff.GetBackoffDuration(ctx); !ok || duration != time.Second*2 {
|
|
t.Fatalf("backoff didn't return 2 seconds on second retry")
|
|
}
|
|
backoff.Backoff(ctx) // noop
|
|
if duration, ok := backoff.GetBackoffDuration(ctx); !ok || duration != time.Second*4 {
|
|
t.Fatalf("backoff didn't return 4 seconds on third retry")
|
|
}
|
|
}
|