page_data_test.go (1249B)
1 // Copyright 2019 The Hugo Authors. All rights reserved.
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License");
4 // you may not use this file except in compliance with the License.
5 // You may obtain a copy of the License at
6 // http://www.apache.org/licenses/LICENSE-2.0
7 //
8 // Unless required by applicable law or agreed to in writing, software
9 // distributed under the License is distributed on an "AS IS" BASIS,
10 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
11 // See the License for the specific language governing permissions and
12 // limitations under the License.
13
14 package page
15
16 import (
17 "bytes"
18 "testing"
19 "text/template"
20
21 qt "github.com/frankban/quicktest"
22 )
23
24 func TestPageData(t *testing.T) {
25 c := qt.New(t)
26
27 data := make(Data)
28
29 c.Assert(data.Pages(), qt.IsNil)
30
31 pages := Pages{
32 &testPage{title: "a1"},
33 &testPage{title: "a2"},
34 }
35
36 data["pages"] = pages
37
38 c.Assert(data.Pages(), eq, pages)
39
40 data["pages"] = func() Pages {
41 return pages
42 }
43
44 c.Assert(data.Pages(), eq, pages)
45
46 templ, err := template.New("").Parse(`Pages: {{ .Pages }}`)
47
48 c.Assert(err, qt.IsNil)
49
50 var buff bytes.Buffer
51
52 c.Assert(templ.Execute(&buff, data), qt.IsNil)
53
54 c.Assert(buff.String(), qt.Contains, "Pages(2)")
55 }