zulu_test.go

package zulu_test

import (
	"testing"
	"text/template"

	"github.com/zulucmd/zulu/v2"
)

func TestAddTemplateFunctions(t *testing.T) {
	zulu.AddTemplateFunc("t", func() bool { return true })
	zulu.AddTemplateFuncs(template.FuncMap{
		"f": func() bool { return false },
		"h": func() string { return "Hello," },
		"w": func() string { return "world." }})

	c := &zulu.Command{}
	c.SetUsageTemplate(`{{if t}}{{h}}{{end}}{{if f}}{{h}}{{end}} {{w}}`)

	const expected = "Hello, world."
	if got := c.UsageString(); got != expected {
		t.Errorf("Expected UsageString: %v\nGot: %v", expected, got)
	}
}