jeresig / testswarm

(function(){
 
	var DEBUG = false;
 
	var doPost = false;
 
	try {
		doPost = !!window.top.postMessage;
	} catch(e){}
 
	var url = window.location.search;
	url = decodeURIComponent( url.slice( url.indexOf("swarmURL=") + 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(fail, total){
			submit({
				fail: fail,
				error: 0,
				total: total
			});
		};
 
		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>";
		};
	}
 
	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();
			}
		}
	}
 
})();