test(codegen): add accId test for RoleAssumerWebIdentity

Author: siddsriv

clients/client-sts/src/defaultStsRoleAssumers.ts

@@ -90,7 +90,7 @@ export const getDefaultRoleAssumer = (
     }
     let accountId;
     try {
-      accountId = AssumedRoleUser.Arn.split(":")[4];
+      accountId = AssumedRoleUser!.Arn!.split(":")[4];
     } catch (error) {
       accountId = undefined;
     }
@@ -147,7 +147,7 @@ export const getDefaultRoleAssumerWithWebIdentity = (
     }
     let accountId;
     try {
-      accountId = AssumedRoleUser.Arn.split(":")[4];
+      accountId = AssumedRoleUser!.Arn!.split(":")[4];
     } catch (error) {
       accountId = undefined;
     }

codegen/smithy-aws-typescript-codegen/src/main/resources/software/amazon/smithy/aws/typescript/codegen/sts-client-defaultRoleAssumers.spec.ts

@@ -218,6 +218,17 @@ describe("getDefaultRoleAssumerWithWebIdentity", () => {
     });
   });
 
+  it("should return accountId in the credentials", async () => {
+    const roleAssumerWithWebIdentity = getDefaultRoleAssumerWithWebIdentity();
+    const params: AssumeRoleWithWebIdentityCommandInput = {
+      RoleArn: "arn:aws:foo",
+      RoleSessionName: "session",
+      WebIdentityToken: "token",
+    };
+    const assumedRole = await roleAssumerWithWebIdentity(params);
+    expect(assumedRole.accountId).toEqual("123456789012");
+  });
+
   it("should use the STS client middleware", async () => {
     const customMiddlewareFunction = jest.fn();
     const roleAssumerWithWebIdentity = getDefaultRoleAssumerWithWebIdentity({}, [

codegen/smithy-aws-typescript-codegen/src/main/resources/software/amazon/smithy/aws/typescript/codegen/sts-client-defaultStsRoleAssumers.ts

@@ -87,7 +87,7 @@ export const getDefaultRoleAssumer = (
     }
     let accountId;
     try {
-      accountId = AssumedRoleUser.Arn.split(":")[4];
+      accountId = AssumedRoleUser!.Arn!.split(":")[4];
     } catch (error) {
       accountId = undefined;
     }
@@ -144,7 +144,7 @@ export const getDefaultRoleAssumerWithWebIdentity = (
     }
     let accountId;
     try {
-      accountId = AssumedRoleUser.Arn.split(":")[4];
+      accountId = AssumedRoleUser!.Arn!.split(":")[4];
     } catch (error) {
       accountId = undefined;
     }