add support for non-string input

Closes #14 Closes #24
alexeyraspopov · Oct 4, 2021 · 3276400 · 3276400
1 parent 2bbb880
commit 3276400
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/picocolors.js b/picocolors.js
@@ -9,7 +9,8 @@ let isColorSupported =
 		"CI" in process.env);
 
 function formatter(open, close, replace = open) {
-	return (string) => {
+	return (input) => {
+		let string = "" + input;
 		let index = string.indexOf(close, open.length);
 		return !~index
 			? open + string + close

diff --git a/tests/test.js b/tests/test.js
@@ -125,6 +125,17 @@ test("close sequence replacement", () => {
 	);
 });
 
+test("non-string input", () => {
+	assert.equal(pc.red(), FMT.red[0] + "undefined" + FMT.red[1]);
+	assert.equal(pc.red(undefined), FMT.red[0] + "undefined" + FMT.red[1]);
+	assert.equal(pc.red(0), FMT.red[0] + "0" + FMT.red[1]);
+	assert.equal(pc.red(NaN), FMT.red[0] + "NaN" + FMT.red[1]);
+	assert.equal(pc.red(null), FMT.red[0] + "null" + FMT.red[1]);
+	assert.equal(pc.red(true), FMT.red[0] + "true" + FMT.red[1]);
+	assert.equal(pc.red(false), FMT.red[0] + "false" + FMT.red[1]);
+	assert.equal(pc.red(Infinity), FMT.red[0] + "Infinity" + FMT.red[1]);
+});
+
 function test(name, fn) {
 	try {
 		fn();