2016-12-12 00:20:14 +00:00
|
|
|
package tq
|
2016-09-28 18:25:37 +00:00
|
|
|
|
|
|
|
import (
|
|
|
|
"testing"
|
|
|
|
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
)
|
|
|
|
|
2016-12-13 20:06:58 +00:00
|
|
|
func TestManifestDefaultsToFixedRetries(t *testing.T) {
|
2017-02-18 20:22:49 +00:00
|
|
|
assert.Equal(t, 8, NewManifest().MaxRetries())
|
2016-12-13 20:06:58 +00:00
|
|
|
}
|
2016-09-28 18:25:37 +00:00
|
|
|
|
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
|
|
|
}
|
|
|
|
|
|
|
|
func TestRetryCounterIncrementsObjects(t *testing.T) {
|
2016-12-13 20:06:58 +00:00
|
|
|
rc := newRetryCounter()
|
2016-09-28 18:25:37 +00:00
|
|
|
rc.Increment("oid")
|
|
|
|
assert.Equal(t, 1, rc.CountFor("oid"))
|
|
|
|
}
|
|
|
|
|
|
|
|
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
|
|
|
|
|
|
|
func TestBatchSizeReturnsBatchSize(t *testing.T) {
|
|
|
|
q := NewTransferQueue(
|
2017-08-25 16:11:52 +00:00
|
|
|
Upload, NewManifest(), "origin", WithBatchSize(3))
|
2017-08-25 15:56:09 +00:00
|
|
|
|
|
|
|
assert.Equal(t, 3, q.BatchSize())
|
|
|
|
}
|