SERVER-47355 Propagate hygienic installation path into shell tests

Author: spencerjackson

buildscripts/resmokelib/core/programs.py

@@ -70,6 +70,7 @@ def make_process(*args, **kwargs):
     # If installDir is provided, add it early to the path
     if config.INSTALL_DIR is not None:
         path.append(config.INSTALL_DIR)
+        env_vars["INSTALL_DIR"] = config.INSTALL_DIR
 
     path.append(env_vars.get("PATH", os.environ.get("PATH", "")))
 

src/mongo/shell/shell_utils_extended.cpp

@@ -435,6 +435,28 @@ BSONObj readDumpFile(const BSONObj& a, void*) {
     return builder.obj();
 }
 
+BSONObj shellGetEnv(const BSONObj& a, void*) {
+    uassert(46712006,
+            "_getEnv() takes one argument: the name of the environment variable",
+            a.nFields() == 1 && a.firstElementType() == String);
+    const auto envName = a.firstElement().String();
+    std::string result{};
+#ifndef _WIN32
+    auto envPtr = getenv(envName.c_str());
+    if (envPtr) {
+        result = std::string(envPtr);
+    }
+#else
+    auto envPtr = _wgetenv(toNativeString(envName.c_str()).c_str());
+    if (envPtr) {
+        result = toUtf8String(envPtr);
+    }
+#endif
+
+    return BSON("" << result.c_str());
+}
+
+
 }  // namespace
 
 void installShellUtilsExtended(Scope& scope) {
@@ -454,6 +476,7 @@ void installShellUtilsExtended(Scope& scope) {
     scope.injectNative("umask", changeUmask);
     scope.injectNative("getFileMode", getFileMode);
     scope.injectNative("_readDumpFile", readDumpFile);
+    scope.injectNative("_getEnv", shellGetEnv);
 }
 
 }  // namespace shell_utils