Reuse worker initial commit

Author: dhimil

bin/cli.js

@@ -32,6 +32,7 @@ var Log = require('../lib/logger'),
     timeout,
     activityTimeout,
     workers = {},
+    browserToWorker = {},
     workerKeys = {},
     logLevel,
     tunnelingAgent,
@@ -176,9 +177,13 @@ function launchBrowsers(config, browser) {
   setTimeout(function () {
     if(Object.prototype.toString.call(config.test_path) === '[object Array]'){
       config.multipleTest = config.test_path.length > 1? true : false;
-      config.test_path.forEach(function(path){
-        launchBrowser(browser, path);
-      });
+      if(config.reuseWorker) {
+        launchBrowser(browser, config.test_path[0]);
+      } else {
+        config.test_path.forEach(function(path){
+          launchBrowser(browser, path);
+        });
+      }
     } else {
       config.multipleTest = false;
       launchBrowser(browser, config.test_path);

lib/server.js

@@ -157,6 +157,29 @@ exports.Server = function Server(bsClient, workers) {
     return details;
   }
 
+  function checkAndTerminateWorker(worker, callback) {
+    if(config.reuseWorker && config.multipleTest) {
+      var next_path = getNextTestPath(worker.test_path);
+      var new_url = 'http://localhost:' + 8888 + '/' + next_path
+        + "_worker_key=" + worker._worker_key + "&_browser_string=" + getTestBrowserInfo(worker) ;
+      bsClient.postNewUrl(worker.id, {url: new_url}, function() {
+        callback(true);
+      });
+    } else {
+      bsClient.terminateWorker(worker.id, callback);
+    }
+  };
+
+  function getNextTestPath(test_path) {
+    // Someday I'll die for codes like these
+    for(var i = 0; i < config.test_path.length; i++) {
+      if(config.test_path[i] == test_path) {
+        return config.test_path[ i + 1 ];
+      }
+    }
+
+  }
+
   handlers = {
     "_progress": function progressHandler(uri, body, request, response) {
       var uuid = request.headers['x-worker-uuid'];
@@ -184,6 +207,7 @@ exports.Server = function Server(bsClient, workers) {
       } catch (e) {}
       var uuid = request.headers['x-worker-uuid'];
       var worker = workers[uuid] || {};
+      worker._worker_key = uuid;
 
       if (query === null) {
         logger.info("[%s] Null response from remote Browser", request.headers['x-browser-string']);
@@ -205,11 +229,16 @@ exports.Server = function Server(bsClient, workers) {
             logger.info('[%s] ' + chalk['yellow']('Screenshot') + ': %s', getTestBrowserInfo(worker), screenshot.url);
           }
 
-          bsClient.terminateWorker(worker.id, function() {
+          checkAndTerminateWorker(worker, function(reusedWorker) {
             if (!workers[uuid]) {
               return;
             }
 
+            if(reusedWorker) {
+              logger.debug('[%s] Reused', getTestBrowserInfo(worker));
+              return;
+            }
+
             logger.debug('[%s] Terminated', getTestBrowserInfo(worker));
 
             clearTimeout(workers[uuid].activityTimeout);