should-color 0.5.2

Determine whether output should use colors or not.
Documentation 
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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98

// ATTENTION: tests should be run with a single thread:
//
// ```bash
// cargo test -- --test-threads 1
// ```

use itertools::*;

use should_color::*;

#[allow(unused_variables)]
fn setup_env(no_color: Option<&str>, clicolor: Option<&str>, clicolor_force: Option<&str>) {
    #[cfg(feature = "no_color")]
    no_color.map_or_else(
        || std::env::remove_var(NO_COLOR),
        |s| std::env::set_var(NO_COLOR, s),
    );
    #[cfg(feature = "clicolor")]
    clicolor.map_or_else(
        || std::env::remove_var(CLICOLOR),
        |s| std::env::set_var(CLICOLOR, s),
    );
    #[cfg(feature = "clicolor_force")]
    clicolor_force.map_or_else(
        || std::env::remove_var(CLICOLOR_FORCE),
        |s| std::env::set_var(CLICOLOR_FORCE, s),
    );
}

#[test]
#[cfg(feature = "clicolor_force")]
fn test_clicolor_force() {
    let any_env = [
        None,
        Some(""),
        Some("0"),
        Some("1"),
        Some("false"),
        Some("true"),
    ];
    let any_cli = [
        None,
        Some(ColorChoice::Never),
        Some(ColorChoice::Auto),
        Some(ColorChoice::Always),
    ];
    let any_set_clicolor_force = [Some("1"), Some("="), Some("false"), Some("true")];

    for (no_color, clicolor, cli, clicolor_force) in
        iproduct!(any_env, any_env, any_cli, any_set_clicolor_force)
    {
        setup_env(no_color, clicolor, clicolor_force);
        assert_eq!(resolve(cli), Some(ColorChoice::Always));
    }
}

#[test]
#[cfg(feature = "clicolor_force")]
fn test_cli() {
    let any_env = [
        None,
        Some(""),
        Some("0"),
        Some("1"),
        Some("false"),
        Some("true"),
    ];
    let any_set_cli = [ColorChoice::Never, ColorChoice::Auto, ColorChoice::Always];
    let any_unset_clicolor_force = [None, Some(""), Some("0")];

    for (no_color, clicolor, cli, clicolor_force) in
        iproduct!(any_env, any_env, any_set_cli, any_unset_clicolor_force)
    {
        setup_env(no_color, clicolor, clicolor_force);
        assert_eq!(resolve(Some(cli)), Some(cli));
    }
}

#[test]
#[cfg(not(feature = "clicolor_force"))]
fn test_cli() {
    let any_env = [
        None,
        Some(""),
        Some("0"),
        Some("1"),
        Some("false"),
        Some("true"),
    ];
    let any_set_cli = [ColorChoice::Never, ColorChoice::Auto, ColorChoice::Always];

    for (no_color, clicolor, cli, clicolor_force) in
        iproduct!(any_env, any_env, any_set_cli, any_env)
    {
        setup_env(no_color, clicolor, clicolor_force);
        assert_eq!(resolve(Some(cli)), Some(cli));
    }
}