Add retry logic when configuring the Go SDK

auth0 · Jun 22, 2023 · f8aa35d · f8aa35d
1 parent 904e472
commit f8aa35d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/internal/acctest/acctest.go b/internal/acctest/acctest.go
@@ -2,6 +2,7 @@ package acctest
 
 import (
 	"context"
+	"net/http"
 	"os"
 	"testing"
 
@@ -65,6 +66,7 @@ func configureTestProviderWithHTTPRecordings(httpRecorder *recorder.Recorder) sc
 			management.WithStaticToken("insecure"),
 			management.WithClient(httpRecorder.GetDefaultClient()),
 			management.WithDebug(debug),
+			management.WithRetries(3, []int{http.StatusTooManyRequests, http.StatusInternalServerError}),
 		}
 
 		if domain != RecordingsDomain {

diff --git a/internal/config/config.go b/internal/config/config.go
@@ -3,6 +3,7 @@ package config
 import (
 	"context"
 	"fmt"
+	"net/http"
 
 	"github.com/auth0/go-auth0"
 	"github.com/auth0/go-auth0/management"
@@ -59,6 +60,7 @@ func ConfigureProvider(terraformVersion *string) schema.ConfigureContextFunc {
 			management.WithDebug(debug),
 			management.WithUserAgent(userAgent(terraformVersion)),
 			management.WithAuth0ClientEnvEntry(providerName, version),
+			management.WithRetries(3, []int{http.StatusTooManyRequests, http.StatusInternalServerError}),
 		)
 		if err != nil {
 			return nil, diag.FromErr(err)