mirror of
https://github.com/tarampampam/error-pages.git
synced 2024-08-30 18:22:40 +00:00
143 lines
3.9 KiB
Go
143 lines
3.9 KiB
Go
package tpl_test
|
|
|
|
import (
|
|
"math/rand"
|
|
"os"
|
|
"strconv"
|
|
"sync"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"gh.tarampamp.am/error-pages/internal/tpl"
|
|
)
|
|
|
|
func Test_Render(t *testing.T) {
|
|
renderer := tpl.NewTemplateRenderer()
|
|
defer func() { _ = renderer.Close() }()
|
|
|
|
require.NoError(t, os.Setenv("TEST_ENV_VAR", "unit-test"))
|
|
|
|
defer func() { require.NoError(t, os.Unsetenv("TEST_ENV_VAR")) }()
|
|
|
|
for name, tt := range map[string]struct {
|
|
giveContent string
|
|
giveProps tpl.Properties
|
|
wantContent string
|
|
wantError bool
|
|
}{
|
|
"common case": {
|
|
giveContent: "{{code}}: {{ message }} {{description}}",
|
|
giveProps: tpl.Properties{Code: "404", Message: "Not found", Description: "Blah"},
|
|
wantContent: "404: Not found Blah",
|
|
},
|
|
"html markup": {
|
|
giveContent: "<!-- comment --><html><body>{{code}}: {{ message }} {{description}}</body></html>",
|
|
giveProps: tpl.Properties{Code: "201", Message: "lorem ipsum"},
|
|
wantContent: "<!-- comment --><html><body>201: lorem ipsum </body></html>",
|
|
},
|
|
"with line breakers": {
|
|
giveContent: "\t {{code}}: {{ message }} {{description}}\n",
|
|
giveProps: tpl.Properties{},
|
|
wantContent: "\t : \n",
|
|
},
|
|
"golang template": {
|
|
giveContent: "\t {{code}} {{ .Code }}{{ if .Message }} Yeah {{end}}",
|
|
giveProps: tpl.Properties{Code: "201", Message: "lorem ipsum"},
|
|
wantContent: "\t 201 201 Yeah ",
|
|
},
|
|
"wrong golang template": {
|
|
giveContent: "{{ if foo() }} Test {{ end }}",
|
|
giveProps: tpl.Properties{},
|
|
wantError: true,
|
|
},
|
|
|
|
"json common case": {
|
|
giveContent: `{"code": {{code | json}}, "message": {"here":[ {{ message | json }} ]}, "desc": "{{description}}"}`,
|
|
giveProps: tpl.Properties{Code: `404'"{`, Message: "Not found\t\r\n"},
|
|
wantContent: `{"code": "404'\"{", "message": {"here":[ "Not found\t\r\n" ]}, "desc": ""}`,
|
|
},
|
|
"json golang template": {
|
|
giveContent: `{"code": "{{code}}", "message": {"here":[ "{{ if .Message }} Yeah {{end}}" ]}}`,
|
|
giveProps: tpl.Properties{Code: "201", Message: "lorem ipsum"},
|
|
wantContent: `{"code": "201", "message": {"here":[ " Yeah " ]}}`,
|
|
},
|
|
|
|
"fn l10n_enabled": {
|
|
giveContent: "{{ if l10n_enabled }}Y{{ else }}N{{ end }}",
|
|
giveProps: tpl.Properties{L10nDisabled: true},
|
|
wantContent: "N",
|
|
},
|
|
"fn l10n_disabled": {
|
|
giveContent: "{{ if l10n_disabled }}Y{{ else }}N{{ end }}",
|
|
giveProps: tpl.Properties{L10nDisabled: true},
|
|
wantContent: "Y",
|
|
},
|
|
|
|
"env": {
|
|
giveContent: `{{ env "TEST_ENV_VAR" }}`,
|
|
wantContent: "unit-test",
|
|
},
|
|
} {
|
|
t.Run(name, func(t *testing.T) {
|
|
content, err := renderer.Render([]byte(tt.giveContent), tt.giveProps)
|
|
|
|
if tt.wantError == true {
|
|
assert.Error(t, err)
|
|
} else {
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, tt.wantContent, string(content))
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestTemplateRenderer_Render_Concurrent(t *testing.T) {
|
|
renderer := tpl.NewTemplateRenderer()
|
|
|
|
var wg sync.WaitGroup
|
|
|
|
for i := 0; i < 100; i++ {
|
|
wg.Add(1)
|
|
|
|
go func() {
|
|
defer wg.Done()
|
|
|
|
props := tpl.Properties{
|
|
Code: strconv.Itoa(rand.Intn(599-300+1) + 300), //nolint:gosec
|
|
Message: "Not found",
|
|
Description: "Blah",
|
|
}
|
|
|
|
content, err := renderer.Render([]byte("{{code}}: {{ message }} {{description}}"), props)
|
|
|
|
assert.NoError(t, err)
|
|
assert.NotEmpty(t, content)
|
|
}()
|
|
}
|
|
|
|
wg.Wait()
|
|
|
|
assert.NoError(t, renderer.Close())
|
|
assert.EqualError(t, renderer.Close(), tpl.ErrClosed.Error())
|
|
|
|
content, err := renderer.Render([]byte{}, tpl.Properties{})
|
|
assert.Nil(t, content)
|
|
assert.EqualError(t, err, tpl.ErrClosed.Error())
|
|
}
|
|
|
|
func BenchmarkRenderHTML(b *testing.B) {
|
|
b.ReportAllocs()
|
|
|
|
renderer := tpl.NewTemplateRenderer()
|
|
defer func() { _ = renderer.Close() }()
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
_, _ = renderer.Render(
|
|
[]byte("{{code}}: {{ message }} {{description}}"),
|
|
tpl.Properties{Code: "404", Message: "Not found", Description: "Blah"},
|
|
)
|
|
}
|
|
}
|