summaryrefslogtreecommitdiffstats
path: root/tpl/collections/collections_test.go
diff options
context:
space:
mode:
authorBjørn Erik Pedersen <bjorn.erik.pedersen@gmail.com>2018-09-08 13:00:36 +0200
committerBjørn Erik Pedersen <bjorn.erik.pedersen@gmail.com>2018-09-08 20:20:26 +0200
commit6667c6d7430acc16b3683fbbacd263f1d00c8672 (patch)
treea238b6bf5a2e02da09e79eafa4784e7f51441144 /tpl/collections/collections_test.go
parentcfda13b36367465016f4458ab9924c948ed02b6f (diff)
tpl/collections: Add group template func
This extends the page grouping in Hugo with a template function that allows for ad-hoc grouping. A made-up example: ``` {{ $cool := where .Site.RegularPages "Params.cool" true | group "cool" }} {{ $blue := where .Site.RegularPages "Params.blue" true | group "blue" }} {{ $paginator := .Paginate (slice $cool $blue) }} ``` Closes #4865
Diffstat (limited to 'tpl/collections/collections_test.go')
-rw-r--r--tpl/collections/collections_test.go41
1 files changed, 41 insertions, 0 deletions
diff --git a/tpl/collections/collections_test.go b/tpl/collections/collections_test.go
index c3e88f36f..cbcf819c7 100644
--- a/tpl/collections/collections_test.go
+++ b/tpl/collections/collections_test.go
@@ -75,6 +75,47 @@ func TestAfter(t *testing.T) {
}
}
+type tstGrouper struct {
+}
+
+type tstGroupers []*tstGrouper
+
+func (g tstGrouper) Group(key interface{}, items interface{}) (interface{}, error) {
+ ilen := reflect.ValueOf(items).Len()
+ return fmt.Sprintf("%v(%d)", key, ilen), nil
+}
+
+func TestGroup(t *testing.T) {
+ t.Parallel()
+
+ ns := New(&deps.Deps{})
+
+ for i, test := range []struct {
+ key interface{}
+ items interface{}
+ expect interface{}
+ }{
+ {"a", []*tstGrouper{&tstGrouper{}, &tstGrouper{}}, "a(2)"},
+ {"b", tstGroupers{&tstGrouper{}, &tstGrouper{}}, "b(2)"},
+ {"a", []tstGrouper{tstGrouper{}, tstGrouper{}}, "a(2)"},
+ {"a", []*tstGrouper{}, "a(0)"},
+ {"a", []string{"a", "b"}, false},
+ {nil, []*tstGrouper{&tstGrouper{}, &tstGrouper{}}, false},
+ } {
+ errMsg := fmt.Sprintf("[%d] %v", i, test)
+
+ result, err := ns.Group(test.key, test.items)
+
+ if b, ok := test.expect.(bool); ok && !b {
+ require.Error(t, err, errMsg)
+ continue
+ }
+
+ require.NoError(t, err, errMsg)
+ require.Equal(t, test.expect, result, errMsg)
+ }
+}
+
func TestDelimit(t *testing.T) {
t.Parallel()