Add setup-tests.sh

.evergreen/config.yml

@@ -227,7 +227,7 @@ functions:
  - command: subprocess.exec
  params:
  binary: bash
- include_expansions_in_env: ["VERSION", "TOPOLOGY"]
+ include_expansions_in_env: ["VERSION", "TOPOLOGY", "ORCHESTRATION_FILE"]
  args:
  - src/.evergreen/scripts/run-with-env.sh
  - src/.evergreen/scripts/bootstrap-mongo-orchestration.sh
@@ -293,6 +293,13 @@ functions:
  - .evergreen/scripts/run-doctests.sh
 
  "run tests":
+ - command: subprocess.exec
+ params:
+ include_expansions_in_env: ["TEST_DATA_LAKE", "TEST_INDEX_MANAGEMENT", "CRYPT_SHARED_LIB_PATH", "test_encryption", "test_encryption_pyopenssl", "test_crypt_shared", "test_pyopenssl", "test_loadbalancer", "test_serverless", "ORCHESTRATION_FILE", "SINGLE_MONGOS_LB_URI", "MULTI_MONGOS_LB_URI"]
+ binary: bash
+ working_dir: "src"
+ args:
+ - .evergreen/scripts/setup-tests.sh
  - command: subprocess.exec
  params:
  working_dir: "src"
@@ -343,6 +350,13 @@ functions:
  - ${DRIVERS_TOOLS}/.evergreen/auth_aws/setup-secrets.sh
 
  "run aws auth test with regular aws credentials":
+ - command: subprocess.exec
+ params:
+ include_expansions_in_env: ["TEST_DATA_LAKE", "TEST_INDEX_MANAGEMENT", "CRYPT_SHARED_LIB_PATH", "test_encryption", "test_encryption_pyopenssl", "test_crypt_shared", "test_pyopenssl", "test_loadbalancer", "test_serverless", "ORCHESTRATION_FILE"]
+ binary: bash
+ working_dir: "src"
+ args:
+ - .evergreen/scripts/setup-tests.sh
  - command: subprocess.exec
  type: test
  params:
@@ -355,6 +369,13 @@ functions:
  - regular
 
  "run aws auth test with assume role credentials":
+ - command: subprocess.exec
+ params:
+ include_expansions_in_env: [ "TEST_DATA_LAKE", "TEST_INDEX_MANAGEMENT", "CRYPT_SHARED_LIB_PATH", "test_encryption", "test_encryption_pyopenssl", "test_crypt_shared", "test_pyopenssl", "test_loadbalancer", "test_serverless", "ORCHESTRATION_FILE" ]
+ binary: bash
+ working_dir: "src"
+ args:
+ - .evergreen/scripts/setup-tests.sh
  - command: subprocess.exec
  type: test
  params:
@@ -367,6 +388,13 @@ functions:
  - assume-role
 
  "run aws auth test with aws EC2 credentials":
+ - command: subprocess.exec
+ params:
+ include_expansions_in_env: [ "TEST_DATA_LAKE", "TEST_INDEX_MANAGEMENT", "CRYPT_SHARED_LIB_PATH", "test_encryption", "test_encryption_pyopenssl", "test_crypt_shared", "test_pyopenssl", "test_loadbalancer", "test_serverless", "ORCHESTRATION_FILE" ]
+ binary: bash
+ working_dir: "src"
+ args:
+ - .evergreen/scripts/setup-tests.sh
  - command: subprocess.exec
  type: test
  params:
@@ -379,6 +407,13 @@ functions:
  - ec2
 
  "run aws auth test with aws web identity credentials":
+ - command: subprocess.exec
+ params:
+ include_expansions_in_env: [ "TEST_DATA_LAKE", "TEST_INDEX_MANAGEMENT", "CRYPT_SHARED_LIB_PATH", "test_encryption", "test_encryption_pyopenssl", "test_crypt_shared", "test_pyopenssl", "test_loadbalancer", "test_serverless", "ORCHESTRATION_FILE" ]
+ binary: bash
+ working_dir: "src"
+ args:
+ - .evergreen/scripts/setup-tests.sh
  - # Test with and without AWS_ROLE_SESSION_NAME set.
  - command: subprocess.exec
  type: test
@@ -402,6 +437,13 @@ functions:
  - web-identity
 
  "run aws auth test with aws credentials as environment variables":
+ - command: subprocess.exec
+ params:
+ include_expansions_in_env: [ "TEST_DATA_LAKE", "TEST_INDEX_MANAGEMENT", "CRYPT_SHARED_LIB_PATH", "test_encryption", "test_encryption_pyopenssl", "test_crypt_shared", "test_pyopenssl", "test_loadbalancer", "test_serverless", "ORCHESTRATION_FILE" ]
+ binary: bash
+ working_dir: "src"
+ args:
+ - .evergreen/scripts/setup-tests.sh
  - command: subprocess.exec
  type: test
  params:
@@ -414,6 +456,13 @@ functions:
  - env-creds
 
  "run aws auth test with aws credentials and session token as environment variables":
+ - command: subprocess.exec
+ params:
+ include_expansions_in_env: [ "TEST_DATA_LAKE", "TEST_INDEX_MANAGEMENT", "CRYPT_SHARED_LIB_PATH", "test_encryption", "test_encryption_pyopenssl", "test_crypt_shared", "test_pyopenssl", "test_loadbalancer", "test_serverless", "ORCHESTRATION_FILE" ]
+ binary: bash
+ working_dir: "src"
+ args:
+ - .evergreen/scripts/setup-tests.sh
  - command: subprocess.exec
  type: test
  params:
