test(templates): use existing template tests for execution fuzzing

flosch · flosch · commit c1d8de1d73e1 · 2022-09-11T04:03:13.000+02:00
diff --git a/pongo2_test.go b/pongo2_test.go
@@ -1,6 +1,10 @@
 package pongo2_test
 
 import (
+	"errors"
+	"io"
+	"io/ioutil"
+	"path/filepath"
 	"testing"
 
 	"github.com/flosch/pongo2/v6"
@@ -99,10 +103,35 @@ func (s *TestSuite) TestImplicitExecCtx(c *C) {
 	c.Check(res, Equals, val)
 }
 
+type DummyLoader struct{}
+
+func (l *DummyLoader) Abs(base, name string) string {
+	return filepath.Join(filepath.Dir(base), name)
+}
+
+func (l *DummyLoader) Get(path string) (io.Reader, error) {
+	return nil, errors.New("dummy not found")
+}
+
 func FuzzSimpleExecution(f *testing.F) {
-	f.Add("{{ foobar }}", "test123")
+	tpls, err := filepath.Glob("template_tests/*.tpl")
+	if err != nil {
+		f.Fatalf("glob: %v", err)
+	}
+	files := []string{"README.md"}
+	files = append(files, tpls...)
+
+	for _, tplPath := range files {
+		buf, err := ioutil.ReadFile(tplPath)
+		if err != nil {
+			f.Fatalf("could not read file '%v': %v", tplPath, err)
+		}
+		f.Add(string(buf), "test-value")
+	}
+
 	f.Fuzz(func(t *testing.T, tpl, contextValue string) {
-		out, err := pongo2.FromString(tpl)
+		ts := pongo2.NewSet("fuzz-test", &DummyLoader{})
+		out, err := ts.FromString(tpl)
 		if err != nil && out != nil {
 			t.Errorf("%v", err)
 		}
diff --git a/testdata/fuzz/FuzzSimpleExecution/6acbb88419cbfbe6407b637f9a4ac280ca68001ce95bdc40c04a497aecc7c9a4 b/testdata/fuzz/FuzzSimpleExecution/6acbb88419cbfbe6407b637f9a4ac280ca68001ce95bdc40c04a497aecc7c9a4
@@ -0,0 +1,3 @@
+go test fuzz v1
+string("{%macro A()%}{%if(0)%}{%endif%}0{{0}}0{{A}}0{%if 0%}{%endif%}0{%endmacro%}{%for A in A%}{%endfor%}")
+string("0")
diff --git a/testdata/fuzz/FuzzSimpleExecution/7240db62ee5a3570dd7bcd9b01a364191f98e7df3f417080d16e50817aa6e3f8 b/testdata/fuzz/FuzzSimpleExecution/7240db62ee5a3570dd7bcd9b01a364191f98e7df3f417080d16e50817aa6e3f8
@@ -0,0 +1,3 @@
+go test fuzz v1
+string("empty single line comment\n{# #}\n\nfilled single line comment\n{# testing single line comment #}\n\nfilled single line comment with valid tags\n{# testing single line comment {% if thing %}{% endif %} #}\n\nfilled single line comment with invalid tags\n{# testing single line comment {% if thing %} #}\n\nfilled single line comment with invalid syntax\n{# testing single line comment {% if thing('') %}wow{% endif %} #}\n\nempty block comment\n{% comment %}{% endcomment %}\n\nfilled text single line block comment\n{% comment %}filled block comment {% endcomment %}\n\nempty multi line block comment\n{% comment %}\n\n\n{% endcomment %}\n\nblock comment with other tags inside of it\n{% comment %}\n  {{ thing_goes_here }}\n  {% if stuff %}do stuff{% endif %}\n{% endcomment %}\n\nblock comment with invalid tags inside of it\n{% comment %}\n  {% if thing %}\n{% endcomment %}\n\nblock comment with invalid syntax inside of it\n{% comment %}\n  {% thing('') %}\n{% endcomment  \x00\n\nRegular tags between comments to verify it doesn't break in the lexer\n{% if hello %}\n{% endif %}\nafter if\n{% comment %}All done{% endcomment %}\n\nend of file\n")
+string("test-vKlue")
diff --git a/testdata/fuzz/FuzzSimpleExecution/975dda3c66610ab39dd34f256ad419591d26c8f193c0f2f83015e1cfb8633dd7 b/testdata/fuzz/FuzzSimpleExecution/975dda3c66610ab39dd34f256ad419591d26c8f193c0f2f83015e1cfb8633dd7
@@ -0,0 +1,3 @@
+go test fuzz v1
+string("empty single line comment\n{# #}\n\nfilled single line comment\n{# testing single line comment #}\n\nfilled single line comment with valid tags\n{# testing single line comment {% if thing %}{% endif %} #}\n\nfilled single line comment with invalid tags\n{# testing single line comment {% if thing %} #}\n\nfilled single line comment with invalid syntax\n{# testing single line comment {% if thing('') %}wow{% endif %} #}\n\nempty block comment\n{% comment %}{% endcommenbl")
+string("test-value")

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+go test fuzz v1`
	`2`	`+string("{%macro A()%}{%if(0)%}{%endif%}0{{0}}0{{A}}0{%if 0%}{%endif%}0{%endmacro%}{%for A in A%}{%endfor%}")`
	`3`	`+string("0")`
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+go test fuzz v1`
	`2`	+string("empty single line comment\n{# #}\n\nfilled single line comment\n{# testing single line comment #}\n\nfilled single line comment with valid tags\n{# testing single line comment {% if thing %}{% endif %} #}\n\nfilled single line comment with invalid tags\n{# testing single line comment {% if thing %} #}\n\nfilled single line comment with invalid syntax\n{# testing single line comment {% if thing('') %}wow{% endif %} #}\n\nempty block comment\n{% comment %}{% endcomment %}\n\nfilled text single line block comment\n{% comment %}filled block comment {% endcomment %}\n\nempty multi line block comment\n{% comment %}\n\n\n{% endcomment %}\n\nblock comment with other tags inside of it\n{% comment %}\n {{ thing_goes_here }}\n {% if stuff %}do stuff{% endif %}\n{% endcomment %}\n\nblock comment with invalid tags inside of it\n{% comment %}\n {% if thing %}\n{% endcomment %}\n\nblock comment with invalid syntax inside of it\n{% comment %}\n {% thing('') %}\n{% endcomment \x00\n\nRegular tags between comments to verify it doesn't break in the lexer\n{% if hello %}\n{% endif %}\nafter if\n{% comment %}All done{% endcomment %}\n\nend of file\n")
	`3`	`+string("test-vKlue")`
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+go test fuzz v1`
	`2`	`+string("empty single line comment\n{# #}\n\nfilled single line comment\n{# testing single line comment #}\n\nfilled single line comment with valid tags\n{# testing single line comment {% if thing %}{% endif %} #}\n\nfilled single line comment with invalid tags\n{# testing single line comment {% if thing %} #}\n\nfilled single line comment with invalid syntax\n{# testing single line comment {% if thing('') %}wow{% endif %} #}\n\nempty block comment\n{% comment %}{% endcommenbl")`
	`3`	`+string("test-value")`