add backoff handling to SDK

Kong · Dec 23, 2024 · f1deb7e · f1deb7e
1 parent 02be60a
commit f1deb7e
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/test/internal/helpers/konnect/control_plane.go b/test/internal/helpers/konnect/control_plane.go
@@ -9,7 +9,8 @@ import (
 	"testing"
 	"time"
 
-	"github.com/avast/retry-go/v4"
+	sdkkonnectgo "github.com/Kong/sdk-konnect-go"
+	"github.com/Kong/sdk-konnect-go/retry"
 	"github.com/google/uuid"
 	"github.com/samber/lo"
 	"github.com/stretchr/testify/assert"
@@ -30,7 +31,16 @@ import (
 func CreateTestControlPlane(ctx context.Context, t *testing.T) string {
 	t.Helper()
 
-	sdk := sdk.New(accessToken(), serverURLOpt())
+	sdk := sdk.New(accessToken(), serverURLOpt(),
+		sdkkonnectgo.WithRetryConfig(retry.Config{
+			Backoff: &retry.BackoffStrategy{
+				InitialInterval: 100,
+				MaxInterval:     2000,
+				Exponent:        1.2,
+				MaxElapsedTime:  10000,
+			},
+		}),
+	)
 
 	var cpID string
 	createRgErr := retry.Do(func() error {