Adds test for SourceIdentity with AssumeRole

Author: gdavison

aws_config_test.go

@@ -208,6 +208,25 @@ func TestGetAwsConfig(t *testing.T) {
 				servicemocks.MockStsGetCallerIdentityValidEndpoint,
 			},
 		},
+		{
+			Config: &Config{
+				AccessKey: servicemocks.MockStaticAccessKey,
+				AssumeRole: &AssumeRole{
+					RoleARN:        servicemocks.MockStsAssumeRoleArn,
+					SessionName:    servicemocks.MockStsAssumeRoleSessionName,
+					SourceIdentity: servicemocks.MockStsAssumeRoleSourceIdentity,
+				},
+				Region:    "us-east-1",
+				SecretKey: servicemocks.MockStaticSecretKey,
+			},
+			Description:              "config AssumeRoleSourceIdentity",
+			ExpectedCredentialsValue: mockdata.MockStsAssumeRoleCredentials,
+			ExpectedRegion:           "us-east-1",
+			MockStsEndpoints: []*servicemocks.MockEndpoint{
+				servicemocks.MockStsAssumeRoleValidEndpointWithOptions(map[string]string{"SourceIdentity": servicemocks.MockStsAssumeRoleSourceIdentity}),
+				servicemocks.MockStsGetCallerIdentityValidEndpoint,
+			},
+		},
 		{
 			Config: &Config{
 				Profile: "SharedCredentialsProfile",

servicemocks/mock.go

@@ -51,6 +51,7 @@ const (
 	MockStsAssumeRoleSecretKey         = `AssumeRoleSecretKey`
 	MockStsAssumeRoleSessionName       = `AssumeRoleSessionName`
 	MockStsAssumeRoleSessionToken      = `AssumeRoleSessionToken`
+	MockStsAssumeRoleSourceIdentity    = `AssumeRoleSourceIdentity`
 	MockStsAssumeRoleTagKey            = `AssumeRoleTagKey`
 	MockStsAssumeRoleTagValue          = `AssumeRoleTagValue`
 	MockStsAssumeRoleTransitiveTagKey  = `AssumeRoleTagKey`

v2/awsv1shim/session_test.go

@@ -253,6 +253,25 @@ func TestGetSession(t *testing.T) {
 				servicemocks.MockStsGetCallerIdentityValidEndpoint,
 			},
 		},
+		{
+			Config: &awsbase.Config{
+				AccessKey: servicemocks.MockStaticAccessKey,
+				AssumeRole: &awsbase.AssumeRole{
+					RoleARN:        servicemocks.MockStsAssumeRoleArn,
+					SessionName:    servicemocks.MockStsAssumeRoleSessionName,
+					SourceIdentity: servicemocks.MockStsAssumeRoleSourceIdentity,
+				},
+				Region:    "us-east-1",
+				SecretKey: servicemocks.MockStaticSecretKey,
+			},
+			Description:              "config AssumeRoleSourceIdentity",
+			ExpectedCredentialsValue: mockdata.MockStsAssumeRoleCredentials,
+			ExpectedRegion:           "us-east-1",
+			MockStsEndpoints: []*servicemocks.MockEndpoint{
+				servicemocks.MockStsAssumeRoleValidEndpointWithOptions(map[string]string{"SourceIdentity": servicemocks.MockStsAssumeRoleSourceIdentity}),
+				servicemocks.MockStsGetCallerIdentityValidEndpoint,
+			},
+		},
 		{
 			Config: &awsbase.Config{
 				Profile: "SharedCredentialsProfile",