add files

Author: blink1073

.evergreen/scripts/kms_tester.py

@@ -5,38 +5,43 @@
 from utils import DRIVERS_TOOLS, LOGGER, ROOT, read_env, run_command, write_env
 
 TMP_DRIVER_FILE = "/tmp/mongo-python-driver.tgz"  # noqa: S108
-CSFLE_DIR = f"{DRIVERS_TOOLS}/.evergreen/csfle"
+DIRS = dict(
+    gcp=f"{DRIVERS_TOOLS}/.evergreen/csfle/gcpkms",
+    azure=f"{DRIVERS_TOOLS}/.evergreen/csfle/azurekms",
+)
 
 
 def _setup_azure_vm(base_env: dict[str, str]) -> None:
     LOGGER.info("Setting up Azure VM...")
+    azure_dir = DIRS["azure"]
     env = base_env.copy()
     env["AZUREKMS_SRC"] = TMP_DRIVER_FILE
     env["AZUREKMS_DST"] = "~/"
-    run_command(f"{CSFLE_DIR}/azurekms/copy-file.sh", env=env)
+    run_command(f"{azure_dir}/copy-file.sh", env=env)
 
     env = base_env.copy()
     env["AZUREKMS_CMD"] = "tar xf mongo-python-driver.tgz"
-    run_command(f"{CSFLE_DIR}/azurekms/run-command.sh", env=env)
+    run_command(f"{azure_dir}/run-command.sh", env=env)
 
     env["AZUREKMS_CMD"] = "bash .evergreen/just.sh setup-tests kms azure-remote"
-    run_command(f"{CSFLE_DIR}/azurekms/run-command.sh", env=env)
+    run_command(f"{azure_dir}/run-command.sh", env=env)
     LOGGER.info("Setting up Azure VM... done.")
 
 
 def _setup_gcp_vm(base_env: dict[str, str]) -> None:
     LOGGER.info("Setting up GCP VM...")
+    gcp_dir = DIRS["gcp"]
     env = base_env.copy()
     env["GCPKMS_SRC"] = TMP_DRIVER_FILE
     env["GCPKMS_DST"] = f"{env['GCPKMS_INSTANCENAME']}:"
-    run_command(f"{CSFLE_DIR}/gcpkms/copy-file.sh", env=env)
+    run_command(f"{gcp_dir}/copy-file.sh", env=env)
 
     env = base_env.copy()
     env["GCPKMS_CMD"] = "tar xf mongo-python-driver.tgz"
-    run_command(f"{CSFLE_DIR}/gcpkms/run-command.sh", env=env)
+    run_command(f"{gcp_dir}/run-command.sh", env=env)
 
     env["GCPKMS_CMD"] = "bash ./.evergreen/just.sh setup-tests kms gcp-remote"
-    run_command(f"{CSFLE_DIR}/gcpkms/run-command.sh", env=env)
+    run_command(f"{gcp_dir}/run-command.sh", env=env)
     LOGGER.info("Setting up GCP VM...")
 
 
@@ -47,7 +52,8 @@ def _create_archive() -> None:
 
 
 def _load_kms_config(sub_test_target: str) -> dict[str, str]:
-    config = read_env(f"{CSFLE_DIR}/{sub_test_target}kms/secrets-export.sh")
+    target_dir = DIRS[sub_test_target]
+    config = read_env(f"{target_dir}/secrets-export.sh")
     base_env = os.environ.copy()
     for key, value in config.items():
         base_env[key] = str(value)
@@ -64,7 +70,7 @@ def setup_kms(sub_test_name: str) -> None:
     assert sub_test_target in ["azure", "gcp"], sub_test_target
     assert sub_test_type in ["", "remote", "fail"], sub_test_type
     success = sub_test_type != "fail"
-    kms_dir = f"{CSFLE_DIR}/{sub_test_target}kms"
+    kms_dir = DIRS[sub_test_target]
 
     if sub_test_target == "azure":
         write_env("TEST_FLE_AZURE_AUTO")
@@ -108,12 +114,12 @@ def test_kms_vm(sub_test_name: str) -> None:
         ] = f'KEY_NAME="{key_name}" KEY_VAULT_ENDPOINT="{key_vault_endpoint}" bash ./.evergreen/just.sh run-tests'
     else:
         env["GCPKMS_CMD"] = "./.evergreen/just.sh run-tests"
-    cmd = f"{CSFLE_DIR}/{sub_test_name}kms/run-command.sh"
+    cmd = f"{DIRS[sub_test_name]}/run-command.sh"
     run_command(cmd, env=env)
 
 
 def teardown_kms(sub_test_name: str) -> None:
-    run_command(f"{CSFLE_DIR}/{sub_test_name}kms/teardown.sh")
+    run_command(f"{DIRS[sub_test_name]}/teardown.sh")
 
 
 if __name__ == "__main__":