mirror of
https://github.com/fatedier/frp.git
synced 2024-12-14 18:50:59 +01:00
23 lines
547 B
Go
23 lines
547 B
Go
|
package cobra
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
"text/template"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestAddTemplateFunctions(t *testing.T) {
|
||
|
|
AddTemplateFunc("t", func() bool { return true })
|
||
|
|
AddTemplateFuncs(template.FuncMap{
|
||
|
|
"f": func() bool { return false },
|
||
|
|
"h": func() string { return "Hello," },
|
||
|
|
"w": func() string { return "world." }})
|
||
|
|
|
||
|
|
c := &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)
|
||
|
|
}
|
||
|
|
}
|