@@ -426,6 +475,13 @@ functions:
  - session-creds
 
  "run oidc auth test with test credentials":
+ - command: subprocess.exec
+ params:
+ include_expansions_in_env: [ "TEST_DATA_LAKE", "TEST_INDEX_MANAGEMENT", "CRYPT_SHARED_LIB_PATH", "test_encryption", "test_encryption_pyopenssl", "test_crypt_shared", "test_pyopenssl", "test_loadbalancer", "test_serverless", "ORCHESTRATION_FILE" ]
+ binary: bash
+ working_dir: "src"
+ args:
+ - .evergreen/scripts/setup-tests.sh
  - command: subprocess.exec
  type: test
  params:
@@ -523,6 +579,13 @@ functions:
  file: atlas-expansion.yml
 
  "run-ocsp-test":
+ - command: subprocess.exec
+ params:
+ include_expansions_in_env: [ "TEST_DATA_LAKE", "TEST_INDEX_MANAGEMENT", "CRYPT_SHARED_LIB_PATH", "test_encryption", "test_encryption_pyopenssl", "test_crypt_shared", "test_pyopenssl", "test_loadbalancer", "test_serverless", "ORCHESTRATION_FILE" ]
+ binary: bash
+ working_dir: "src"
+ args:
+ - .evergreen/scripts/setup-tests.sh
  - command: subprocess.exec
  type: test
  params:

.evergreen/scripts/bootstrap-mongo-orchestration.sh

@@ -39,7 +39,7 @@ MONGODB_VERSION=${VERSION} \
  SSL=${SSL} \
  STORAGE_ENGINE=${STORAGE_ENGINE} \
  DISABLE_TEST_COMMANDS=${DISABLE_TEST_COMMANDS} \
- ORCHESTRATION_FILE=${ORCHESTRATION_FILE} \
+ ORCHESTRATION_FILE=${ORCHESTRATION_FILE:-} \
  REQUIRE_API_VERSION=${REQUIRE_API_VERSION} \
  LOAD_BALANCER=${LOAD_BALANCER} \
  bash ${DRIVERS_TOOLS}/.evergreen/run-orchestration.sh

.evergreen/scripts/configure-env.sh

@@ -54,15 +54,8 @@ export MONGODB_API_VERSION="${MONGODB_API_VERSION:-}"
 export SKIP_HATCH="${SKIP_HATCH:-}"
 export skip_crypt_shared="${skip_crypt_shared:-}"
 export STORAGE_ENGINE="${STORAGE_ENGINE:-}"
-export ORCHESTRATION_FILE="${ORCHESTRATION_FILE:-}"
 export REQUIRE_API_VERSION="${REQUIRE_API_VERSION:-}"
 export LOAD_BALANCER="${LOAD_BALANCER:-}"
-test_encryption="${test_encryption:-}"
-test_encryption_pyopenssl="${test_encryption_pyopenssl:-}"
-test_crypt_shared="${test_crypt_shared:-}"
-test_pyopenssl="${test_pyopenssl:-}"
-test_loadbalancer="${test_loadbalancer:-}"
-test_serverless="${test_serverless:-}"
 
 export TMPDIR="$MONGO_ORCHESTRATION_HOME/db"
 export PATH="$MONGODB_BINARIES:$PATH"

.evergreen/scripts/run-with-env.sh

@@ -1,12 +1,18 @@
-#!/bin/bash
+#!/bin/bash -eu
 
 # Example use: bash run-with-env.sh run-tests.sh {args...}
 
 # Parameter expansion to get just the current directory's name
 if [ "${PWD##*/}" == "src" ]; then
  . .evergreen/scripts/env.sh
+ if [ -f ".evergreen/scripts/test-env.sh" ]; then
+ . .evergreen/scripts/test-env.sh
+ fi
 else
  . src/.evergreen/scripts/env.sh
+ if [ -f "src/.evergreen/scripts/test-env.sh" ]; then
+ . src/.evergreen/scripts/test-env.sh
+ fi
 fi
 
 set -eu

.evergreen/scripts/setup-tests.sh

@@ -0,0 +1,25 @@
+#!/bin/bash -eux
+
+PROJECT_DIRECTORY="$(pwd)"
+SCRIPT_DIR="$PROJECT_DIRECTORY/.evergreen/scripts"
+
+if [ -f "$SCRIPT_DIR/test-env.sh" ]; then
+ echo "Reading $SCRIPT_DIR/test-env.sh file"
+ . "$SCRIPT_DIR/test-env.sh"
+ exit 0
+fi
+
+cat <<EOT > "$SCRIPT_DIR"/test-env.sh
+echo "Sourcing test-env.sh now, ORCHESTRATION_FILE: $ORCHESTRATION_FILE"
+export test_encryption="${test_encryption:-}"
+export test_encryption_pyopenssl="${test_encryption_pyopenssl:-}"
+export test_crypt_shared="${test_crypt_shared:-}"
+export test_pyopenssl="${test_pyopenssl:-}"
+export test_loadbalancer="${test_loadbalancer:-}"
+export test_serverless="${test_serverless:-}"
+export TEST_INDEX_MANAGEMENT="${TEST_INDEX_MANAGEMENT:-}"
+export TEST_DATA_LAKE="${TEST_DATA_LAKE:-}"
+export ORCHESTRATION_FILE="${ORCHESTRATION_FILE:-}"
+EOT
+
+chmod +x "$SCRIPT_DIR"/test-env.sh