tpl: Add strings.Repeat

This commit is contained in:
David E. Wheeler
2018-06-03 02:55:37 -04:00
committed by Bjørn Erik Pedersen
parent 07b96d16e8
commit 13435a6f60
5 changed files with 109 additions and 0 deletions

View File

@@ -16,6 +16,7 @@ package strings
import (
"fmt"
"html/template"
"math"
"testing"
"github.com/gohugoio/hugo/deps"
@@ -709,3 +710,38 @@ func TestTrimSuffix(t *testing.T) {
assert.Equal(t, test.expect, result, errMsg)
}
}
func TestRepeat(t *testing.T) {
t.Parallel()
for i, test := range []struct {
s interface{}
n interface{}
expect interface{}
}{
{"yo", "2", "yoyo"},
{"~", "16", "~~~~~~~~~~~~~~~~"},
{"<tag>", "0", ""},
{"yay", "1", "yay"},
{1221, "1", "1221"},
{1221, 2, "12211221"},
{template.HTML("<tag>"), "2", "<tag><tag>"},
{[]byte("<tag>"), 2, "<tag><tag>"},
// errors
{"", tstNoStringer{}, false},
{tstNoStringer{}, "", false},
{"hi", math.MaxInt16 + 1, false},
} {
errMsg := fmt.Sprintf("[%d] %v", i, test)
result, err := ns.Repeat(test.n, test.s)
if b, ok := test.expect.(bool); ok && !b {
require.Error(t, err, errMsg)
continue
}
require.NoError(t, err, errMsg)
assert.Equal(t, test.expect, result, errMsg)
}
}