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)
@@ -70,9 +71,10 @@ exports.run = () => {
 			'tap',
 			'verbose',
 			'watch',
-			'update-snapshots'
+			'update-snapshots',
+			'color'
 		],
-		default: conf,
+		default: Object.assign({color: true}, conf),
 		alias: {
 			t: 'tap',
 			v: 'verbose',
@@ -121,17 +123,23 @@ exports.run = () => {
 		pkgDir,
 		timeout: cli.flags.timeout,
 		concurrency: cli.flags.concurrency ? parseInt(cli.flags.concurrency, 10) : 0,
-		updateSnapshots: cli.flags.updateSnapshots
+		updateSnapshots: cli.flags.updateSnapshots,
+		color: cli.flags.color
 	});
 
 	let reporter;
 
 	if (cli.flags.tap && !cli.flags.watch) {
 		reporter = tapReporter();
 	} else if (cli.flags.verbose || isCi) {
-		reporter = verboseReporter();
+		reporter = verboseReporter({
+			color: cli.flags.color
+		});
 	} else {
-		reporter = miniReporter({watching: cli.flags.watch});
+		reporter = miniReporter({
+			watching: cli.flags.watch,
+			color: cli.flags.color
+		});
 	}
 
 	reporter.api = api;

lib/fork.js

@@ -36,7 +36,9 @@ module.exports = (file, opts, execArgv) => {
 		} : false
 	}, opts);
 
-	const ps = childProcess.fork(path.join(__dirname, 'test-worker.js'), [JSON.stringify(opts)], {
+	var args = [JSON.stringify(opts), opts.color ? '--color' : '--no-color'];
+
+	const ps = childProcess.fork(path.join(__dirname, 'test-worker.js'), args, {
 		cwd: opts.pkgDir,
 		silent: true,
 		env,

lib/reporters/mini.js

@@ -11,9 +11,9 @@ var repeating = require('repeating');
 var colors = require('../colors');
 
 chalk.enabled = true;
-Object.keys(colors).forEach(function (key) {
+for (let key of Object.keys(colors)) {
 	colors[key].enabled = true;
-});
+}
 
 function MiniReporter(options) {
 	if (!(this instanceof MiniReporter)) {
@@ -28,6 +28,13 @@ function MiniReporter(options) {
 
 	this.options = Object.assign({}, options);
 
+	if (this.options.color === false) {
+		chalk.enabled = false;
+		for (let key of Object.keys(colors)) {
+			colors[key].enabled = false;
+		}
+	}
+
 	this.reset();
 	this.stream = process.stderr;
 	this.stringDecoder = new StringDecoder();

lib/reporters/verbose.js

@@ -6,13 +6,23 @@ var plur = require('plur');
 var repeating = require('repeating');
 var colors = require('../colors');
 
-Object.keys(colors).forEach(function (key) {
+chalk.enabled = true;
+for (let key of Object.keys(colors)) {
 	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.color === false) {
+		chalk.enabled = false;
+		for (let key of Object.keys(colors)) {
+			colors[key].enabled = false;
+		}
 	}
 }
 

readme.md

@@ -159,6 +159,7 @@ $ ava --help
     --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)

test/reporters/mini.js

@@ -671,3 +671,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({
+		color: false
+	});
+
+	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

@@ -571,6 +571,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({
+		color: false
+	});
+	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();
 }