blob: 68fe85dedfec38ff976ffa683b11b4405877ab70 [file] [log] [blame]
package ansi
import (
"fmt"
"strings"
"testing"
)
func TestPlain(t *testing.T) {
DisableColors(true)
PrintStyles()
}
func TestStyles(t *testing.T) {
DisableColors(false)
PrintStyles()
}
func TestDisableColors(t *testing.T) {
fn := ColorFunc("red")
buf := colorCode("off")
if buf.String() != "" {
t.Fail()
}
DisableColors(true)
if Black != "" {
t.Fail()
}
code := ColorCode("red")
if code != "" {
t.Fail()
}
s := fn("foo")
if s != "foo" {
t.Fail()
}
DisableColors(false)
if Black == "" {
t.Fail()
}
code = ColorCode("red")
if code == "" {
t.Fail()
}
// will have escape codes around it
index := strings.Index(fn("foo"), "foo")
if index <= 0 {
t.Fail()
}
}
func TestAttributeReset(t *testing.T) {
boldRed := ColorCode("red+b")
greenUnderline := ColorCode("green+u")
s := fmt.Sprintf("normal %s bold red %s green underline %s", boldRed, greenUnderline, Reset)
// See the results on the terminal for regression tests.
fmt.Printf("Colored string: %s\n", s)
fmt.Printf("Escaped string: %q\n", s)
if s != "normal \x1b[0;1;31m bold red \x1b[0;4;32m green underline \x1b[0m" {
t.Error("Attributes are not being reset")
}
}