initial import

core/fx/retry_test.go

@@ -0,0 +1,42 @@
+package fx
+
+import (
+	"errors"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestRetry(t *testing.T) {
+	assert.NotNil(t, DoWithRetries(func() error {
+		return errors.New("any")
+	}))
+
+	var times int
+	assert.Nil(t, DoWithRetries(func() error {
+		times++
+		if times == defaultRetryTimes {
+			return nil
+		}
+		return errors.New("any")
+	}))
+
+	times = 0
+	assert.NotNil(t, DoWithRetries(func() error {
+		times++
+		if times == defaultRetryTimes+1 {
+			return nil
+		}
+		return errors.New("any")
+	}))
+
+	var total = 2 * defaultRetryTimes
+	times = 0
+	assert.Nil(t, DoWithRetries(func() error {
+		times++
+		if times == total {
+			return nil
+		}
+		return errors.New("any")
+	}, WithRetries(total)))
+}