Merge pull request #170 from LambdaTest/MLE-10664

support for setup of sys env on machine

Author: asad9711

commands/utils/batch/batch_runner.js

@@ -182,7 +182,7 @@ async function run(lt_config, batches, env, i = 0) {
           });
       })
       .catch(function (err) {
-        console.log("No able to archive the project");
+        console.log("Unable to archive the project");
         console.log(err);
         reject(err);
       });

commands/utils/constants.js

@@ -9,6 +9,7 @@ module.exports = {
   DEFAULT_TEST_PATH: ".",
   LAMBDA_CONFIG: "./lambdatest-config.json",
   SUPPORTED_CYPRESS_VERSIONS: ["5", "6"],
+  WHITELISTED_ENV_VARS: ["CI_BUILD_ID"],
   BUILD_END_STATES:
     "&status=running,queued,created,initiated,pqueued,error,lambda error,failed",
   BUILD_ERROR_STATES: "&status=error,lambda error,failed",

commands/utils/set_args.js

@@ -2,6 +2,7 @@ const constants = require("./constants.js");
 const fs = require("fs");
 const path = require("path");
 const process = require("process");
+const { type } = require("os");
 
 function write_file(file_path, content) {
   fs.writeFileSync(file_path, content, function (err) {
@@ -322,6 +323,33 @@ function sync_args_from_cmd(args) {
     } else {
       lt_config["run_settings"]["reject_unauthorized"] = false;
     }
+
+    //Set the env variables
+    let sys_env_vars = undefined;
+    let envs = {};
+    if ("sys-envs" in args) {
+      sys_env_vars = args["sys-envs"];
+    } else if (lt_config["run_settings"]["sys_envs"]) {
+      sys_env_vars = lt_config["run_settings"]["sys_envs"];
+    }
+
+    if (sys_env_vars){
+      sys_env_vars = sys_env_vars.trim();
+      sys_env_vars = sys_env_vars.split(";");
+      
+      for (index in sys_env_vars) {
+        envItem = sys_env_vars[index];
+        if (envItem){
+          envKeyValue = envItem.split("=");
+          envKey = envKeyValue[0];
+          envValue = envKeyValue[1];
+          envs[envKey] = envValue;
+        }
+      }
+    }
+    lt_config["run_settings"]["sys_envs"] = envs;
+    
+
     //get specs from current directory if specs are not passed in config or cli
     if (
       (lt_config["run_settings"]["specs"] == undefined ||

commands/utils/validate.js

@@ -1,6 +1,7 @@
 const fs = require("fs");
 const constants = require("./constants.js");
 module.exports = validate_config = function (lt_config, validation_configs) {
+  console.log("validating config");
   return new Promise(function (resolve, reject) {
     //validate auth keys are present
     if (
@@ -319,6 +320,22 @@ module.exports = validate_config = function (lt_config, validation_configs) {
         reject("Error!! boolean value is expected in reject_unauthorized key");
       }
     }
+
+    // validate system env variables to be set up
+    if ("sys_envs" in lt_config["run_settings"]) {
+      let sys_envs = lt_config["run_settings"]["sys_envs"];
+      let envValue;
+      Object.keys(sys_envs).forEach(function(envKey) {
+        envValue = sys_envs[envKey];
+        if (envKey && ! constants.WHITELISTED_ENV_VARS.includes(envKey)){
+          reject(`Usage of unwanted environment variable detected. Allowed variables are - ${constants.WHITELISTED_ENV_VARS}`);
+        }
+        if (envValue == undefined || envValue === ""){
+          reject("Value of environment variable cannot be left blank");
+        }
+      })
+    
+    }
     resolve("Validated the Config");
   });
 };

index.js

@@ -127,6 +127,11 @@ const argv = require("yargs")
           alias: "build-tags",
           describe: "build tags",
           type: "string",
+        })
+        .option("sys-envs", {
+          alias: "sys-env-variables",
+          describe: "system environment variables",
+          type: "string",
         });
     },
     function (argv) {