builder_test.go

package marker

import (
	"fmt"
	"testing"

	"github.com/fatih/color"
	"github.com/stretchr/testify/assert"
)

func Test_Builder(t *testing.T) {
	blueFg := color.New(color.FgBlue)
	blueFg.EnableColor()
	blue := blueFg.SprintFunc()

	redFg := color.New(color.FgRed)
	redFg.EnableColor()
	red := redFg.SprintFunc()

	b := MarkBuilder{}

	actualString := b.SetString("Skydome is a data company.").
		Mark(MatchAll("Skydome"), blueFg).
		MarkMany(redFg, MatchAll("data"), MatchAll("company")).
		Build()

	expectedString := fmt.Sprintf("%s is a %s %s.", blue("Skydome"), red("data"), red("company"))

	assert.Equal(t, expectedString, actualString)
}

func Benchmark_Builder(b *testing.B) {
	blueFg := color.New(color.FgBlue)
	blueFg.EnableColor()

	redFg := color.New(color.FgRed)
	redFg.EnableColor()

	b.ReportAllocs()
	builder := MarkBuilder{}

	skydomeMatcher := MatchAll("Skydome")

	for i := 0; i < b.N; i++ {
		builder.SetString("Skydome is a data company.").
			Mark(skydomeMatcher, blueFg).
			Build()
	}
}