2016-12-12 00:20:14 +00:00
|
|
|
package tq
|
2016-09-28 18:25:37 +00:00
|
|
|
|
|
|
|
import (
|
|
|
|
"testing"
|
2020-04-15 03:13:27 +00:00
|
|
|
"time"
|
2016-09-28 18:25:37 +00:00
|
|
|
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
)
|
|
|
|
|
2016-12-13 20:06:58 +00:00
|
|
|
func TestManifestDefaultsToFixedRetries(t *testing.T) {
|
2017-10-25 17:46:37 +00:00
|
|
|
assert.Equal(t, 8, NewManifest(nil, nil, "", "").MaxRetries())
|
2016-12-13 20:06:58 +00:00
|
|
|
}
|
2016-09-28 18:25:37 +00:00
|
|
|
|
2020-04-15 03:13:27 +00:00
|
|
|
func TestManifestDefaultsToFixedRetryDelay(t *testing.T) {
|
|
|
|
assert.Equal(t, 10, NewManifest(nil, nil, "", "").MaxRetryDelay())
|
|
|
|
}
|
|
|
|
|
2016-12-13 20:06:58 +00:00
|
|
|
func TestRetryCounterDefaultsToFixedRetries(t *testing.T) {
|
|
|
|
rc := newRetryCounter()
|
2017-02-18 20:22:49 +00:00
|
|
|
assert.Equal(t, 8, rc.MaxRetries)
|
2016-09-28 18:25:37 +00:00
|
|
|
}
|
|
|
|
|
2020-04-15 03:13:27 +00:00
|
|
|
func TestRetryCounterDefaultsToFixedRetryDelay(t *testing.T) {
|
|
|
|
rc := newRetryCounter()
|
|
|
|
assert.Equal(t, 10, rc.MaxRetryDelay)
|
|
|
|
}
|
|
|
|
|
2016-09-28 18:25:37 +00:00
|
|
|
func TestRetryCounterIncrementsObjects(t *testing.T) {
|
2016-12-13 20:06:58 +00:00
|
|
|
rc := newRetryCounter()
|
2020-04-15 03:13:27 +00:00
|
|
|
assert.Equal(t, 1, rc.Increment("oid"))
|
2016-09-28 18:25:37 +00:00
|
|
|
assert.Equal(t, 1, rc.CountFor("oid"))
|
2020-04-15 03:13:27 +00:00
|
|
|
|
|
|
|
assert.Equal(t, 2, rc.Increment("oid"))
|
|
|
|
assert.Equal(t, 2, rc.CountFor("oid"))
|
2016-09-28 18:25:37 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func TestRetryCounterCanNotRetryAfterExceedingRetryCount(t *testing.T) {
|
2016-12-13 20:06:58 +00:00
|
|
|
rc := newRetryCounter()
|
2017-02-18 20:22:49 +00:00
|
|
|
rc.MaxRetries = 1
|
2016-09-28 18:25:37 +00:00
|
|
|
rc.Increment("oid")
|
|
|
|
|
2016-12-13 20:06:58 +00:00
|
|
|
count, canRetry := rc.CanRetry("oid")
|
2016-09-28 18:25:37 +00:00
|
|
|
assert.Equal(t, 1, count)
|
|
|
|
assert.False(t, canRetry)
|
|
|
|
}
|
2017-08-25 15:56:09 +00:00
|
|
|
|
2020-04-15 03:13:27 +00:00
|
|
|
func TestRetryCounterDoesNotDelayFirstAttempt(t *testing.T) {
|
|
|
|
rc := newRetryCounter()
|
|
|
|
assert.Equal(t, time.Time{}, rc.ReadyTime("oid"))
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestRetryCounterDelaysExponentially(t *testing.T) {
|
|
|
|
rc := newRetryCounter()
|
|
|
|
start := time.Now()
|
|
|
|
|
|
|
|
rc.Increment("oid")
|
|
|
|
ready1 := rc.ReadyTime("oid")
|
|
|
|
assert.GreaterOrEqual(t, int64(ready1.Sub(start)/time.Millisecond), int64(baseRetryDelayMs))
|
|
|
|
|
|
|
|
rc.Increment("oid")
|
|
|
|
ready2 := rc.ReadyTime("oid")
|
|
|
|
assert.GreaterOrEqual(t, int64(ready2.Sub(start)/time.Millisecond), int64(2*baseRetryDelayMs))
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestRetryCounterLimitsDelay(t *testing.T) {
|
|
|
|
rc := newRetryCounter()
|
|
|
|
rc.MaxRetryDelay = 1
|
|
|
|
|
|
|
|
for i := 0; i < 4; i++ {
|
|
|
|
rc.Increment("oid")
|
|
|
|
}
|
|
|
|
|
|
|
|
rt := rc.ReadyTime("oid")
|
|
|
|
assert.WithinDuration(t, time.Now(), rt, 1*time.Second)
|
|
|
|
}
|
|
|
|
|
2017-08-25 15:56:09 +00:00
|
|
|
func TestBatchSizeReturnsBatchSize(t *testing.T) {
|
|
|
|
q := NewTransferQueue(
|
2017-10-25 17:46:37 +00:00
|
|
|
Upload, NewManifest(nil, nil, "", ""), "origin", WithBatchSize(3))
|
2017-08-25 15:56:09 +00:00
|
|
|
|
|
|
|
assert.Equal(t, 3, q.BatchSize())
|
|
|
|
}
|