@@ -2,6 +2,7 @@ package integration
22
33import (
44 "fmt"
5+ awsClient "github.com/aws/aws-sdk-go/aws/client"
56 "os"
67 "strings"
78 "time"
@@ -50,7 +51,18 @@ var _ = Describe("RAM Share", Ordered, func() {
5051 BeforeAll (func () {
5152 secondaryTestRoleArn = os .Getenv ("SECONDARY_ACCOUNT_TEST_ROLE_ARN" )
5253
53- primarySess := session .Must (session .NewSession (& aws.Config {Region : aws .String (config .Region )}))
54+ retryer := awsClient.DefaultRetryer {
55+ MinThrottleDelay : 1 * time .Second ,
56+ MinRetryDelay : 1 * time .Second ,
57+ MaxThrottleDelay : 5 * time .Second ,
58+ MaxRetryDelay : 5 * time .Second ,
59+ }
60+
61+ primarySess := session .Must (session .NewSession (& aws.Config {
62+ Region : aws .String (config .Region ),
63+ Retryer : retryer ,
64+ }))
65+
5466 stsClient := sts .New (primarySess )
5567 assumeRoleInput := & sts.AssumeRoleInput {
5668 RoleArn : aws .String (secondaryTestRoleArn ),
@@ -63,7 +75,8 @@ var _ = Describe("RAM Share", Ordered, func() {
6375 creds := assumeRoleResult .Credentials
6476
6577 secondarySess := session .Must (session .NewSession (& aws.Config {
66- Region : aws .String (config .Region ),
78+ Region : aws .String (config .Region ),
79+ Retryer : retryer ,
6780 Credentials : credentials .NewStaticCredentials (
6881 * creds .AccessKeyId ,
6982 * creds .SecretAccessKey ,
0 commit comments