@@ -46,14 +46,18 @@ func (tc testCase) Run(t *testing.T) {
4646 require .Error (t , err )
4747 } else {
4848 require .NoError (t , err )
49- tc .assertConfig (t , cfg )
50- creds , _ := cfg .Credentials .Retrieve (ctx )
51- if tc .authSettings .GetAuthType () == AuthTypeKeys && tc .authSettings .SessionToken != "" {
52- assert .Equal (t , tc .authSettings .SessionToken , creds .SessionToken )
49+ creds , err := cfg .Credentials .Retrieve (ctx )
50+ if tc .assumeRoleShouldFail {
51+ require .Error (t , err )
52+ } else {
53+ tc .assertConfig (t , cfg )
54+ if tc .authSettings .GetAuthType () == AuthTypeKeys && tc .authSettings .SessionToken != "" {
55+ assert .Equal (t , tc .authSettings .SessionToken , creds .SessionToken )
56+ }
57+ accessKey , secret := tc .getExpectedKeyAndSecret (t )
58+ assert .Equal (t , accessKey , creds .AccessKeyID )
59+ assert .Equal (t , secret , creds .SecretAccessKey )
5360 }
54- accessKey , secret := tc .getExpectedKeyAndSecret (t )
55- assert .Equal (t , accessKey , creds .AccessKeyID )
56- assert .Equal (t , secret , creds .SecretAccessKey )
5761 }
5862 if isStsEndpoint (& tc .authSettings .Endpoint ) {
5963 assert .Equal (t , tc .authSettings .Endpoint , * client .assumeRoleClient .stsConfig .BaseEndpoint )
@@ -208,7 +212,6 @@ func TestGetAWSConfig_Keys_AssumeRule(t *testing.T) {
208212 AssumeRoleARN : "arn:aws:iam::1234567890:role/aws-service-role" ,
209213 },
210214 assumeRoleShouldFail : true ,
211- shouldError : true ,
212215 },
213216 }.runAll (t )
214217}
0 commit comments