145 lines
3.2 KiB
Go
145 lines
3.2 KiB
Go
package iostreams
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
func TestEnvColorDisabled(t *testing.T) {
|
|
orig_NO_COLOR := os.Getenv("NO_COLOR")
|
|
orig_CLICOLOR := os.Getenv("CLICOLOR")
|
|
orig_CLICOLOR_FORCE := os.Getenv("CLICOLOR_FORCE")
|
|
t.Cleanup(func() {
|
|
os.Setenv("NO_COLOR", orig_NO_COLOR)
|
|
os.Setenv("CLICOLOR", orig_CLICOLOR)
|
|
os.Setenv("CLICOLOR_FORCE", orig_CLICOLOR_FORCE)
|
|
})
|
|
|
|
tests := []struct {
|
|
name string
|
|
NO_COLOR string
|
|
CLICOLOR string
|
|
CLICOLOR_FORCE string
|
|
want bool
|
|
}{
|
|
{
|
|
name: "pristine env",
|
|
NO_COLOR: "",
|
|
CLICOLOR: "",
|
|
CLICOLOR_FORCE: "",
|
|
want: false,
|
|
},
|
|
{
|
|
name: "NO_COLOR enabled",
|
|
NO_COLOR: "1",
|
|
CLICOLOR: "",
|
|
CLICOLOR_FORCE: "",
|
|
want: true,
|
|
},
|
|
{
|
|
name: "CLICOLOR disabled",
|
|
NO_COLOR: "",
|
|
CLICOLOR: "0",
|
|
CLICOLOR_FORCE: "",
|
|
want: true,
|
|
},
|
|
{
|
|
name: "CLICOLOR enabled",
|
|
NO_COLOR: "",
|
|
CLICOLOR: "1",
|
|
CLICOLOR_FORCE: "",
|
|
want: false,
|
|
},
|
|
{
|
|
name: "CLICOLOR_FORCE has no effect",
|
|
NO_COLOR: "",
|
|
CLICOLOR: "",
|
|
CLICOLOR_FORCE: "1",
|
|
want: false,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
os.Setenv("NO_COLOR", tt.NO_COLOR)
|
|
os.Setenv("CLICOLOR", tt.CLICOLOR)
|
|
os.Setenv("CLICOLOR_FORCE", tt.CLICOLOR_FORCE)
|
|
|
|
if got := EnvColorDisabled(); got != tt.want {
|
|
t.Errorf("EnvColorDisabled(): want %v, got %v", tt.want, got)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestEnvColorForced(t *testing.T) {
|
|
orig_NO_COLOR := os.Getenv("NO_COLOR")
|
|
orig_CLICOLOR := os.Getenv("CLICOLOR")
|
|
orig_CLICOLOR_FORCE := os.Getenv("CLICOLOR_FORCE")
|
|
t.Cleanup(func() {
|
|
os.Setenv("NO_COLOR", orig_NO_COLOR)
|
|
os.Setenv("CLICOLOR", orig_CLICOLOR)
|
|
os.Setenv("CLICOLOR_FORCE", orig_CLICOLOR_FORCE)
|
|
})
|
|
|
|
tests := []struct {
|
|
name string
|
|
NO_COLOR string
|
|
CLICOLOR string
|
|
CLICOLOR_FORCE string
|
|
want bool
|
|
}{
|
|
{
|
|
name: "pristine env",
|
|
NO_COLOR: "",
|
|
CLICOLOR: "",
|
|
CLICOLOR_FORCE: "",
|
|
want: false,
|
|
},
|
|
{
|
|
name: "NO_COLOR enabled",
|
|
NO_COLOR: "1",
|
|
CLICOLOR: "",
|
|
CLICOLOR_FORCE: "",
|
|
want: false,
|
|
},
|
|
{
|
|
name: "CLICOLOR disabled",
|
|
NO_COLOR: "",
|
|
CLICOLOR: "0",
|
|
CLICOLOR_FORCE: "",
|
|
want: false,
|
|
},
|
|
{
|
|
name: "CLICOLOR enabled",
|
|
NO_COLOR: "",
|
|
CLICOLOR: "1",
|
|
CLICOLOR_FORCE: "",
|
|
want: false,
|
|
},
|
|
{
|
|
name: "CLICOLOR_FORCE enabled",
|
|
NO_COLOR: "",
|
|
CLICOLOR: "",
|
|
CLICOLOR_FORCE: "1",
|
|
want: true,
|
|
},
|
|
{
|
|
name: "CLICOLOR_FORCE disabled",
|
|
NO_COLOR: "",
|
|
CLICOLOR: "",
|
|
CLICOLOR_FORCE: "0",
|
|
want: false,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
os.Setenv("NO_COLOR", tt.NO_COLOR)
|
|
os.Setenv("CLICOLOR", tt.CLICOLOR)
|
|
os.Setenv("CLICOLOR_FORCE", tt.CLICOLOR_FORCE)
|
|
|
|
if got := EnvColorForced(); got != tt.want {
|
|
t.Errorf("EnvColorForced(): want %v, got %v", tt.want, got)
|
|
}
|
|
})
|
|
}
|
|
}
|