PYTHON-5138 Convert setup_tests.py to a cli (#2154)

Author: blink1073

.evergreen/config.yml

@@ -42,7 +42,7 @@ functions:
     # Make an evergreen expansion file with dynamic values
     - command: subprocess.exec
       params:
-        include_expansions_in_env: ["is_patch", "project", "version_id", "AUTH", "SSL", "TEST_ENCRYPTION", "TEST_ENCRYPTION_PYOPENSSL", "TEST_CRYPT_SHARED", "TEST_PYOPENSSL", "SETDEFAULTENCODING", "TEST_LOADBALANCER", "TEST_SEVERLESS", "SKIP_CSOT_TESTS", "MONGODB_STARTED", "DISABLE_TEST_COMMANDS", "GREEN_FRAMEWORK", "NO_EXT", "COVERAGE", "COMPRESSORS", "MONGODB_API_VERSION", "skip_crypt_shared", "VERSION", "TOPOLOGY", "STORAGE_ENGINE", "ORCHESTRATION_FILE", "REQUIRE_API_VERSION", "LOAD_BALANCER", "skip_web_identity_auth_test", "skip_ECS_auth_test"]
+        include_expansions_in_env: ["is_patch", "project", "version_id", "skip_web_identity_auth_test", "skip_ECS_auth_test"]
         binary: bash
         working_dir: "src"
         args:
@@ -205,18 +205,13 @@ functions:
     - command: subprocess.exec
       params:
         binary: bash
-        include_expansions_in_env: ["VERSION", "TOPOLOGY", "AUTH", "SSL", "ORCHESTRATION_FILE", "LOAD_BALANCER"]
+        include_expansions_in_env: [VERSION, TOPOLOGY, AUTH, SSL, ORCHESTRATION_FILE, LOAD_BALANCER,
+          STORAGE_ENGINE, REQUIRE_API_VERSION, DRIVERS_TOOLS, TEST_CRYPT_SHARED]
         args:
-          - src/.evergreen/scripts/run-with-env.sh
           - src/.evergreen/scripts/bootstrap-mongo-orchestration.sh
     - command: expansions.update
       params:
         file: mo-expansion.yml
-    - command: expansions.update
-      params:
-        updates:
-          - key: MONGODB_STARTED
-            value: "1"
 
   "bootstrap data lake":
     - command: subprocess.exec
@@ -250,17 +245,6 @@ functions:
           - .evergreen/scripts/run-with-env.sh
           - .evergreen/scripts/run-mod-wsgi-tests.sh
 
-  "run mockupdb tests":
-    - command: subprocess.exec
-      type: test
-      params:
-        include_expansions_in_env: ["PYTHON_BINARY"]
-        working_dir: "src"
-        binary: bash
-        args:
-          - .evergreen/scripts/run-with-env.sh
-          - .evergreen/scripts/run-mockupdb-tests.sh
-
   "run doctests":
      - command: subprocess.exec
        type: test
@@ -276,14 +260,12 @@ functions:
     - command: subprocess.exec
       type: test
       params:
-        include_expansions_in_env: ["TEST_DATA_LAKE", "PYTHON_BINARY", "AUTH", "SSL",
-          "AWS_ACCESS_KEY_ID", "AWS_SECRET_ACCESS_KEY", "AWS_SESSION_TOKEN", "TEST_SUITES",
-          "TEST_INDEX_MANAGEMENT", "CRYPT_SHARED_LIB_PATH", "TEST_ENCRYPTION", "TEST_ENCRYPTION_PYOPENSSL",
-          "TEST_CRYPT_SHARED", "TEST_PYOPENSSL", "TEST_LOADBALANCER", "TEST_SEVERLESS", "MONGODB_URI"]
+        include_expansions_in_env: [AUTH, SSL, AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY,
+          AWS_SESSION_TOKEN, COVERAGE, PYTHON_BINARY, LIBMONGOCRYPT_URL, MONGODB_URI,
+          DISABLE_TEST_COMMANDS, GREEN_FRAMEWORK, NO_EXT, COMPRESSORS]
         binary: bash
         working_dir: "src"
-        args:
-          - .evergreen/scripts/setup-tests.sh
+        args: [.evergreen/just.sh, setup-test, "${TEST_NAME}", "${SUB_TEST_NAME}"]
     - command: subprocess.exec
       type: test
       params:
@@ -755,7 +737,9 @@ tasks:
     - name: "mockupdb"
       tags: ["mockupdb"]
       commands:
-        - func: "run mockupdb tests"
+        - func: "run tests"
+          vars:
+            TEST_NAME: mockupdb
 
     - name: "doctests"
       tags: ["doctests"]
@@ -770,6 +754,8 @@ tasks:
       tags: ["serverless"]
       commands:
         - func: "run tests"
+          vars:
+            TEST_NAME: serverless
 
     - name: "test-enterprise-auth"
       tags: ["enterprise-auth"]
@@ -789,7 +775,7 @@ tasks:
             TOPOLOGY: "replica_set"
         - func: "run tests"
           vars:
-            TEST_INDEX_MANAGEMENT: "1"
+            TEST_NAME: index_management
             AUTH: "auth"
 
     - name: "mod-wsgi-standalone"
@@ -857,7 +843,7 @@ tasks:
         - func: "bootstrap data lake"
         - func: "run tests"
           vars:
-            TEST_DATA_LAKE: "true"
+            TEST_NAME: "data_lake"
 
     - name: "test-aws-lambda-deployed"
       commands: