cheat/internal/sheet/colorize_test.go

package sheet

import (
	"testing"

	"github.com/cheat/cheat/internal/config"
)

// TestColorize asserts that syntax-highlighting is correctly applied
func TestColorize(t *testing.T) {

	// mock configs
	conf := config.Config{
		Formatter: "terminal16m",
		Style:     "solarized-dark",
	}

	// mock a sheet
	s := Sheet{
		Text: "echo 'foo'",
	}

	// colorize the sheet text
	s.Colorize(conf)

	// initialize expectations
	want := "echo"
	want += " 'foo'"

	// assert
	if s.Text != want {
		t.Errorf("failed to colorize sheet: want: %s, got: %s", want, s.Text)
	}
}