101 lines
2.5 KiB
Go
101 lines
2.5 KiB
Go
package handlebars
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"path"
|
|
"strconv"
|
|
"testing"
|
|
|
|
"github.com/aymerick/raymond"
|
|
)
|
|
|
|
// cf. https://github.com/aymerick/go-fuzz-tests/raymond
|
|
const DUMP_TPL = false
|
|
|
|
var dump_tpl_nb = 0
|
|
|
|
type Test struct {
|
|
name string
|
|
input string
|
|
data interface{}
|
|
privData map[string]interface{}
|
|
helpers map[string]interface{}
|
|
partials map[string]string
|
|
output interface{}
|
|
}
|
|
|
|
func launchTests(t *testing.T, tests []Test) {
|
|
t.Parallel()
|
|
|
|
for _, test := range tests {
|
|
var err error
|
|
var tpl *raymond.Template
|
|
|
|
if DUMP_TPL {
|
|
filename := strconv.Itoa(dump_tpl_nb)
|
|
if err := ioutil.WriteFile(path.Join(".", "dump_tpl", filename), []byte(test.input), 0644); err != nil {
|
|
panic(err)
|
|
}
|
|
dump_tpl_nb += 1
|
|
}
|
|
|
|
// parse template
|
|
tpl, err = raymond.Parse(test.input)
|
|
if err != nil {
|
|
t.Errorf("Test '%s' failed - Failed to parse template\ninput:\n\t'%s'\nerror:\n\t%s", test.name, test.input, err)
|
|
} else {
|
|
if len(test.helpers) > 0 {
|
|
// register helpers
|
|
tpl.RegisterHelpers(test.helpers)
|
|
}
|
|
|
|
if len(test.partials) > 0 {
|
|
// register partials
|
|
tpl.RegisterPartials(test.partials)
|
|
}
|
|
|
|
// setup private data frame
|
|
var privData *raymond.DataFrame
|
|
if test.privData != nil {
|
|
privData = raymond.NewDataFrame()
|
|
for k, v := range test.privData {
|
|
privData.Set(k, v)
|
|
}
|
|
}
|
|
|
|
// render template
|
|
output, err := tpl.ExecWith(test.data, privData)
|
|
if err != nil {
|
|
t.Errorf("Test '%s' failed\ninput:\n\t'%s'\ndata:\n\t%s\nerror:\n\t%s\nAST:\n\t%s", test.name, test.input, raymond.Str(test.data), err, tpl.PrintAST())
|
|
} else {
|
|
// check output
|
|
var expectedArr []string
|
|
expectedArr, ok := test.output.([]string)
|
|
if ok {
|
|
match := false
|
|
for _, expectedStr := range expectedArr {
|
|
if expectedStr == output {
|
|
match = true
|
|
break
|
|
}
|
|
}
|
|
|
|
if !match {
|
|
t.Errorf("Test '%s' failed\ninput:\n\t'%s'\ndata:\n\t%s\npartials:\n\t%s\nexpected\n\t%q\ngot\n\t%q\nAST:\n%s", test.name, test.input, raymond.Str(test.data), raymond.Str(test.partials), expectedArr, output, tpl.PrintAST())
|
|
}
|
|
} else {
|
|
expectedStr, ok := test.output.(string)
|
|
if !ok {
|
|
panic(fmt.Errorf("Erroneous test output description: %q", test.output))
|
|
}
|
|
|
|
if expectedStr != output {
|
|
t.Errorf("Test '%s' failed\ninput:\n\t'%s'\ndata:\n\t%s\npartials:\n\t%s\nexpected\n\t%q\ngot\n\t%q\nAST:\n%s", test.name, test.input, raymond.Str(test.data), raymond.Str(test.partials), expectedStr, output, tpl.PrintAST())
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|