add tests to retry mechanism

utils/backoff_test.go

@@ -0,0 +1,55 @@
+package utils
+
+import (
+	"errors"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestWithMaxRetries(t *testing.T) {
+	t.Run("NotEnoughRetry", func(t *testing.T) {
+		retryable := newMockRetryableFn(3)
+		err := WithMaxRetries(
+			func() (err error) {
+				_, err = retryable.Run()
+				return err
+			},
+			2,
+		)
+		require.Error(t, err)
+	})
+	t.Run("EnoughRetry", func(t *testing.T) {
+		retryable := newMockRetryableFn(2)
+		var res bool
+		err := WithMaxRetries(
+			func() (err error) {
+				res, err = retryable.Run()
+				return err
+			},
+			2,
+		)
+		require.NoError(t, err)
+		require.True(t, res)
+	})
+}
+
+type mockRetryableFn struct {
+	counter uint64
+	trigger uint64
+}
+
+func newMockRetryableFn(trigger uint64) mockRetryableFn {
+	return mockRetryableFn{
+		counter: 0,
+		trigger: trigger,
+	}
+}
+
+func (m *mockRetryableFn) Run() (bool, error) {
+	if m.counter == m.trigger {
+		return true, nil
+	}
+	m.counter++
+	return false, errors.New("error")
+}