src/html/template/multi_test.go - go - Git at Google

 // Copyright 2011 The Go Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style
 // license that can be found in the LICENSE file.

 // Tests for multiple-template execution, copied from text/template.

 package template

 import (
 	"archive/zip"
 	"bytes"
 	"os"
 	"testing"
 	"text/template/parse"
 )

 var multiExecTests = []execTest{
 	{"empty", "", "", nil, true},
 	{"text", "some text", "some text", nil, true},
 	{"invoke x", `{{template "x" .SI}}`, "TEXT", tVal, true},
 	{"invoke x no args", `{{template "x"}}`, "TEXT", tVal, true},
 	{"invoke dot int", `{{template "dot" .I}}`, "17", tVal, true},
 	{"invoke dot []int", `{{template "dot" .SI}}`, "[3 4 5]", tVal, true},
 	{"invoke dotV", `{{template "dotV" .U}}`, "v", tVal, true},
 	{"invoke nested int", `{{template "nested" .I}}`, "17", tVal, true},
 	{"variable declared by template", `{{template "nested" $x:=.SI}},{{index $x 1}}`, "[3 4 5],4", tVal, true},

 	// User-defined function: test argument evaluator.
 	{"testFunc literal", `{{oneArg "joe"}}`, "oneArg=joe", tVal, true},
 	{"testFunc .", `{{oneArg .}}`, "oneArg=joe", "joe", true},
 }

 // These strings are also in testdata/*.
 const multiText1 = `
 	{{define "x"}}TEXT{{end}}
 	{{define "dotV"}}{{.V}}{{end}}
 `

 const multiText2 = `
 	{{define "dot"}}{{.}}{{end}}
 	{{define "nested"}}{{template "dot" .}}{{end}}
 `

 func TestMultiExecute(t *testing.T) {
 	// Declare a couple of templates first.
 	template, err := New("root").Parse(multiText1)
 	if err != nil {
 		t.Fatalf("parse error for 1: %s", err)
 	}
 	_, err = template.Parse(multiText2)
 	if err != nil {
 		t.Fatalf("parse error for 2: %s", err)
 	}
 	testExecute(multiExecTests, template, t)
 }

 func TestParseFiles(t *testing.T) {
 	_, err := ParseFiles("DOES NOT EXIST")
 	if err == nil {
 		t.Error("expected error for non-existent file; got none")
 	}
 	template := New("root")
 	_, err = template.ParseFiles("testdata/file1.tmpl", "testdata/file2.tmpl")
 	if err != nil {
 		t.Fatalf("error parsing files: %v", err)
 	}
 	testExecute(multiExecTests, template, t)
 }

 func TestParseGlob(t *testing.T) {
 	_, err := ParseGlob("DOES NOT EXIST")
 	if err == nil {
 		t.Error("expected error for non-existent file; got none")
 	}
 	_, err = New("error").ParseGlob("[x")
 	if err == nil {
 		t.Error("expected error for bad pattern; got none")
 	}
 	template := New("root")
 	_, err = template.ParseGlob("testdata/file*.tmpl")
 	if err != nil {
 		t.Fatalf("error parsing files: %v", err)
 	}
 	testExecute(multiExecTests, template, t)
 }

 func TestParseFS(t *testing.T) {
 	fs := os.DirFS("testdata")

 	{
 		_, err := ParseFS(fs, "DOES NOT EXIST")
 		if err == nil {
 			t.Error("expected error for non-existent file; got none")
 		}
 	}

 	{
 		template := New("root")
 		_, err := template.ParseFS(fs, "file1.tmpl", "file2.tmpl")
 		if err != nil {
 			t.Fatalf("error parsing files: %v", err)
 		}
 		testExecute(multiExecTests, template, t)
 	}

 	{
 		template := New("root")
 		_, err := template.ParseFS(fs, "file*.tmpl")
 		if err != nil {
 			t.Fatalf("error parsing files: %v", err)
 		}
 		testExecute(multiExecTests, template, t)
 	}
 }

 // In these tests, actual content (not just template definitions) comes from the parsed files.

 var templateFileExecTests = []execTest{
 	{"test", `{{template "tmpl1.tmpl"}}{{template "tmpl2.tmpl"}}`, "template1\n\ny\ntemplate2\n\nx\n", 0, true},
 }

 func TestParseFilesWithData(t *testing.T) {
 	template, err := New("root").ParseFiles("testdata/tmpl1.tmpl", "testdata/tmpl2.tmpl")
 	if err != nil {
 		t.Fatalf("error parsing files: %v", err)
 	}
 	testExecute(templateFileExecTests, template, t)
 }

 func TestParseGlobWithData(t *testing.T) {
 	template, err := New("root").ParseGlob("testdata/tmpl*.tmpl")
 	if err != nil {
 		t.Fatalf("error parsing files: %v", err)
 	}
 	testExecute(templateFileExecTests, template, t)
 }

 func TestParseZipFS(t *testing.T) {
 	z, err := zip.OpenReader("testdata/fs.zip")
 	if err != nil {
 		t.Fatalf("error parsing zip: %v", err)
 	}
 	template, err := New("root").ParseFS(z, "tmpl*.tmpl")
 	if err != nil {
 		t.Fatalf("error parsing files: %v", err)
 	}
 	testExecute(templateFileExecTests, template, t)
 }

 const (
 	cloneText1 = `{{define "a"}}{{template "b"}}{{template "c"}}{{end}}`
 	cloneText2 = `{{define "b"}}b{{end}}`
 	cloneText3 = `{{define "c"}}root{{end}}`
 	cloneText4 = `{{define "c"}}clone{{end}}`
 )

 // Issue 7032
 func TestAddParseTreeToUnparsedTemplate(t *testing.T) {
 	master := "{{define \"master\"}}{{end}}"
 	tmpl := New("master")
 	tree, err := parse.Parse("master", master, "", "", nil)
 	if err != nil {
 		t.Fatalf("unexpected parse err: %v", err)
 	}
 	masterTree := tree["master"]
 	tmpl.AddParseTree("master", masterTree) // used to panic
 }

 func TestRedefinition(t *testing.T) {
 	var tmpl *Template
 	var err error
 	if tmpl, err = New("tmpl1").Parse(`{{define "test"}}foo{{end}}`); err != nil {
 		t.Fatalf("parse 1: %v", err)
 	}
 	if _, err = tmpl.Parse(`{{define "test"}}bar{{end}}`); err != nil {
 		t.Fatalf("got error %v, expected nil", err)
 	}
 	if _, err = tmpl.New("tmpl2").Parse(`{{define "test"}}bar{{end}}`); err != nil {
 		t.Fatalf("got error %v, expected nil", err)
 	}
 }

 // Issue 10879
 func TestEmptyTemplateCloneCrash(t *testing.T) {
 	t1 := New("base")
 	t1.Clone() // used to panic
 }

 // Issue 10910, 10926
 func TestTemplateLookUp(t *testing.T) {
 	t.Skip("broken on html/template") // TODO
 	t1 := New("foo")
 	if t1.Lookup("foo") != nil {
 		t.Error("Lookup returned non-nil value for undefined template foo")
 	}
 	t1.New("bar")
 	if t1.Lookup("bar") != nil {
 		t.Error("Lookup returned non-nil value for undefined template bar")
 	}
 	t1.Parse(`{{define "foo"}}test{{end}}`)
 	if t1.Lookup("foo") == nil {
 		t.Error("Lookup returned nil value for defined template")
 	}
 }

 func TestParse(t *testing.T) {
 	// In multiple calls to Parse with the same receiver template, only one call
 	// can contain text other than space, comments, and template definitions
 	t1 := New("test")
 	if _, err := t1.Parse(`{{define "test"}}{{end}}`); err != nil {
 		t.Fatalf("parsing test: %s", err)
 	}
 	if _, err := t1.Parse(`{{define "test"}}{{/* this is a comment */}}{{end}}`); err != nil {
 		t.Fatalf("parsing test: %s", err)
 	}
 	if _, err := t1.Parse(`{{define "test"}}foo{{end}}`); err != nil {
 		t.Fatalf("parsing test: %s", err)
 	}
 }

 func TestEmptyTemplate(t *testing.T) {
 	cases := []struct {
 		defn []string
 		in   string
 		want string
 	}{
 		{[]string{"x", "y"}, "", "y"},
 		{[]string{""}, "once", ""},
 		{[]string{"", ""}, "twice", ""},
 		{[]string{"{{.}}", "{{.}}"}, "twice", "twice"},
 		{[]string{"{{/* a comment */}}", "{{/* a comment */}}"}, "comment", ""},
 		{[]string{"{{.}}", ""}, "twice", "twice"}, // TODO: should want "" not "twice"
 	}

 	for i, c := range cases {
 		root := New("root")

 		var (
 			m   *Template
 			err error
 		)
 		for _, d := range c.defn {
 			m, err = root.New(c.in).Parse(d)
 			if err != nil {
 				t.Fatal(err)
 			}
 		}
 		buf := &bytes.Buffer{}
 		if err := m.Execute(buf, c.in); err != nil {
 			t.Error(i, err)
 			continue
 		}
 		if buf.String() != c.want {
 			t.Errorf("expected string %q: got %q", c.want, buf.String())
 		}
 	}
 }

 // Issue 19249 was a regression in 1.8 caused by the handling of empty
 // templates added in that release, which got different answers depending
 // on the order templates appeared in the internal map.
 func TestIssue19294(t *testing.T) {
 	// The empty block in "xhtml" should be replaced during execution
 	// by the contents of "stylesheet", but if the internal map associating
 	// names with templates is built in the wrong order, the empty block
 	// looks non-empty and this doesn't happen.
 	var inlined = map[string]string{
 		"stylesheet": `{{define "stylesheet"}}stylesheet{{end}}`,
 		"xhtml":      `{{block "stylesheet" .}}{{end}}`,
 	}
 	all := []string{"stylesheet", "xhtml"}
 	for i := 0; i < 100; i++ {
 		res, err := New("title.xhtml").Parse(`{{template "xhtml" .}}`)
 		if err != nil {
 			t.Fatal(err)
 		}
 		for _, name := range all {
 			_, err := res.New(name).Parse(inlined[name])
 			if err != nil {
 				t.Fatal(err)
 			}
 		}
 		var buf bytes.Buffer
 		res.Execute(&buf, 0)
 		if buf.String() != "stylesheet" {
 			t.Fatalf("iteration %d: got %q; expected %q", i, buf.String(), "stylesheet")
 		}
 	}
 }
	// Copyright 2011 The Go Authors. All rights reserved.
	// Use of this source code is governed by a BSD-style
	// license that can be found in the LICENSE file.

	// Tests for multiple-template execution, copied from text/template.

	package template

	import (
	"archive/zip"
	"bytes"
	"os"
	"testing"
	"text/template/parse"
	)

	var multiExecTests = []execTest{
	{"empty", "", "", nil, true},
	{"text", "some text", "some text", nil, true},
	{"invoke x", `{{template "x" .SI}}`, "TEXT", tVal, true},
	{"invoke x no args", `{{template "x"}}`, "TEXT", tVal, true},
	{"invoke dot int", `{{template "dot" .I}}`, "17", tVal, true},
	{"invoke dot []int", `{{template "dot" .SI}}`, "[3 4 5]", tVal, true},
	{"invoke dotV", `{{template "dotV" .U}}`, "v", tVal, true},
	{"invoke nested int", `{{template "nested" .I}}`, "17", tVal, true},
	{"variable declared by template", `{{template "nested" $x:=.SI}},{{index $x 1}}`, "[3 4 5],4", tVal, true},

	// User-defined function: test argument evaluator.
	{"testFunc literal", `{{oneArg "joe"}}`, "oneArg=joe", tVal, true},
	{"testFunc .", `{{oneArg .}}`, "oneArg=joe", "joe", true},
	}

	// These strings are also in testdata/*.
	const multiText1 = `
	{{define "x"}}TEXT{{end}}
	{{define "dotV"}}{{.V}}{{end}}
	`

	const multiText2 = `
	{{define "dot"}}{{.}}{{end}}
	{{define "nested"}}{{template "dot" .}}{{end}}
	`

	func TestMultiExecute(t *testing.T) {
	// Declare a couple of templates first.
	template, err := New("root").Parse(multiText1)
	if err != nil {
	t.Fatalf("parse error for 1: %s", err)
	}
	_, err = template.Parse(multiText2)
	if err != nil {
	t.Fatalf("parse error for 2: %s", err)
	}
	testExecute(multiExecTests, template, t)
	}

	func TestParseFiles(t *testing.T) {
	_, err := ParseFiles("DOES NOT EXIST")
	if err == nil {
	t.Error("expected error for non-existent file; got none")
	}
	template := New("root")
	_, err = template.ParseFiles("testdata/file1.tmpl", "testdata/file2.tmpl")
	if err != nil {
	t.Fatalf("error parsing files: %v", err)
	}
	testExecute(multiExecTests, template, t)
	}

	func TestParseGlob(t *testing.T) {
	_, err := ParseGlob("DOES NOT EXIST")
	if err == nil {
	t.Error("expected error for non-existent file; got none")
	}
	_, err = New("error").ParseGlob("[x")
	if err == nil {
	t.Error("expected error for bad pattern; got none")
	}
	template := New("root")
	_, err = template.ParseGlob("testdata/file*.tmpl")
	if err != nil {
	t.Fatalf("error parsing files: %v", err)
	}
	testExecute(multiExecTests, template, t)
	}

	func TestParseFS(t *testing.T) {
	fs := os.DirFS("testdata")

	{
	_, err := ParseFS(fs, "DOES NOT EXIST")
	if err == nil {
	t.Error("expected error for non-existent file; got none")
	}
	}

	{
	template := New("root")
	_, err := template.ParseFS(fs, "file1.tmpl", "file2.tmpl")
	if err != nil {
	t.Fatalf("error parsing files: %v", err)
	}
	testExecute(multiExecTests, template, t)
	}

	{
	template := New("root")
	_, err := template.ParseFS(fs, "file*.tmpl")
	if err != nil {
	t.Fatalf("error parsing files: %v", err)
	}
	testExecute(multiExecTests, template, t)
	}
	}

	// In these tests, actual content (not just template definitions) comes from the parsed files.

	var templateFileExecTests = []execTest{
	{"test", `{{template "tmpl1.tmpl"}}{{template "tmpl2.tmpl"}}`, "template1\n\ny\ntemplate2\n\nx\n", 0, true},
	}

	func TestParseFilesWithData(t *testing.T) {
	template, err := New("root").ParseFiles("testdata/tmpl1.tmpl", "testdata/tmpl2.tmpl")
	if err != nil {
	t.Fatalf("error parsing files: %v", err)
	}
	testExecute(templateFileExecTests, template, t)
	}

	func TestParseGlobWithData(t *testing.T) {
	template, err := New("root").ParseGlob("testdata/tmpl*.tmpl")
	if err != nil {
	t.Fatalf("error parsing files: %v", err)
	}
	testExecute(templateFileExecTests, template, t)
	}

	func TestParseZipFS(t *testing.T) {
	z, err := zip.OpenReader("testdata/fs.zip")
	if err != nil {
	t.Fatalf("error parsing zip: %v", err)
	}
	template, err := New("root").ParseFS(z, "tmpl*.tmpl")
	if err != nil {
	t.Fatalf("error parsing files: %v", err)
	}
	testExecute(templateFileExecTests, template, t)
	}

	const (
	cloneText1 = `{{define "a"}}{{template "b"}}{{template "c"}}{{end}}`
	cloneText2 = `{{define "b"}}b{{end}}`
	cloneText3 = `{{define "c"}}root{{end}}`
	cloneText4 = `{{define "c"}}clone{{end}}`
	)

	// Issue 7032
	func TestAddParseTreeToUnparsedTemplate(t *testing.T) {
	master := "{{define \"master\"}}{{end}}"
	tmpl := New("master")
	tree, err := parse.Parse("master", master, "", "", nil)
	if err != nil {
	t.Fatalf("unexpected parse err: %v", err)
	}
	masterTree := tree["master"]
	tmpl.AddParseTree("master", masterTree) // used to panic
	}

	func TestRedefinition(t *testing.T) {
	var tmpl *Template
	var err error
	if tmpl, err = New("tmpl1").Parse(`{{define "test"}}foo{{end}}`); err != nil {
	t.Fatalf("parse 1: %v", err)
	}
	if _, err = tmpl.Parse(`{{define "test"}}bar{{end}}`); err != nil {
	t.Fatalf("got error %v, expected nil", err)
	}
	if _, err = tmpl.New("tmpl2").Parse(`{{define "test"}}bar{{end}}`); err != nil {
	t.Fatalf("got error %v, expected nil", err)
	}
	}

	// Issue 10879
	func TestEmptyTemplateCloneCrash(t *testing.T) {
	t1 := New("base")
	t1.Clone() // used to panic
	}

	// Issue 10910, 10926
	func TestTemplateLookUp(t *testing.T) {
	t.Skip("broken on html/template") // TODO
	t1 := New("foo")
	if t1.Lookup("foo") != nil {
	t.Error("Lookup returned non-nil value for undefined template foo")
	}
	t1.New("bar")
	if t1.Lookup("bar") != nil {
	t.Error("Lookup returned non-nil value for undefined template bar")
	}
	t1.Parse(`{{define "foo"}}test{{end}}`)
	if t1.Lookup("foo") == nil {
	t.Error("Lookup returned nil value for defined template")
	}
	}

	func TestParse(t *testing.T) {
	// In multiple calls to Parse with the same receiver template, only one call
	// can contain text other than space, comments, and template definitions
	t1 := New("test")
	if _, err := t1.Parse(`{{define "test"}}{{end}}`); err != nil {
	t.Fatalf("parsing test: %s", err)
	}
	if _, err := t1.Parse(`{{define "test"}}{{/* this is a comment */}}{{end}}`); err != nil {
	t.Fatalf("parsing test: %s", err)
	}
	if _, err := t1.Parse(`{{define "test"}}foo{{end}}`); err != nil {
	t.Fatalf("parsing test: %s", err)
	}
	}

	func TestEmptyTemplate(t *testing.T) {
	cases := []struct {
	defn []string
	in string
	want string
	}{
	{[]string{"x", "y"}, "", "y"},
	{[]string{""}, "once", ""},
	{[]string{"", ""}, "twice", ""},
	{[]string{"{{.}}", "{{.}}"}, "twice", "twice"},
	{[]string{"{{/* a comment /}}", "{{/ a comment */}}"}, "comment", ""},
	{[]string{"{{.}}", ""}, "twice", "twice"}, // TODO: should want "" not "twice"
	}

	for i, c := range cases {
	root := New("root")

	var (
	m *Template
	err error
	)
	for _, d := range c.defn {
	m, err = root.New(c.in).Parse(d)
	if err != nil {
	t.Fatal(err)
	}
	}
	buf := &bytes.Buffer{}
	if err := m.Execute(buf, c.in); err != nil {
	t.Error(i, err)
	continue
	}
	if buf.String() != c.want {
	t.Errorf("expected string %q: got %q", c.want, buf.String())
	}
	}
	}

	// Issue 19249 was a regression in 1.8 caused by the handling of empty
	// templates added in that release, which got different answers depending
	// on the order templates appeared in the internal map.
	func TestIssue19294(t *testing.T) {
	// The empty block in "xhtml" should be replaced during execution
	// by the contents of "stylesheet", but if the internal map associating
	// names with templates is built in the wrong order, the empty block
	// looks non-empty and this doesn't happen.
	var inlined = map[string]string{
	"stylesheet": `{{define "stylesheet"}}stylesheet{{end}}`,
	"xhtml": `{{block "stylesheet" .}}{{end}}`,
	}
	all := []string{"stylesheet", "xhtml"}
	for i := 0; i < 100; i++ {
	res, err := New("title.xhtml").Parse(`{{template "xhtml" .}}`)
	if err != nil {
	t.Fatal(err)
	}
	for _, name := range all {
	_, err := res.New(name).Parse(inlined[name])
	if err != nil {
	t.Fatal(err)
	}
	}
	var buf bytes.Buffer
	res.Execute(&buf, 0)
	if buf.String() != "stylesheet" {
	t.Fatalf("iteration %d: got %q; expected %q", i, buf.String(), "stylesheet")
	}
	}
	}