chalk_test.go

package chalk

import (
	"fmt"
	"github.com/stretchr/testify/assert"
	"testing"
)

type test struct {
	expected interface{}
	actual   interface{}
}

func TestColor(t *testing.T) {
	assertions := assert.New(t)

	color := Color{1}
	tests := []test{
		{color.Value(), 1},
		{"\u001b[31m", color.String()},
		{fmt.Sprintf("%shello%s", "\u001b[31m", ResetColor), color.Set("hello")},
	}

	for _, tc := range tests {
		assertions.Equalf(tc.expected, tc.actual, "%#v should be %#v", tc.actual, tc.expected)
	}
}

func TestBrightColor(t *testing.T) {
	assertions := assert.New(t)

	color := BrightColor{Color{1}}
	tests := []test{
		{color.Value(), 1},
		{"\u001b[91m", color.String()},
		{fmt.Sprintf("%shello%s", "\u001b[91m", ResetColor), color.Set("hello")},
	}

	for _, tc := range tests {
		assertions.Equalf(tc.expected, tc.actual, "%#v should be %#v", tc.actual, tc.expected)
	}
}

func TestTextStyle(t *testing.T) {
	assertions := assert.New(t)

	textStyle := TextStyle{4, 24}
	tests := []test{
		{"\u001b[4m\u001b[24m", textStyle.String()},
		{"\u001b[4mhello\u001b[24m", textStyle.TextStyle("hello")},
		{"hello", TextStyle{}.TextStyle("hello")},
	}
	for _, tc := range tests {
		assertions.Equalf(tc.expected, tc.actual, "%#v should be %#v", tc.actual, tc.expected)
	}
}

func TestStyle(t *testing.T) {
	assertions := assert.New(t)

	colorStyle := Red.NewStyle()
	colorStyle.Foreground(ResetColor)
	colorStyle.Background(ResetColor)

	builtStyle := Red.NewStyle().
		WithForeground(Red).
		WithBackground(Blue).
		WithTextStyle(Underline)

	tests := []test{
		{"\u001b[40m\u001b[31m", Red.NewStyle().String()},
		{"\u001b[40m\u001b[30m\u001b[1mhello\u001b[22m\u001b[49m\u001b[39m", Bold.NewStyle().Style("hello")},
		{"\u001b[49m\u001b[39m", colorStyle.String()},
		{"\u001b[44m\u001b[31m\u001b[4mhello\u001b[24m\u001b[49m\u001b[39m", builtStyle.Style("hello")},
	}

	for _, tc := range tests {
		assertions.Equalf(tc.expected, tc.actual, "%#v should be %#v", tc.actual, tc.expected)
	}
}