ansi_test.go

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")
	}
}