Updates test framework to check for config and give a better error message

dresende · dresende · commit b888d9e8b66e · 2013-07-30T11:56:32.000+01:00
Also, don't run a protocol if not configured (and specially don't
return error code for it)
diff --git a/test/common.js b/test/common.js
@@ -17,6 +17,18 @@ common.createConnection = function(cb) {
 	ORM.connect(this.getConnectionString(), cb);
 };
 
+common.hasConfig = function (proto) {
+	var config;
+
+	try {
+		config = require("./config");
+	} catch (ex) {
+		return 'not-found';
+	}
+
+	return (config.hasOwnProperty(proto) ? 'found' : 'not-defined');
+};
+
 common.getConfig = function () {
 	if (common.isTravis()) {
 		switch (this.protocol()) {
diff --git a/test/logging.js b/test/logging.js
@@ -0,0 +1,29 @@
+var util = require("util");
+
+exports.info = buildMethod(process.stdout, "[i]", 34);
+exports.error = buildMethod(process.stderr, "[!]", 31);
+
+function buildMethod(stream, prefix, color) {
+	return function () {
+		var params = Array.prototype.slice.apply(arguments);
+		var text   = params.shift();
+
+		return printTo(stream, prefix + " ", color, text, params);
+	};
+}
+
+function printTo(stream, prefix, color, text, params) {
+	params.unshift(text);
+	text = util.format.apply(util, params);
+
+	stream.write(printColor(color, true) + prefix + printColor(color) + text.replace(/\*\*(.+?)\*\*/, function (m, t) {
+		return printColor(color, true) + t + printColor(color);
+	}) + printColor(null) + "\n");
+}
+
+function printColor(color, bold) {
+	if (color === null) {
+		return "\033[0m";
+	}
+	return "\033[" + (bold ? "1" : "0") + ";" + color + "m";
+}
diff --git a/test/run.js b/test/run.js
@@ -1,11 +1,22 @@
-var Mocha    = require('mocha');
-var fs       = require('fs');
-var path     = require('path');
+var Mocha    = require("mocha");
+var fs       = require("fs");
+var path     = require("path");
+var common   = require("./common");
+var logging  = require("./logging");
 var location = path.normalize(path.join(__dirname, "integration"));
 var mocha    = new Mocha({
 	reporter: "progress"
 });
 
+switch (common.hasConfig(common.protocol())) {
+	case 'not-defined':
+		logging.error("There's no configuration for protocol **%s**", common.protocol());
+		process.exit(0);
+	case 'not-found':
+		logging.error("**test/config.js** missing. Take a look at **test/config.example.js**");
+		process.exit(0);
+}
+
 runTests();
 
 function runTests() {
@@ -19,8 +30,7 @@ function runTests() {
 		);
 	});
 
-	process.stdout.write("\033[1;34m[i] \033[0;34mTesting \033[1;34m" + process.env.ORM_PROTOCOL + "\033[0m\n");
-	process.stdout.write("\033[1;34m[i] \033[0;34mURI: \033[1;34m" + require('./common').getConnectionString() + "\033[0m\n");
+	logging.info("Testing **%s**", common.getConnectionString());
 
 	mocha.run(function (failures) {
 		process.exit(failures);
@@ -29,7 +39,7 @@ function runTests() {
 
 function shouldRunTest(file) {
 	var name  = file.substr(0, file.length - 3);
-	var proto = process.env.ORM_PROTOCOL;
+	var proto = common.protocol();
 
 	if (proto == "mongodb" && [ "model-aggregate",
 	                            "property-number-size", "smart-types" ].indexOf(name) >= 0) return false;
