Added support for --no-color flag in verbose and mini reporters

Author: Thomas

lib/cli.js

@@ -38,6 +38,7 @@ exports.run = () => {
 		  --verbose, -v           Enable verbose output
 		  --no-cache              Disable the transpiler cache
 		  --no-power-assert       Disable Power Assert
+		  --no-color              Disable color output
 		  --match, -m             Only run tests with matching title (Can be repeated)
 		  --watch, -w             Re-run tests when tests and source files change
 		  --source, -S            Pattern to match source files so tests can be re-run (Can be repeated)
@@ -124,9 +125,14 @@ exports.run = () => {
 	if (cli.flags.tap && !cli.flags.watch) {
 		reporter = tapReporter();
 	} else if (cli.flags.verbose || isCi) {
-		reporter = verboseReporter();
+		reporter = verboseReporter({
+			disableColors: cli.flags.color !== undefined
+		});
 	} else {
-		reporter = miniReporter({watching: cli.flags.watch});
+		reporter = miniReporter({
+			watching: cli.flags.watch,
+			disableColors: cli.flags.color !== undefined
+		});
 	}
 
 	reporter.api = api;

lib/reporters/mini.js

@@ -28,6 +28,13 @@ function MiniReporter(options) {
 
 	this.options = Object.assign({}, options);
 
+	if (this.options.disableColors) {
+		chalk.enabled = false;
+		Object.keys(colors).forEach(function (key) {
+			colors[key].enabled = false;
+		});
+	}
+
 	this.reset();
 	this.stream = process.stderr;
 	this.stringDecoder = new StringDecoder();

lib/reporters/verbose.js

@@ -10,9 +10,18 @@ Object.keys(colors).forEach(function (key) {
 	colors[key].enabled = true;
 });
 
-function VerboseReporter() {
+function VerboseReporter(options) {
 	if (!(this instanceof VerboseReporter)) {
-		return new VerboseReporter();
+		return new VerboseReporter(options);
+	}
+
+	this.options = Object.assign({}, options);
+
+	if (this.options.disableColors) {
+		chalk.enabled = false;
+		Object.keys(colors).forEach(function (key) {
+			colors[key].enabled = false;
+		});
 	}
 }
 

test/reporters/mini.js

@@ -643,3 +643,26 @@ test('results when hasExclusive is enabled, but there are multiple remaining tes
 	t.end();
 });
 
+test('Result when no-color flag is set', function (t) {
+	var reporter = miniReporter({
+		disableColors: true
+	});
+
+	var runStatus = {
+		hasExclusive: true,
+		testCount: 3,
+		passCount: 1,
+		failCount: 0,
+		remainingCount: 2
+	};
+
+	var output = reporter.finish(runStatus);
+	var expectedOutput = [
+		'',
+		'',
+		'  The .only() modifier is used in some tests. 2 tests were not run',
+		'\n'
+	].join('\n');
+	t.is(output, expectedOutput);
+	t.end();
+});

test/reporters/verbose.js

@@ -530,6 +530,31 @@ test('results when hasExclusive is enabled, but there are multiple remaining tes
 	t.end();
 });
 
+test('Result when no-color flag is set', function (t) {
+	var reporter = verboseReporter({
+		disableColors: true
+	});
+	var runStatus = createRunStatus();
+	runStatus.hasExclusive = true;
+	runStatus.testCount = 3;
+	runStatus.passCount = 1;
+	runStatus.failCount = 0;
+	runStatus.remainingCount = 2;
+
+	var output = reporter.finish(runStatus);
+	var expectedOutput = [
+		'',
+		'  1 test passed [17:19:12]',
+		'',
+		'',
+		'  The .only() modifier is used in some tests. 2 tests were not run',
+		''
+	].join('\n');
+
+	t.is(output, expectedOutput);
+	t.end();
+});
+
 function fooFunc() {
 	barFunc();
 }