From d7d29e0e90aef1062348fa68cf25eeb5360db2bb Mon Sep 17 00:00:00 2001 From: Dan Sun Date: Thu, 6 Feb 2025 08:19:49 -0500 Subject: [PATCH] test: Support AWS Session Token in real provider test Signed-off-by: Dan Sun --- tests/extproc/real_providers_test.go | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/tests/extproc/real_providers_test.go b/tests/extproc/real_providers_test.go index 893bb5bce..b6332184f 100644 --- a/tests/extproc/real_providers_test.go +++ b/tests/extproc/real_providers_test.go @@ -46,7 +46,15 @@ func TestWithRealProviders(t *testing.T) { // Set up credential file for AWS. awsAccessKeyID := getEnvVarOrSkip(t, "TEST_AWS_ACCESS_KEY_ID") awsSecretAccessKey := getEnvVarOrSkip(t, "TEST_AWS_SECRET_ACCESS_KEY") - awsCredentialsBody := fmt.Sprintf("[default]\nAWS_ACCESS_KEY_ID=%s\nAWS_SECRET_ACCESS_KEY=%s\n", awsAccessKeyID, awsSecretAccessKey) + awsSessionToken := os.Getenv("TEST_AWS_SESSION_TOKEN") + var awsCredentialsBody string + if awsSessionToken != "" { + awsCredentialsBody = fmt.Sprintf("[default]\nAWS_ACCESS_KEY_ID=%s\nAWS_SECRET_ACCESS_KEY=%s\nTEST_AWS_SESSION_TOKEN=%s\n", + awsAccessKeyID, awsSecretAccessKey, awsSessionToken) + } else { + awsCredentialsBody = fmt.Sprintf("[default]\nAWS_ACCESS_KEY_ID=%s\nAWS_SECRET_ACCESS_KEY=%s\n", + awsAccessKeyID, awsSecretAccessKey) + } // Test with AWS Credential File. awsFilePath := t.TempDir() + "/aws-credential-file"