INTPYTHON-622 Fix handling of assumed credentials (#69)

Author: blink1073

.evergreen/config.yml

@@ -52,7 +52,7 @@ functions:
   "assume role":
     - command: ec2.assume_role
       params:
-        role_arn: ${assume_role_arn}
+        role_arn: ${drivers_test_secrets_role}
 
   "execute tests":
     - command: subprocess.exec
@@ -68,7 +68,7 @@ functions:
       type: test
       retry_on_failure: true
       params:
-        include_expansions_in_env: [DIR]
+        include_expansions_in_env: [DIR, AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY, AWS_SESSION_TOKEN]
         working_dir: "src"
         binary: bash
         args:
@@ -78,14 +78,15 @@ functions:
     - command: subprocess.exec
       type: test
       params:
-        include_expansions_in_env: [DIR]
+        include_expansions_in_env: [DIR, AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY, AWS_SESSION_TOKEN]
         working_dir: "src"
         binary: bash
         args: [.evergreen/setup-remote.sh]
 
 pre_error_fails_task: true
 post_error_fails_task: true
 pre:
+  - func: "assume role"
   - func: "fetch source"
   - func: "fetch secrets"
 post:
@@ -198,15 +199,13 @@ tasks:
     tags: [local]
     commands:
       - func: "fetch repo"
-      - func: "assume role"
       - func: "setup local atlas"
       - func: "execute tests"
 
   - name: test-pymongo-voyageai-remote
     tags: [remote]
     commands:
       - func: "fetch repo"
-      - func: "assume role"
       - func: "setup remote atlas"
       - func: "execute tests"