From 2c43eb7efedddcfef541cce0eb0130b90352bd7d Mon Sep 17 00:00:00 2001
From: Joey Zhao <5253430+joeyzhao2018@users.noreply.github.com>
Date: Mon, 28 Aug 2023 16:34:55 -0400
Subject: [PATCH] Update aws-vault usage to sso

---
 scripts/publish_prod.sh          | 6 +++---
 scripts/publish_sandbox.sh       | 4 ++--
 scripts/run_integration_tests.sh | 4 ++--
 3 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/scripts/publish_prod.sh b/scripts/publish_prod.sh
index a6ea0dee..e8838c5a 100755
--- a/scripts/publish_prod.sh
+++ b/scripts/publish_prod.sh
@@ -34,7 +34,7 @@ fi
 # Ensure AWS access before proceeding
 ddsaml2aws login -a govcloud-us1-fed-human-engineering
 AWS_PROFILE=govcloud-us1-fed-human-engineering aws sts get-caller-identity
-aws-vault exec prod-engineering -- aws sts get-caller-identity
+aws-vault exec sso-prod-engineering -- aws sts get-caller-identity
 
 # Ensure pypi registry access
 read -p "Do you have access to PyPI (y/n)?" CONT
@@ -64,11 +64,11 @@ echo "Building layers..."
 
 echo
 echo "Signing layers for commercial AWS regions"
-aws-vault exec prod-engineering -- ./scripts/sign_layers.sh prod
+aws-vault exec sso-prod-engineering -- ./scripts/sign_layers.sh prod
 
 echo
 echo "Publishing layers to commercial AWS regions"
-VERSION=$LAYER_VERSION aws-vault exec prod-engineering -- ./scripts/publish_layers.sh
+VERSION=$LAYER_VERSION aws-vault exec sso-prod-engineering -- ./scripts/publish_layers.sh
 
 echo "Publishing layers to GovCloud AWS regions"
 ddsaml2aws login -a govcloud-us1-fed-human-engineering
diff --git a/scripts/publish_sandbox.sh b/scripts/publish_sandbox.sh
index 4dec5889..8f7d1de5 100755
--- a/scripts/publish_sandbox.sh
+++ b/scripts/publish_sandbox.sh
@@ -4,8 +4,8 @@
 set -e
 
 ./scripts/build_layers.sh
-aws-vault exec serverless-sandbox-account-admin -- ./scripts/sign_layers.sh sandbox
-aws-vault exec serverless-sandbox-account-admin -- ./scripts/publish_layers.sh
+aws-vault exec sso-serverless-sandbox-account-admin -- ./scripts/sign_layers.sh sandbox
+aws-vault exec sso-serverless-sandbox-account-admin -- ./scripts/publish_layers.sh
 
 # Automatically create PR against github.com/DataDog/documentation
 # If you'd like to test, please uncomment the below line
diff --git a/scripts/run_integration_tests.sh b/scripts/run_integration_tests.sh
index f3ee5f36..beb91921 100755
--- a/scripts/run_integration_tests.sh
+++ b/scripts/run_integration_tests.sh
@@ -2,9 +2,9 @@
 
 # Usage - run commands from repo root:
 # To check if new changes to the layer cause changes to any snapshots:
-#   BUILD_LAYERS=true DD_API_KEY=XXXX aws-vault exec serverless-sandbox-account-admin -- ./scripts/run_integration_tests
+#   BUILD_LAYERS=true DD_API_KEY=XXXX aws-vault exec sso-serverless-sandbox-account-admin -- ./scripts/run_integration_tests
 # To regenerate snapshots:
-#   UPDATE_SNAPSHOTS=true DD_API_KEY=XXXX aws-vault exec serverless-sandbox-account-admin -- ./scripts/run_integration_tests
+#   UPDATE_SNAPSHOTS=true DD_API_KEY=XXXX aws-vault exec sso-serverless-sandbox-account-admin -- ./scripts/run_integration_tests
 
 set -e