support multi page flow

lib/driver.js

@@ -14,7 +14,7 @@
  * Module dependencies.
  */
 
-//require('debug').enable('totoro*');
+require('debug').enable('totoro*');
 var debug = require('debug')('totoro-phantomjs-driver');
 var TotoroDriver = require('totoro-driver-base');
 var phantomjs = require('phantomjs');
@@ -55,8 +55,8 @@ proto.onAdd = function (data) {
     path.join(__dirname, 'phantom-openurl.js'),
     data.url,
     this.includeScripts,
-    this.script,
-    this.ignoreLog
+    this.ignoreLog,
+    JSON.stringify(this.subTasks)
   ];
   var child = childProcess.execFile(phantomjs.path, args);
   child.stdout.on('data', function (out) {

lib/phantom-openurl.js

@@ -21,59 +21,119 @@ var async = require('async');
 
 var url = system.args[1];
 var includeScripts = system.args[2];
-var script = system.args[3];
-var ignoreLog = system.args[4];
+var ignoreLog = system.args[3];
+var subTasks = system.args[4];
+
+if (subTasks) {
+  try {
+    subTasks = JSON.parse(subTasks);
+  } catch(e) {
+    subTasks = null;
+  }
+}
 
 if (includeScripts) {
   includeScripts = includeScripts.split(',');
 } else {
   includeScripts = [];
 }
 
-var isInserted = false;
 log('opening ' + url);
 page.open(url, function (status) {
   log('opened ' + url + ', status ' + status);
-  isInserted = false;
 });
 
+var script;
 // To avoid triggering this method repeatedly.
 page.onLoadFinished = function(status) {
   if (status !== 'success') {
     log('opened url error');
     exit(1);
     return;
   }
-
-  if (!isInserted) {
+  if (script) {
     insertScripts(page, includeScripts, function() {
       executeScript(script);
+      script = null;
     });
   }
+};
 
-  isInserted = true;
+page.onUrlChanged = function(a, b) {
+  script = findMappingScript(page.url);
 };
 
 //include scripts;
 function insertScripts(page, scripts, cb) {
   async.eachSeries(scripts, function(script, callback) {
-    page.includeJs(script, function() {
-      log('loaded ' + script);
+    if (script.indexOf('http') < 0) {
+      page.injectJs(script);
       callback();
-    });
+    } else {
+      page.includeJs(script, function() {
+        log('loaded ' + script);
+        callback();
+        // 有可能是缓存的原因, 第二次加载 js 的时候, 这个方法会触发两遍.
+        callback = function(){};
+      });
+    }
   }, function() {
     log('scripts loaded!');
     cb();
   });
 }
 
+page.onError = function(msg, trace) {
+  var msgStack = ['ERROR: ' + msg];
+
+  if (trace && trace.length) {
+    msgStack.push('TRACE:');
+    trace.forEach(function(t) {
+      msgStack.push(' -> ' + t.file + ': ' + t.line + (t.function ? ' (in function "' + t.function +'")' : ''));
+    });
+  }
+
+  console.error(msgStack.join('\n'));
+};
+
 function executeScript(script) {
   if (script) {
     page.evaluateJavaScript(eval(script));
   }
 }
 
+page.onConsoleMessage = function(msg, lineNum, sourceId) {
+  log('CONSOLE: ' + msg + ' (from line #' + lineNum + ' in "' + sourceId + '")');
+};
+
+function findMappingScript(url) {
+  // 如果支持多页面, 需要提前传入 runner 和 测试脚本的对应关系, 也就是 subTasks
+  var script;
+  if (subTasks) {
+    Object.keys(subTasks).some(function(subUrl) {
+      if (isMatch(url, subUrl)) {
+        script = subTasks[subUrl];
+        delete subTasks[subUrl];
+      }
+    });
+  }
+  return script;
+};
+
+// 检查 url 是否匹配
+function isMatch(url, subUrl) {
+  if (url.indexOf(subUrl) === 0) {
+    return true;
+  }
+
+  if (subUrl.indexOf('/') === 0) {
+    return url.indexOf(subUrl) > -1;
+  }
+  return false;
+}
+
 function log(str) {
+  return;
   if (ignoreLog) {
     return;
   }