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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
|
package tq
import (
"testing"
"time"
"github.com/stretchr/testify/assert"
)
func TestManifestDefaultsToFixedRetries(t *testing.T) {
assert.Equal(t, 8, NewManifest(nil, nil, "", "").MaxRetries())
}
func TestManifestDefaultsToFixedRetryDelay(t *testing.T) {
assert.Equal(t, 10, NewManifest(nil, nil, "", "").MaxRetryDelay())
}
func TestRetryCounterDefaultsToFixedRetries(t *testing.T) {
rc := newRetryCounter()
assert.Equal(t, 8, rc.MaxRetries)
}
func TestRetryCounterDefaultsToFixedRetryDelay(t *testing.T) {
rc := newRetryCounter()
assert.Equal(t, 10, rc.MaxRetryDelay)
}
func TestRetryCounterIncrementsObjects(t *testing.T) {
rc := newRetryCounter()
assert.Equal(t, 1, rc.Increment("oid"))
assert.Equal(t, 1, rc.CountFor("oid"))
assert.Equal(t, 2, rc.Increment("oid"))
assert.Equal(t, 2, rc.CountFor("oid"))
}
func TestRetryCounterCanNotRetryAfterExceedingRetryCount(t *testing.T) {
rc := newRetryCounter()
rc.MaxRetries = 1
rc.Increment("oid")
count, canRetry := rc.CanRetry("oid")
assert.Equal(t, 1, count)
assert.False(t, canRetry)
}
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)
}
func TestBatchSizeReturnsBatchSize(t *testing.T) {
q := NewTransferQueue(
Upload, NewManifest(nil, nil, "", ""), "origin", WithBatchSize(3))
assert.Equal(t, 3, q.BatchSize())
}
|
