mirror of
https://github.com/gohugoio/hugo.git
synced 2025-08-16 20:53:59 +02:00
Add Substr and Split template functions
Both of these can take any type the cast lib can turn into a string.
This commit is contained in:
@@ -276,6 +276,63 @@ func TestIn(t *testing.T) {
|
||||
}
|
||||
}
|
||||
|
||||
func TestSubstr(t *testing.T) {
|
||||
for i, this := range []struct {
|
||||
v1 interface{}
|
||||
v2 int
|
||||
v3 int
|
||||
expect string
|
||||
}{
|
||||
{"abc", 1, 2, "b"},
|
||||
{"abc", 1, 3, "bc"},
|
||||
{"abc", 0, 1, "a"},
|
||||
} {
|
||||
result, err := Substr(this.v1, this.v2, this.v3)
|
||||
|
||||
if err != nil {
|
||||
t.Errorf("[%d] failed: %s", i, err)
|
||||
continue
|
||||
}
|
||||
|
||||
if result != this.expect {
|
||||
t.Errorf("[%d] Got %v but expected %v", i, result, this.expect)
|
||||
}
|
||||
}
|
||||
|
||||
_, err := Substr(tstNoStringer{}, 0, 1)
|
||||
if err == nil {
|
||||
t.Error("Expected error for non-string-convertable variable")
|
||||
}
|
||||
}
|
||||
|
||||
func TestSplit(t *testing.T) {
|
||||
for i, this := range []struct {
|
||||
v1 interface{}
|
||||
v2 string
|
||||
expect []string
|
||||
}{
|
||||
{"a, b", ", ", []string{"a", "b"}},
|
||||
{"a & b & c", " & ", []string{"a", "b", "c"}},
|
||||
{"http://exmaple.com", "http://", []string{"", "exmaple.com"}},
|
||||
} {
|
||||
result, err := Split(this.v1, this.v2)
|
||||
|
||||
if err != nil {
|
||||
t.Errorf("[%d] failed: %s", i, err)
|
||||
continue
|
||||
}
|
||||
|
||||
if !reflect.DeepEqual(result, this.expect) {
|
||||
t.Errorf("[%d] Got %s but expected %s", i, result, this.expect)
|
||||
}
|
||||
}
|
||||
|
||||
_, err := Split(tstNoStringer{}, ",")
|
||||
if err == nil {
|
||||
t.Error("Expected error for non-string-convertable variable")
|
||||
}
|
||||
}
|
||||
|
||||
func TestIntersect(t *testing.T) {
|
||||
for i, this := range []struct {
|
||||
sequence1 interface{}
|
||||
|
Reference in New Issue
Block a user