forked from evergreen-ci/gimlet
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathrender_test.go
139 lines (113 loc) · 3.45 KB
/
render_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
package gimlet
import (
"bytes"
"net/http"
"net/http/httptest"
"testing"
"github.com/stretchr/testify/suite"
)
type RenderSuite struct {
render Renderer
opts RendererOptions
constructor func(RendererOptions) Renderer
testData interface{}
expected string
suite.Suite
}
func TestHTMLRenderSuite(t *testing.T) {
s := new(RenderSuite)
s.constructor = func(opts RendererOptions) Renderer {
return NewHTMLRenderer(opts)
}
suite.Run(t, s)
}
func TestTextRenderSuite(t *testing.T) {
s := new(RenderSuite)
s.constructor = func(opts RendererOptions) Renderer {
return NewTextRenderer(opts)
}
suite.Run(t, s)
}
func (s *RenderSuite) SetupSuite() {
s.expected = `<html><body><div>menu</div><p>hello Socrates setarcoS</p></body></html>`
s.testData = struct {
Name string
}{"Socrates"}
s.opts = RendererOptions{
Directory: "testdata",
Functions: map[string]interface{}{
"Reverse": func(in string) string {
runes := []rune(in)
for i, j := 0, len(runes)-1; i < j; i, j = i+1, j-1 {
runes[i], runes[j] = runes[j], runes[i]
}
return string(runes)
},
},
}
}
func (s *RenderSuite) SetupTest() {
s.render = s.constructor(s.opts)
}
func (s *RenderSuite) TestBaseHTML() {
for i := 0; i < 10; i++ {
out := &bytes.Buffer{}
err := s.render.Render(out, s.testData, "base", "test1.html", "test2.html")
s.NoError(err)
s.Equal(s.expected, out.String())
}
if r, ok := s.render.(*htmlRenderer); ok {
for k, v := range r.cache {
out := &bytes.Buffer{}
err := v.ExecuteTemplate(out, "base", s.testData)
s.NoError(err)
r.cache[k] = v
}
out := &bytes.Buffer{}
err := s.render.Render(out, s.testData, "base", "test1.html", "test2.html")
s.Error(err)
}
}
func (s *RenderSuite) TestBadTemplates() {
out := &bytes.Buffer{}
err := s.render.Render(out, s.testData, "base", "invalid_template.html")
s.Error(err)
err = s.render.Render(out, s.testData, "base", "template_does_not_exist.html")
s.Error(err)
err = s.render.Render(out, s.testData, "base", "badtemplate.html")
s.Error(err)
req, err := http.NewRequest("GET", "http://example.com/foo", nil)
s.Require().NoError(err)
htmlHandler := func(w http.ResponseWriter, r *http.Request) {
s.render.WriteResponse(w, http.StatusOK, s.testData, "badtemplate.html")
}
w := httptest.NewRecorder()
htmlHandler(w, req)
s.Equal(http.StatusInternalServerError, w.Code)
}
func (s *RenderSuite) TestWriteHTTP() {
req, err := http.NewRequest("GET", "http://example.com/foo", nil)
s.Require().NoError(err)
/* Test a handler that writes a rendered HTML template */
w := httptest.NewRecorder()
htmlHandler := func(w http.ResponseWriter, r *http.Request) {
s.render.WriteResponse(w, http.StatusOK, s.testData, "base", "test1.html", "test2.html")
}
htmlHandler(w, req)
s.Equal(s.expected, w.Body.String())
s.Equal(http.StatusOK, w.Code)
w = httptest.NewRecorder()
s.render.Stream(w, http.StatusOK, s.testData, "base", "test1.html", "test2.html")
s.Equal(http.StatusOK, w.Code)
w = httptest.NewRecorder()
s.render.Stream(w, http.StatusOK, s.testData, "base", "test1.html", "test2.html")
s.Equal(http.StatusOK, w.Code)
}
func (s *RenderSuite) TestRenderingErrorToResponse() {
w := httptest.NewRecorder()
s.render.Stream(w, http.StatusOK, s.testData, "base2", "test1.html", "test2.html2")
s.Equal(http.StatusOK, w.Code)
w = httptest.NewRecorder()
s.render.Stream(w, http.StatusOK, s.testData, "base2", "test1.html", "test2.html2")
s.Equal(http.StatusOK, w.Code)
}