From 2d0afb02bc82cb150379bdc159630560ca320443 Mon Sep 17 00:00:00 2001
From: Lee Chiang Fong <lee_chiang_fong@tech.gov.sg>
Date: Thu, 9 Jan 2025 15:28:48 +0800
Subject: [PATCH] Make mock stsClient fail when region is missing

Signed-off-by: Lee Chiang Fong <lee_chiang_fong@tech.gov.sg>
---
 .../src/DefaultAwsCredentialsManager.test.ts         | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/packages/integration-aws-node/src/DefaultAwsCredentialsManager.test.ts b/packages/integration-aws-node/src/DefaultAwsCredentialsManager.test.ts
index d81f717a58..a4e13916ff 100644
--- a/packages/integration-aws-node/src/DefaultAwsCredentialsManager.test.ts
+++ b/packages/integration-aws-node/src/DefaultAwsCredentialsManager.test.ts
@@ -91,6 +91,18 @@ describe('DefaultAwsCredentialsManager', () => {
       Account: '123456789012',
     });
 
+    stsMock
+      .on(GetCallerIdentityCommand)
+      .callsFake(async (_input, getClient) => {
+        const client = getClient();
+        const region = await client.config.region();
+        if (!region) {
+          throw new Error('Region is missing');
+        }
+        return {
+          Account: '123456789012',
+        };
+      });
     stsMock
       .on(AssumeRoleCommand, {
         RoleArn: 'arn:aws:iam::111111111111:role/hello',