ensure FORCE_COLOR maxes out at a value of three

chalk · Oct 14, 2018 · e6b4bbc · e6b4bbc
1 parent bdf2cb2
commit e6b4bbc
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/index.js b/index.js
@@ -23,7 +23,7 @@ if ('FORCE_COLOR' in env) {
 	} else {
 		forceColor = env.FORCE_COLOR.length === 0 ?
 			1 :
-      parseInt(env.FORCE_COLOR, 10);
+      Math.min(parseInt(env.FORCE_COLOR, 10), 3);
 	}
 }
 

diff --git a/test.js b/test.js
@@ -329,6 +329,17 @@ test('supports setting a color value using FORCE_COLOR', t => {
 	result = importFresh('.');
 	t.truthy(result.stdout);
 	t.is(result.stdout.level, 3);
+
+	process.env.FORCE_COLOR = '0';
+	result = importFresh('.');
+	t.false(result.stdout);
+});
+
+test('FORCE_COLOR maxes out at a value of 3', t => {
+	process.env.FORCE_COLOR = '4';
+	const result = importFresh('.');
+	t.truthy(result.stdout);
+	t.is(result.stdout.level, 3);
 });
 
 test('FORCE_COLOR works when set via command line (all values are strings)', t => {