Improve color support tests

Author: jhnns

test/cli.js

@@ -1,4 +1,8 @@
 'use strict';
+// tap runs tests in a non-TTY environment, that's why supports-color will return false.
+// Since we're going to check for colored output in our tests, we need to override that.
+process.env.FORCE_COLOR = true;
+
 const fs = require('fs');
 const path = require('path');
 const childProcess = require('child_process');
@@ -16,12 +20,6 @@ const colors = require('../lib/colors');
 
 const cliPath = path.join(__dirname, '../cli.js');
 
-// For some reason chalk is disabled by default
-chalk.enabled = true;
-for (const key of Object.keys(colors)) {
-	colors[key].enabled = true;
-}
-
 function execCli(args, opts, cb) {
 	let dirname;
 	let env;

test/fixture/chalk-disabled.js

@@ -0,0 +1,6 @@
+import chalk from 'chalk';
+import test from '../../';
+
+test('should not support colors', t => {
+	t.false(chalk.enabled);
+});

test/fixture/chalk-enabled.js

@@ -0,0 +1,6 @@
+import chalk from 'chalk';
+import test from '../../';
+
+test('should support colors', t => {
+	t.true(chalk.enabled);
+});

test/fork.js

@@ -7,15 +7,15 @@ const CachingPrecompiler = require('../lib/caching-precompiler');
 const cacheDir = path.join(__dirname, '../node_modules/.cache/ava');
 const precompiler = new CachingPrecompiler({path: cacheDir});
 
-function fork(testPath) {
+function fork(testPath, options) {
 	const hash = precompiler.precompileFile(testPath);
 	const precompiled = {};
 	precompiled[testPath] = hash;
 
-	return _fork(testPath, {
+	return _fork(testPath, Object.assign({
 		cacheDir,
 		precompiled
-	});
+	}, options));
 }
 
 function fixture(name) {
@@ -125,3 +125,20 @@ test('babelrc is ignored', t => {
 			t.end();
 		});
 });
+
+test('color support is initialized correctly', t => {
+	t.plan(1);
+
+	return Promise.all([
+		fork(fixture('chalk-enabled.js'), {color: true}).run({}),
+		fork(fixture('chalk-disabled.js'), {color: false}).run({}),
+		fork(fixture('chalk-disabled.js'), {}).run({})
+	]).then(info => {
+		info.forEach(info => {
+			if (info.stats.failCount > 0) {
+				throw new Error(`${info.file} failed`);
+			}
+		});
+		t.is(info.length, 3);
+	});
+});