snapshot of https://github.com/cortezaproject/corteza.git (2024.9.x, 2026-09-18; blobs > 131072B omitted)

server/system/renderer/generic_html_test.go

@@ -0,0 +1,103 @@
+package renderer
+
+import (
+	"bytes"
+	"context"
+	"io"
+	"strings"
+	"testing"
+)
+
+func TestGenericHTMLHeaderFooter(t *testing.T) {
+	d := &genericHTMLDriver{}
+
+	pl := &driverPayload{
+		Template:  bytes.NewBufferString(`<main>{{template "pp" .}}</main>`),
+		Header:    bytes.NewBufferString(`<header>{{.name}} {{template "pp" .}}</header>`),
+		Footer:    bytes.NewBufferString(`<footer>{{.name}}</footer>`),
+		Variables: map[string]interface{}{"name": "Human"},
+		Partials: map[string]io.Reader{
+			"pp": bytes.NewBufferString(`{{define "pp"}}partial:{{.name}}{{end}}`),
+		},
+	}
+
+	out, err := d.Render(context.Background(), pl)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	bb, _ := io.ReadAll(out)
+	got := string(bb)
+	want := `<header>Human partial:Human</header><main>partial:Human</main><footer>Human</footer>`
+	if got != want {
+		t.Fatalf("got %q, want %q", got, want)
+	}
+
+	// no header/footer keeps the single-pass path working
+	pl2 := &driverPayload{
+		Template:  bytes.NewBufferString(`<main>ok</main>`),
+		Variables: map[string]interface{}{},
+	}
+	out2, err := d.Render(context.Background(), pl2)
+	if err != nil {
+		t.Fatal(err)
+	}
+	bb2, _ := io.ReadAll(out2)
+	if !strings.Contains(string(bb2), "<main>ok</main>") {
+		t.Fatalf("plain render broken: %q", string(bb2))
+	}
+}
+
+func TestGenericHTMLSafeHTML(t *testing.T) {
+	d := &genericHTMLDriver{}
+	rte := `<h1>Title 1</h1>`
+
+	pl := &driverPayload{
+		Template:  bytes.NewBufferString(`<div>{{.body}}</div><div>{{.body | safeHTML}}</div>`),
+		Variables: map[string]interface{}{"body": rte},
+	}
+
+	out, err := d.Render(context.Background(), pl)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	bb, _ := io.ReadAll(out)
+	got := string(bb)
+	want := `<div>&lt;h1&gt;Title 1&lt;/h1&gt;</div><div><h1>Title 1</h1></div>`
+	if got != want {
+		t.Fatalf("got %q, want %q", got, want)
+	}
+
+	// the case the partial workaround could not cover: rich text in a list
+	pl2 := &driverPayload{
+		Template:  bytes.NewBufferString(`{{range .items}}<td>{{.desc | safeHTML}}</td>{{end}}`),
+		Variables: map[string]interface{}{"items": []map[string]any{{"desc": rte}, {"desc": rte}}},
+	}
+
+	out2, err := d.Render(context.Background(), pl2)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	bb2, _ := io.ReadAll(out2)
+	if got, want := string(bb2), `<td><h1>Title 1</h1></td><td><h1>Title 1</h1></td>`; got != want {
+		t.Fatalf("got %q, want %q", got, want)
+	}
+
+	// an unset or empty value must not break the render of the whole document
+	pl3 := &driverPayload{
+		Template:  bytes.NewBufferString(`<td>{{.unset | safeHTML}}</td><td>{{.blank | safeHTML}}</td><td>{{.count | safeHTML}}</td>`),
+		Variables: map[string]interface{}{"blank": nil, "count": 3},
+	}
+
+	out3, err := d.Render(context.Background(), pl3)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	bb3, _ := io.ReadAll(out3)
+	if got, want := string(bb3), `<td></td><td></td><td>3</td>`; got != want {
+		t.Fatalf("got %q, want %q", got, want)
+	}
+}