picocolors/tests/environments.js at 571ede8d08aadb7065ce5bed5887567893ba7165 · alexeyraspopov/picocolors · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
let vm = require("vm")
let fs = require("fs")
let pc = require("../picocolors.js")
let assert = require("assert")
let source = fs.readFileSync(__dirname + "/../picocolors.js", "utf-8")

test("ci server", () => {
	let pc = initModuleEnv({ env: { TERM: "dumb", CI: "1" } })
	assert.equal(pc.isColorSupported, true)
	assert.equal(pc.red("text"), pc.createColors(true).red("text"))
})

test("arg --color", () => {
	let pc = initModuleEnv({ env: { TERM: "dumb" }, argv: ["--color"] })
	assert.equal(pc.isColorSupported, true)
	assert.equal(pc.red("text"), pc.createColors(true).red("text"))
})

test("env NO_COLOR", () => {
	let pc = initModuleEnv({ env: { FORCE_COLOR: "1", NO_COLOR: "1" } })
	assert.equal(pc.isColorSupported, false)
	assert.equal(pc.red("text"), pc.createColors(false).red("text"))
})

test("env FORCE_COLOR", () => {
	let pc = initModuleEnv({ env: { TERM: "dumb", FORCE_COLOR: "1" } })
	assert.equal(pc.isColorSupported, true)
	assert.equal(pc.red("text"), pc.createColors(true).red("text"))
})

test("arg --no-color", () => {
	let pc = initModuleEnv({ env: { FORCE_COLOR: "1" }, argv: ["--no-color"] })
	assert.equal(pc.isColorSupported, false)
	assert.equal(pc.red("text"), pc.createColors(false).red("text"))
})

test("no term", () => {
	let pc = initModuleEnv({ env: { TERM: "dumb" } })
	assert.equal(pc.isColorSupported, false)
	assert.equal(pc.red("text"), pc.createColors(false).red("text"))
})

test("windows", () => {
	let pc = initModuleEnv({ env: { TERM: "dumb" }, platform: "win32" })
	assert.equal(pc.isColorSupported, true)
	assert.equal(pc.red("text"), pc.createColors(true).red("text"))
})

test("edge runtime", () => {
	let pc = initModuleEnv({ env: { FORCE_COLOR: "1" }, argv: undefined, require: undefined })
	assert.equal(pc.isColorSupported, true)
	assert.equal(pc.red("text"), pc.createColors(true).red("text"))
})

function test(name, fn) {
	try {
		fn()
		console.log(pc.green("✓ " + name))
	} catch (error) {
		console.log(pc.red("✗ " + name))
		throw error
	}
}

function initModuleEnv({ env, argv = [], platform = "darwin", require = global.require }) {
	let process = { env, argv, platform }
	let context = vm.createContext({ require, process, module: { exports: {} } })
	let script = new vm.Script(source)
	script.runInContext(context)
	return context.module.exports
}