blob: d55780c45ef3c7cde30cbaeaa18030717a6a270f [file] [log] [blame]
// Copyright 2018 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
package retry
import (
"testing"
"time"
)
func TestZeroBackoff(t *testing.T) {
backoff := ZeroBackoff{}
backoff.Reset()
if backoff.Next() != 0 {
t.Error("invalid interval")
}
}
func TestConstantBackoff(t *testing.T) {
backoff := NewConstantBackoff(time.Second)
backoff.Reset()
if backoff.Next() != time.Second {
t.Error("invalid interval")
}
}
func TestMaxTriesBackoff(t *testing.T) {
backoff := WithMaxRetries(&ZeroBackoff{}, 10)
backoff.Reset()
for i := 0; i < 10; i++ {
if backoff.Next() != 0 {
t.Error("invalid interval")
}
}
if backoff.Next() != Stop {
t.Error("did not stop")
}
}
func TestMaxDurationBackoff(t *testing.T) {
c := &fakeClock{t: time.Now()}
backoff := &maxDurationBackoff{backOff: &ZeroBackoff{}, maxDuration: 10 * time.Second, c: c}
backoff.Reset()
if backoff.Next() != 0 {
t.Error("invalid interval")
}
c.Tick(9 * time.Second)
if backoff.Next() != 0 {
t.Error("invalid interval")
}
c.Tick(1 * time.Second)
if backoff.Next() != Stop {
t.Error("did not stop")
}
}