Fix benchmark tests

This commit is contained in:
Ken Hibino
2020-01-18 10:17:39 -08:00
parent c29200b1fc
commit 207a6d2d1a
5 changed files with 28 additions and 13 deletions

View File

@@ -18,9 +18,13 @@ func BenchmarkEndToEndSimple(b *testing.B) {
const count = 100000
for n := 0; n < b.N; n++ {
b.StopTimer() // begin setup
r := setup(b)
client := NewClient(r)
bg := NewBackground(r, &Config{
setup(b)
redis := &RedisClientOpt{
Addr: redisAddr,
DB: redisDB,
}
client := NewClient(redis)
bg := NewBackground(redis, &Config{
Concurrency: 10,
RetryDelayFunc: func(n int, err error, t *Task) time.Duration {
return time.Second
@@ -55,9 +59,13 @@ func BenchmarkEndToEnd(b *testing.B) {
for n := 0; n < b.N; n++ {
b.StopTimer() // begin setup
rand.Seed(time.Now().UnixNano())
r := setup(b)
client := NewClient(r)
bg := NewBackground(r, &Config{
setup(b)
redis := &RedisClientOpt{
Addr: redisAddr,
DB: redisDB,
}
client := NewClient(redis)
bg := NewBackground(redis, &Config{
Concurrency: 10,
RetryDelayFunc: func(n int, err error, t *Task) time.Duration {
return time.Second