Skip to content

Commit 9003b25

Browse files
committed
fix enterprise auth
1 parent 73e7666 commit 9003b25

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

.evergreen/scripts/setup-tests.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -207,28 +207,28 @@ def handle_test_env() -> None:
207207
config = read_env(f"{ROOT}/secrets-export.sh")
208208
if os.name == "nt":
209209
LOGGER.info("Setting GSSAPI_PASS")
210-
write_env("GSSAPI_PASS", os.environ["SASL_PASS"])
210+
write_env("GSSAPI_PASS", config["SASL_PASS"])
211211
write_env("GSSAPI_CANONICALIZE", "true")
212212
else:
213213
# BUILD-3830
214214
krb_conf = ROOT / ".evergreen/krb5.conf.empty"
215215
krb_conf.touch()
216216
write_env("KRB5_CONFIG", krb_conf)
217217
LOGGER.info("Writing keytab")
218-
keytab = base64.b64decode(os.environ["KEYTAB_BASE64"])
218+
keytab = base64.b64decode(config["KEYTAB_BASE64"])
219219
keytab_file = ROOT / ".evergreen/drivers.keytab"
220220
with keytab_file.open("wb") as fid:
221221
fid.write(keytab)
222-
principal = os.environ["PRINCIPAL"]
222+
principal = config["PRINCIPAL"]
223223
LOGGER.info("Running kinit")
224224
os.environ["KRB5_CONFIG"] = str(krb_conf)
225225
cmd = f"kinit -k -t {keytab_file} -p {principal}"
226226
run_command(cmd)
227227

228228
LOGGER.info("Setting GSSAPI variables")
229-
write_env("GSSAPI_HOST", os.environ["SASL_HOST"])
230-
write_env("GSSAPI_PORT", os.environ["SASL_PORT"])
231-
write_env("GSSAPI_PRINCIPAL", os.environ["PRINCIPAL"])
229+
write_env("GSSAPI_HOST", config["SASL_HOST"])
230+
write_env("GSSAPI_PORT", config["SASL_PORT"])
231+
write_env("GSSAPI_PRINCIPAL", config["PRINCIPAL"])
232232

233233
if test_name == "load_balancer":
234234
SINGLE_MONGOS_LB_URI = os.environ.get(

0 commit comments

Comments
 (0)