From 49fb620f4a0f8f0a9af67f7293f30a342e670df9 Mon Sep 17 00:00:00 2001
From: George Fu <kuhe@users.noreply.github.com>
Date: Wed, 30 Oct 2024 21:54:23 +0000
Subject: [PATCH] test: what is even going on

---
 tests/e2e/delete-stale-changesets.js  |  2 +-
 tests/e2e/ensure-test-stack.js        |  2 +-
 tests/e2e/get-integ-test-resources.js | 20 +++++++++++---------
 3 files changed, 13 insertions(+), 11 deletions(-)

diff --git a/tests/e2e/delete-stale-changesets.js b/tests/e2e/delete-stale-changesets.js
index 45634c3a0541d..cb523d84fe2c1 100644
--- a/tests/e2e/delete-stale-changesets.js
+++ b/tests/e2e/delete-stale-changesets.js
@@ -1,4 +1,4 @@
-const { ListChangeSetsCommand, DeleteChangeSetCommand } = require("../../clients/client-cloudformation");
+const { ListChangeSetsCommand, DeleteChangeSetCommand } = require("@aws-sdk/client-cloudformation");
 
 exports.deleteStaleChangesets = async (client, stackName) => {
   const changesets = await client.send(
diff --git a/tests/e2e/ensure-test-stack.js b/tests/e2e/ensure-test-stack.js
index 5c153b11423d9..00daa44c8af84 100644
--- a/tests/e2e/ensure-test-stack.js
+++ b/tests/e2e/ensure-test-stack.js
@@ -8,7 +8,7 @@ const {
   waitUntilStackCreateComplete,
   DescribeChangeSetCommand,
   DeleteChangeSetCommand,
-} = require("../../clients/client-cloudformation");
+} = require("@aws-sdk/client-cloudformation");
 
 /**
  * Deploy the integration test stack if it does not exist. Update the
diff --git a/tests/e2e/get-integ-test-resources.js b/tests/e2e/get-integ-test-resources.js
index 85dd86cbed769..d8eda250351d2 100644
--- a/tests/e2e/get-integ-test-resources.js
+++ b/tests/e2e/get-integ-test-resources.js
@@ -1,31 +1,33 @@
 const { readFileSync } = require("fs");
 const { join } = require("path");
-const { STSClient, GetCallerIdentityCommand } = require("../../clients/client-sts");
-const { CloudFormationClient, DescribeStackResourcesCommand } = require("../../clients/client-cloudformation");
-const { S3ControlClient, ListMultiRegionAccessPointsCommand } = require("../../clients/client-s3-control");
+const { STSClient, GetCallerIdentityCommand } = require("@aws-sdk/client-sts");
+const { CloudFormationClient, DescribeStackResourcesCommand } = require("@aws-sdk/client-cloudformation");
+const { S3ControlClient, ListMultiRegionAccessPointsCommand } = require("@aws-sdk/client-s3-control");
 const { ensureTestStack } = require("./ensure-test-stack");
 const { deleteStaleChangesets } = require("./delete-stale-changesets");
 const { loadSharedConfigFiles } = require("@smithy/shared-ini-file-loader");
-const { fromIni } = require("@aws-sdk/credential-providers");
+const { createCredentialChain, fromIni, fromHttp } = require("@aws-sdk/credential-providers");
 
 exports.getIntegTestResources = async () => {
   const ini = await loadSharedConfigFiles();
-  if (ini.configFile["sdk-integ-test"] || ini.credentialsFile["sdk-integ-test"]) {
+  const profileData = ini.configFile["sdk-integ-test"] ?? ini.credentialsFile["sdk-integ-test"];
+  if (profileData) {
     process.env.AWS_PROFILE = "sdk-integ-test";
     console.log("Setting AWS_PROFILE=sdk-integ-test");
   } else {
     console.log("AWS_PROFILE is", process.env.AWS_PROFILE);
   }
 
-  // TODO(debug)
-  console.log({
-    config: ini.configFile,
-    credentials: ini.credentialsFile,
+  const credentials = fromIni({
+    ...(profileData.role_arn ? { awsContainerCredentialsRelativeUri: `/role-arn/${profileData.role_arn}` } : {}),
+    profile: process.env.AWS_PROFILE ?? "default",
+    logger: console,
   });
 
   const region = "us-west-2";
   const cloudformation = new CloudFormationClient({
     region,
+    credentials,
   });
   const stackName = "SdkReleaseV3IntegTestResourcesStack";