diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-16 17:15:09 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-16 17:15:09 +0000 |
commit | ea9f089f6ad153f432b8265a99bce7d261b9c820 (patch) | |
tree | 8590b318bb54874bf1f0c597ff503f8c3b9ace75 /cobra_test.go | |
parent | Initial commit. (diff) | |
download | golang-github-spf13-cobra-ea9f089f6ad153f432b8265a99bce7d261b9c820.tar.xz golang-github-spf13-cobra-ea9f089f6ad153f432b8265a99bce7d261b9c820.zip |
Adding upstream version 1.8.0.upstream/1.8.0upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'cobra_test.go')
-rw-r--r-- | cobra_test.go | 42 |
1 files changed, 42 insertions, 0 deletions
diff --git a/cobra_test.go b/cobra_test.go new file mode 100644 index 0000000..fbb07f9 --- /dev/null +++ b/cobra_test.go @@ -0,0 +1,42 @@ +// Copyright 2013-2023 The Cobra Authors +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. + +package cobra + +import ( + "testing" + "text/template" +) + +func assertNoErr(t *testing.T, e error) { + if e != nil { + t.Error(e) + } +} + +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) + } +} |