You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
117 lines
2.4 KiB
Go
117 lines
2.4 KiB
Go
package fx
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestRetry(t *testing.T) {
|
|
assert.NotNil(t, DoWithRetry(func() error {
|
|
return errors.New("any")
|
|
}))
|
|
|
|
times1 := 0
|
|
assert.Nil(t, DoWithRetry(func() error {
|
|
times1++
|
|
if times1 == defaultRetryTimes {
|
|
return nil
|
|
}
|
|
return errors.New("any")
|
|
}))
|
|
|
|
times2 := 0
|
|
assert.NotNil(t, DoWithRetry(func() error {
|
|
times2++
|
|
if times2 == defaultRetryTimes+1 {
|
|
return nil
|
|
}
|
|
return errors.New("any")
|
|
}))
|
|
|
|
total := 2 * defaultRetryTimes
|
|
times3 := 0
|
|
assert.Nil(t, DoWithRetry(func() error {
|
|
times3++
|
|
if times3 == total {
|
|
return nil
|
|
}
|
|
return errors.New("any")
|
|
}, WithRetry(total)))
|
|
}
|
|
|
|
func TestRetryWithTimeout(t *testing.T) {
|
|
assert.Nil(t, DoWithRetry(func() error {
|
|
return nil
|
|
}, WithTimeout(time.Second*10)))
|
|
|
|
times1 := 0
|
|
assert.Nil(t, DoWithRetry(func() error {
|
|
times1++
|
|
if times1 == 1 {
|
|
return errors.New("any ")
|
|
}
|
|
time.Sleep(time.Second * 3)
|
|
return nil
|
|
}, WithTimeout(time.Second*5)))
|
|
|
|
total := defaultRetryTimes
|
|
times2 := 0
|
|
assert.Nil(t, DoWithRetry(func() error {
|
|
times2++
|
|
if times2 == total {
|
|
return nil
|
|
}
|
|
time.Sleep(time.Second)
|
|
return errors.New("any")
|
|
}, WithTimeout(time.Second*(time.Duration(total)+2))))
|
|
|
|
assert.NotNil(t, DoWithRetry(func() error {
|
|
return errors.New("any")
|
|
}, WithTimeout(time.Second*5)))
|
|
}
|
|
|
|
func TestRetryWithInterval(t *testing.T) {
|
|
times1 := 0
|
|
assert.NotNil(t, DoWithRetry(func() error {
|
|
times1++
|
|
if times1 == 1 {
|
|
return errors.New("any")
|
|
}
|
|
time.Sleep(time.Second * 3)
|
|
return nil
|
|
}, WithTimeout(time.Second*5), WithInterval(time.Second*3)))
|
|
|
|
times2 := 0
|
|
assert.NotNil(t, DoWithRetry(func() error {
|
|
times2++
|
|
if times2 == 2 {
|
|
return nil
|
|
}
|
|
time.Sleep(time.Second * 3)
|
|
return errors.New("any ")
|
|
}, WithTimeout(time.Second*5), WithInterval(time.Second*3)))
|
|
|
|
}
|
|
|
|
func TestRetryCtx(t *testing.T) {
|
|
assert.NotNil(t, DoWithRetryCtx(func(ctx context.Context, retryCount int) error {
|
|
if retryCount == 0 {
|
|
return errors.New("any")
|
|
}
|
|
time.Sleep(time.Second * 3)
|
|
return nil
|
|
}, WithTimeout(time.Second*5), WithInterval(time.Second*3)))
|
|
|
|
assert.NotNil(t, DoWithRetryCtx(func(ctx context.Context, retryCount int) error {
|
|
if retryCount == 1 {
|
|
return nil
|
|
}
|
|
time.Sleep(time.Second * 3)
|
|
return errors.New("any ")
|
|
}, WithTimeout(time.Second*5), WithInterval(time.Second*3)))
|
|
}
|