Handle binary spawned by CLT

Author: Karan Nagpal

bin/commands/runs.js

@@ -204,7 +204,6 @@ module.exports = function run(args, rawArgs) {
                       }).finally(() =>{
                         let terminalWidth = (process.stdout.columns) * 0.9;
                         let lineSeparator = "\n" + "-".repeat(terminalWidth);
-                        console.log(lineSeparator)
                         logger.info(Constants.userMessages.BUILD_FAILED_ERROR)
                         process.exitCode = Constants.BUILD_FAILED_EXIT_CODE;
                       });

bin/helpers/capabilityHelper.js

@@ -71,21 +71,29 @@ const caps = (bsConfig, zip) => {
       obj.local = true;
     }
 
-    obj.localMode = null;
-    // Local Mode
-    if (obj.local === true && bsConfig.connection_settings.local_mode) {
-      obj.localMode = bsConfig.connection_settings.local_mode;
-      if (bsConfig.connection_settings.user_defined_local_mode_warning) {
-        logger.warn(Constants.userMessages.INVALID_LOCAL_MODE_WARNING);
+    // binary was spawned locally
+    if(obj.local === true) {
+      if (!Utils.isUndefined(process.env.BSTACK_CYPRESS_RUN_LOCAL_BINARY) && process.env.BSTACK_CYPRESS_RUN_LOCAL_BINARY == "true") {
+        obj.localMode = null;
+
+        // Local Mode
+        if (obj.local === true && bsConfig.connection_settings.local_mode) {
+          obj.localMode = bsConfig.connection_settings.local_mode;
+          if (bsConfig.connection_settings.user_defined_local_mode_warning) {
+            logger.warn(Constants.userMessages.INVALID_LOCAL_MODE_WARNING);
+          }
+          logger.info(`Local testing set up in ${obj.localMode} mode.`);
+        }
+
+        // Local Identifier
+        obj.localIdentifier = null;
+        if (obj.local === true && (bsConfig.connection_settings.localIdentifier || bsConfig.connection_settings.local_identifier)) {
+          obj.localIdentifier = bsConfig.connection_settings.localIdentifier || bsConfig.connection_settings.local_identifier;
+          logger.info(`Local testing identifier: ${obj.localIdentifier}`);
+        }
+      } else {
+        logger.info(Constants.userMessages.LOCAL_BINARY_ALREADY_RUNNING);
       }
-      logger.info(`Local testing set up in ${obj.localMode} mode.`);
-    }
-
-    // Local Identifier
-    obj.localIdentifier = null;
-    if (obj.local === true && (bsConfig.connection_settings.localIdentifier || bsConfig.connection_settings.local_identifier)) {
-      obj.localIdentifier = bsConfig.connection_settings.localIdentifier || bsConfig.connection_settings.local_identifier;
-      logger.info(`Local testing identifier: ${obj.localIdentifier}`);
     }
 
     logger.info(`Local is set to: ${obj.local} (${obj.local ? Constants.userMessages.LOCAL_TRUE : Constants.userMessages.LOCAL_FALSE})`);
@@ -164,8 +172,6 @@ const validate = (bsConfig, args) => {
     // validate local args i.e --local-mode and --local-identifier
 
     if( Utils.searchForOption('--local-identifier') && (Utils.isUndefined(args.localIdentifier) || (!Utils.isUndefined(args.localIdentifier) && !args.localIdentifier.trim()))) reject(Constants.validationMessages.INVALID_CLI_LOCAL_IDENTIFIER);
-    
-    if( Utils.getLocalFlag(bsConfig.connection_settings) && (Utils.isUndefined(bsConfig["connection_settings"]["local_identifier"]) || ( !Utils.isUndefined(bsConfig["connection_settings"]["local_identifier"]) && !bsConfig["connection_settings"]["local_identifier"].trim()))) reject(Constants.validationMessages.INVALID_LOCAL_IDENTIFIER);
 
     if( Utils.searchForOption('--local-mode') && ( Utils.isUndefined(args.localMode) || (!Utils.isUndefined(args.localMode) && !["always-on","on-demand"].includes(args.localMode)))) reject(Constants.validationMessages.INVALID_LOCAL_MODE);
 

bin/helpers/constants.js

@@ -56,6 +56,7 @@ const userMessages = {
   LOCAL_START_FAILED: "Local Testing setup failed.",
   LOCAL_STOP_FAILED: "Local Binary stop failed.",
   INVALID_LOCAL_MODE_WARNING: "Invalid value specified for local_mode. local_mode: (\"always-on\" | \"on-demand\"). For more info, check out https://www.browserstack.com/docs/automate/cypress/cli-reference",
+  LOCAL_BINARY_ALREADY_RUNNING: "We found an existing BrowserStack Local connection running from your account. Using the existing connection for this build. If you wish to use a new Local connection for your build, please specify a value for 'local_identifier' within 'connection_settings' in your browserstack.json config.",
   SPEC_LIMIT_WARNING: "You might not see all your results on the dashboard because of high spec count, please consider reducing the number of spec files in this folder.",
   DOWNLOAD_BUILD_ARTIFACTS_FAILED: "Downloading build artifacts for the build <build-id> failed for <machine-count> machines.",
   ASYNC_DOWNLOADS: "Test artifacts as specified under 'downloads' can be downloaded after the build has completed its run, using 'browserstack-cypress generate-downloads <build-id>'",

bin/helpers/utils.js

@@ -593,11 +593,6 @@ exports.setLocalIdentifier = (bsConfig, args) => {
       !this.isUndefined(bsConfig["connection_settings"]["local_identifier"])
     ){
     bsConfig['connection_settings']['local_mode'] = 'always-on';
-  } else if (
-      bsConfig['connection_settings']['local'] &&
-      this.isUndefined(bsConfig["connection_settings"]["local_identifier"])
-    ){
-    bsConfig["connection_settings"]["local_identifier"] = this.generateLocalIdentifier(bsConfig['connection_settings']['local_mode']);
   }
 };
 
@@ -637,9 +632,17 @@ exports.setLocalMode = (bsConfig, args) => {
 exports.setupLocalTesting = (bsConfig, args, rawArgs) => {
   return new Promise(async (resolve, reject) => {
     if( bsConfig['connection_settings'] && bsConfig['connection_settings']['local'] && String(bsConfig['connection_settings']['local']) === "true" ){
-      let localIdentifierRunning = await this.checkLocalIdentifierRunning(
-        bsConfig, bsConfig['connection_settings']['local_identifier']
-      );
+      let localBinaryRunning = await this.checklocalBinaryRunning(bsConfig, bsConfig['connection_settings']['local_identifier']);
+      let localIdentifierRunning;
+      if (localBinaryRunning['should_spawn_binary'] == true) {
+        localIdentifierRunning = false;
+        if(this.isUndefined(bsConfig["connection_settings"]["local_identifier"])) {
+          bsConfig["connection_settings"]["local_identifier"] = this.generateLocalIdentifier(bsConfig['connection_settings']['local_mode']);
+        }
+      } else {
+        localIdentifierRunning = true;
+        process.env.BSTACK_CYPRESS_RUN_LOCAL_BINARY = "true";
+      }
       if (!localIdentifierRunning){
         bsConfig.connection_settings.usedAutoLocal = true;
         var bs_local = this.getLocalBinary();
@@ -675,22 +678,6 @@ exports.setupLocalTesting = (bsConfig, args, rawArgs) => {
 
 exports.stopLocalBinary = (bsConfig, bs_local, args, rawArgs) => {
   return new Promise(async (resolve, reject) => {
-    if(bsConfig['connection_settings'] && bsConfig['connection_settings']['local']){
-      let localIdentifierRunning = await this.checkLocalIdentifierRunning(bsConfig,bsConfig["connection_settings"]["local_identifier"]);
-      if(!localIdentifierRunning){
-        let message = `Local Binary not running.`,
-          errorCode = 'local_identifier_error';
-        this.sendUsageReport(
-          bsConfig,
-          args,
-          message,
-          Constants.messageTypes.ERROR,
-          errorCode,
-          null,
-          rawArgs
-        );
-      }
-    }
     if (!this.isUndefined(bs_local) && bs_local.isRunning() && bsConfig['connection_settings'] && bsConfig['connection_settings']['local_mode'].toLowerCase() != "always-on") {
       let that = this;
       bs_local.stop(function (localStopError) {
@@ -744,7 +731,7 @@ exports.generateLocalIdentifier = (mode) => {
   return Buffer.from(local_identifier).toString("base64");
 };
 
-exports.checkLocalIdentifierRunning = (bsConfig, localIdentifier) => {
+exports.checklocalBinaryRunning = (bsConfig, localIdentifier) => {
   let options = {
     url: `${config.cypress_v1}/local_binary_running_check`,
     auth: {
@@ -762,13 +749,8 @@ exports.checkLocalIdentifierRunning = (bsConfig, localIdentifier) => {
         if(err){
           reject(err);
         }
-        console.log(body);
         let response = JSON.parse(body);
-        if(response['should_spawn_binary'] == true){
-          resolve(false);
-        } else {
-          resolve(true);
-        }
+        resolve(response);
     });
   });
 };
@@ -1043,7 +1025,6 @@ exports.stopBrowserStackBuild = async (bsConfig, args, buildId, rawArgs) => {
             logger.info(message);
           }
         } catch(err) {
-          console.log(err);
           message = Constants.userMessages.BUILD_STOP_FAILED;
           messageType = Constants.messageTypes.ERROR;
           errorCode = 'api_failed_build_stop';