mirror of
https://github.com/helm/helm.git
synced 2026-08-03 22:50:49 +00:00
add test for template recursion
Signed-off-by: Daniel Cheng <dcheng@us.ibm.com>
This commit is contained in:
@@ -643,3 +643,58 @@ func TestAlterFuncMap_tplinclude(t *testing.T) {
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func TestRenderRecursionLimit(t *testing.T) {
|
||||
// endless recursion should produce an error
|
||||
c := &chart.Chart{
|
||||
Metadata: &chart.Metadata{Name: "bad"},
|
||||
Templates: []*chart.File{
|
||||
{Name: "templates/base", Data: []byte(`{{include "recursion" . }}`)},
|
||||
{Name: "templates/recursion", Data: []byte(`{{define "recursion"}}{{include "recursion" . }}{{end}}`)},
|
||||
},
|
||||
}
|
||||
v := chartutil.Values{
|
||||
"Values": "",
|
||||
"Chart": c.Metadata,
|
||||
"Release": chartutil.Values{
|
||||
"Name": "TestRelease",
|
||||
},
|
||||
}
|
||||
expectErr := "rendering template has a nested reference name: recursion: unable to execute template"
|
||||
|
||||
_, err := Render(c, v)
|
||||
if err == nil || !strings.HasSuffix(err.Error(), expectErr) {
|
||||
t.Errorf("Expected err with suffix: %s", expectErr)
|
||||
}
|
||||
|
||||
// calling the same function many times is ok
|
||||
times := 4000
|
||||
phrase := "All work and no play makes Jack a dull boy"
|
||||
printFunc := `{{define "overlook"}}{{printf "` + phrase + `\n"}}{{end}}`
|
||||
var repeatedIncl string
|
||||
for i := 0; i < times; i++ {
|
||||
repeatedIncl += `{{include "overlook" . }}`
|
||||
}
|
||||
|
||||
d := &chart.Chart{
|
||||
Metadata: &chart.Metadata{Name: "overlook"},
|
||||
Templates: []*chart.File{
|
||||
{Name: "templates/quote", Data: []byte(repeatedIncl)},
|
||||
{Name: "templates/_function", Data: []byte(printFunc)},
|
||||
},
|
||||
}
|
||||
|
||||
out, err := Render(d, v)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
var expect string
|
||||
for i := 0; i < times; i++ {
|
||||
expect += phrase + "\n"
|
||||
}
|
||||
if got := out["overlook/templates/quote"]; got != expect {
|
||||
t.Errorf("Expected %q, got %q (%v)", expect, got, out)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user