Skip to content

Commit

Permalink
[#898] Added in injection script
Browse files Browse the repository at this point in the history
  • Loading branch information
dseif committed Jan 24, 2012
1 parent 2aaf4b1 commit 587878f
Showing 1 changed file with 324 additions and 0 deletions.
324 changes: 324 additions & 0 deletions test/inject.js
@@ -0,0 +1,324 @@
(function(){

var DEBUG = false;

var doPost = false;

try {
doPost = !!window.top.postMessage;
} catch(e){}

var search = window.location.search,
url, index;
if( ( index = search.indexOf( "swarmURL=" ) ) != -1 )
url = decodeURIComponent( search.slice( index + 9 ) );

if ( !DEBUG && (!url || url.indexOf("http") !== 0) ) {
return;
}

var submitTimeout = 5;

var curHeartbeat;
var beatRate = 20;

// Expose the TestSwarm API
window.TestSwarm = {
submit: submit,
heartbeat: function(){
if ( curHeartbeat ) {
clearTimeout( curHeartbeat );
}

curHeartbeat = setTimeout(function(){
submit({ fail: -1, total: -1 });
}, beatRate * 1000);
},
serialize: function(){
return trimSerialize();
}
};

// Prevent careless things from executing
window.print = window.confirm = window.alert = window.open = function(){};

window.onerror = function(e){
document.body.appendChild( document.createTextNode( "ERROR: " + e ));
submit({ fail: 0, error: 1, total: 1 });
return false;
};

// QUnit (jQuery)
// http://docs.jquery.com/QUnit
if ( typeof QUnit !== "undefined" ) {
QUnit.done = function(failed, passed){
submit({
fail: failed,
error: 0,
total: passed + failed
});
};

QUnit.log = window.TestSwarm.heartbeat;
window.TestSwarm.heartbeat();

window.TestSwarm.serialize = function(){
// Clean up the HTML (remove any un-needed test markup)
remove("nothiddendiv");
remove("loadediframe");
remove("dl");
remove("main");

// Show any collapsed results
var ol = document.getElementsByTagName("ol");
for ( var i = 0; i < ol.length; i++ ) {
ol[i].style.display = "block";
}

return trimSerialize();
};

// UnitTestJS (Prototype, Scriptaculous)
// http://github.com/tobie/unittest_js/tree/master
} else if ( typeof Test !== "undefined" && Test && Test.Unit && Test.Unit.runners ) {
var total_runners = Test.Unit.runners.length, cur_runners = 0;
var total = 0, fail = 0, error = 0;

for (var i = 0; i < Test.Unit.runners.length; i++) (function(i){
var finish = Test.Unit.runners[i].finish;
Test.Unit.runners[i].finish = function(){
finish.call( this );

var results = this.getResult();
total += results.assertions;
fail += results.failures;
error += results.errors;

if ( ++cur_runners === total_runners ) {
submit({
fail: fail,
error: error,
total: total
});
}
};
})(i);

// JSSpec (MooTools)
// http://jania.pe.kr/aw/moin.cgi/JSSpec
} else if ( typeof JSSpec !== "undefined" && JSSpec && JSSpec.Logger ) {
var onRunnerEnd = JSSpec.Logger.prototype.onRunnerEnd;
JSSpec.Logger.prototype.onRunnerEnd = function(){
onRunnerEnd.call(this);

// Show any collapsed results
var ul = document.getElementsByTagName("ul");
for ( var i = 0; i < ul.length; i++ ) {
ul[i].style.display = "block";
}

submit({
fail: JSSpec.runner.getTotalFailures(),
error: JSSpec.runner.getTotalErrors(),
total: JSSpec.runner.totalExamples
});
};

window.TestSwarm.serialize = function(){
// Show any collapsed results
var ul = document.getElementsByTagName("ul");
for ( var i = 0; i < ul.length; i++ ) {
ul[i].style.display = "block";
}

return trimSerialize();
};

// JSUnit
// http://www.jsunit.net/
// Note: Injection file must be included before the frames
// are document.write()d into the page.
} else if ( typeof JsUnitTestManager !== "undefined" ) {
var _done = JsUnitTestManager.prototype._done;
JsUnitTestManager.prototype._done = function(){
_done.call(this);

submit({
fail: this.failureCount,
error: this.errorCount,
total: this.totalCount
});
};

window.TestSwarm.serialize = function(){
return "<pre>" + this.log.join("\n") + "</pre>";
};

// Selenium Core
// http://seleniumhq.org/projects/core/
} else if ( typeof SeleniumTestResult !== "undefined" && typeof LOG !== "undefined" ) {
// Completely overwrite the postback
SeleniumTestResult.prototype.post = function(){
submit({
fail: this.metrics.numCommandFailures,
error: this.metrics.numCommandErrors,
total: this.metrics.numCommandPasses + this.metrics.numCommandFailures + this.metrics.numCommandErrors
});
};

window.TestSwarm.serialize = function(){
var results = [];
while ( LOG.pendingMessages.length ) {
var msg = LOG.pendingMessages.shift();
results.push( msg.type + ": " + msg.msg );
}

return "<pre>" + results.join("\n") + "</pre>";
};

// Dojo Objective Harness
// http://docs.dojocampus.org/quickstart/doh
} else if ( typeof doh !== "undefined" && doh._report ) {
var _report = doh._report;
doh._report = function(){
_report.apply(this, arguments);

submit({
fail: doh._failureCount,
error: doh._errorCount,
total: doh._testCount
});
};

window.TestSwarm.serialize = function(){
return "<pre>" + document.getElementById("logBody").innerHTML + "</pre>";
};
// Screw.Unit
// git://github.com/nathansobo/screw-unit.git
} else if ( typeof Screw !== "undefined" && typeof jQuery !== "undefined" && Screw && Screw.Unit ) {
$(Screw).bind("after", function() {
var passed = $(".passed").length;
var failed = $(".failed").length;
submit({
fail: failed,
error: 0,
total: failed + passed
});
});

$(Screw).bind("loaded", function() {
$(".it")
.bind("passed", window.TestSwarm.heartbeat)
.bind("failed", window.TestSwarm.heartbeat);
window.TestSwarm.heartbeat();
});

window.TestSwarm.serialize = function(){
return trimSerialize();
};
}

function trimSerialize(doc) {
doc = doc || document;

var scripts = doc.getElementsByTagName("script");
while ( scripts.length ) {
remove( scripts[0] );
}

var root = window.location.href.replace(/(https?:\/\/.*?)\/.*/, "$1");
var cur = window.location.href.replace(/[^\/]*$/, "");

var links = doc.getElementsByTagName("link");
for ( var i = 0; i < links.length; i++ ) {
var href = links[i].href;
if ( href.indexOf("/") === 0 ) {
href = root + href;
} else if ( !/^https?:\/\//.test( href ) ) {
href = cur + href;
}
links[i].href = href;
}

return ("<html>" + doc.documentElement.innerHTML + "</html>")
.replace(/\s+/g, " ");
}

function remove(elem){
if ( typeof elem === "string" ) {
elem = document.getElementById( elem );
}

if ( elem ) {
elem.parentNode.removeChild( elem );
}
}

function submit(params){
if ( curHeartbeat ) {
clearTimeout( curHeartbeat );
}

var paramItems = (url.split("?")[1] || "").split("&");

for ( var i = 0; i < paramItems.length; i++ ) {
if ( paramItems[i] ) {
var parts = paramItems[i].split("=");
if ( !params[ parts[0] ] ) {
params[ parts[0] ] = parts[1];
}
}
}

if ( !params.state ) {
params.state = "saverun";
}

if ( !params.results ) {
params.results = window.TestSwarm.serialize();
}

if ( doPost ) {
// Build Query String
var query = "";

for ( var i in params ) {
query += ( query ? "&" : "" ) + i + "=" +
encodeURIComponent(params[i]);
}

if ( DEBUG ) {
alert( query );
} else {
window.top.postMessage( query, "*" );
}

} else {
var form = document.createElement("form");
form.action = url;
form.method = "POST";

for ( var i in params ) {
var input = document.createElement("input");
input.type = "hidden";
input.name = i;
input.value = params[i];
form.appendChild( input );
}

if ( DEBUG ) {
alert( form.innerHTML );
} else {

// Watch for the result submission timing out
setTimeout(function(){
submit( params );
}, submitTimeout * 1000);

document.body.appendChild( form );
form.submit();
}
}
}

})();

0 comments on commit 587878f

Please sign in to comment.