1
2
3
4
5 package parse
6
7 import (
8 "flag"
9 "fmt"
10 "strings"
11 "testing"
12 )
13
14 var debug = flag.Bool("debug", false, "show the errors produced by the main tests")
15
16 type numberTest struct {
17 text string
18 isInt bool
19 isUint bool
20 isFloat bool
21 isComplex bool
22 int64
23 uint64
24 float64
25 complex128
26 }
27
28 var numberTests = []numberTest{
29
30 {"0", true, true, true, false, 0, 0, 0, 0},
31 {"-0", true, true, true, false, 0, 0, 0, 0},
32 {"73", true, true, true, false, 73, 73, 73, 0},
33 {"7_3", true, true, true, false, 73, 73, 73, 0},
34 {"0b10_010_01", true, true, true, false, 73, 73, 73, 0},
35 {"0B10_010_01", true, true, true, false, 73, 73, 73, 0},
36 {"073", true, true, true, false, 073, 073, 073, 0},
37 {"0o73", true, true, true, false, 073, 073, 073, 0},
38 {"0O73", true, true, true, false, 073, 073, 073, 0},
39 {"0x73", true, true, true, false, 0x73, 0x73, 0x73, 0},
40 {"0X73", true, true, true, false, 0x73, 0x73, 0x73, 0},
41 {"0x7_3", true, true, true, false, 0x73, 0x73, 0x73, 0},
42 {"-73", true, false, true, false, -73, 0, -73, 0},
43 {"+73", true, false, true, false, 73, 0, 73, 0},
44 {"100", true, true, true, false, 100, 100, 100, 0},
45 {"1e9", true, true, true, false, 1e9, 1e9, 1e9, 0},
46 {"-1e9", true, false, true, false, -1e9, 0, -1e9, 0},
47 {"-1.2", false, false, true, false, 0, 0, -1.2, 0},
48 {"1e19", false, true, true, false, 0, 1e19, 1e19, 0},
49 {"1e1_9", false, true, true, false, 0, 1e19, 1e19, 0},
50 {"1E19", false, true, true, false, 0, 1e19, 1e19, 0},
51 {"-1e19", false, false, true, false, 0, 0, -1e19, 0},
52 {"0x_1p4", true, true, true, false, 16, 16, 16, 0},
53 {"0X_1P4", true, true, true, false, 16, 16, 16, 0},
54 {"0x_1p-4", false, false, true, false, 0, 0, 1 / 16., 0},
55 {"4i", false, false, false, true, 0, 0, 0, 4i},
56 {"-1.2+4.2i", false, false, false, true, 0, 0, 0, -1.2 + 4.2i},
57 {"073i", false, false, false, true, 0, 0, 0, 73i},
58
59 {"0i", true, true, true, true, 0, 0, 0, 0},
60 {"-1.2+0i", false, false, true, true, 0, 0, -1.2, -1.2},
61 {"-12+0i", true, false, true, true, -12, 0, -12, -12},
62 {"13+0i", true, true, true, true, 13, 13, 13, 13},
63
64 {"0123", true, true, true, false, 0123, 0123, 0123, 0},
65 {"-0x0", true, true, true, false, 0, 0, 0, 0},
66 {"0xdeadbeef", true, true, true, false, 0xdeadbeef, 0xdeadbeef, 0xdeadbeef, 0},
67
68 {`'a'`, true, true, true, false, 'a', 'a', 'a', 0},
69 {`'\n'`, true, true, true, false, '\n', '\n', '\n', 0},
70 {`'\\'`, true, true, true, false, '\\', '\\', '\\', 0},
71 {`'\''`, true, true, true, false, '\'', '\'', '\'', 0},
72 {`'\xFF'`, true, true, true, false, 0xFF, 0xFF, 0xFF, 0},
73 {`'パ'`, true, true, true, false, 0x30d1, 0x30d1, 0x30d1, 0},
74 {`'\u30d1'`, true, true, true, false, 0x30d1, 0x30d1, 0x30d1, 0},
75 {`'\U000030d1'`, true, true, true, false, 0x30d1, 0x30d1, 0x30d1, 0},
76
77 {text: "+-2"},
78 {text: "0x123."},
79 {text: "1e."},
80 {text: "0xi."},
81 {text: "1+2."},
82 {text: "'x"},
83 {text: "'xx'"},
84 {text: "'433937734937734969526500969526500'"},
85
86 {"0xef", true, true, true, false, 0xef, 0xef, 0xef, 0},
87 }
88
89 func TestNumberParse(t *testing.T) {
90 for _, test := range numberTests {
91
92
93 var c complex128
94 typ := itemNumber
95 var tree *Tree
96 if test.text[0] == '\'' {
97 typ = itemCharConstant
98 } else {
99 _, err := fmt.Sscan(test.text, &c)
100 if err == nil {
101 typ = itemComplex
102 }
103 }
104 n, err := tree.newNumber(0, test.text, typ)
105 ok := test.isInt || test.isUint || test.isFloat || test.isComplex
106 if ok && err != nil {
107 t.Errorf("unexpected error for %q: %s", test.text, err)
108 continue
109 }
110 if !ok && err == nil {
111 t.Errorf("expected error for %q", test.text)
112 continue
113 }
114 if !ok {
115 if *debug {
116 fmt.Printf("%s\n\t%s\n", test.text, err)
117 }
118 continue
119 }
120 if n.IsComplex != test.isComplex {
121 t.Errorf("complex incorrect for %q; should be %t", test.text, test.isComplex)
122 }
123 if test.isInt {
124 if !n.IsInt {
125 t.Errorf("expected integer for %q", test.text)
126 }
127 if n.Int64 != test.int64 {
128 t.Errorf("int64 for %q should be %d Is %d", test.text, test.int64, n.Int64)
129 }
130 } else if n.IsInt {
131 t.Errorf("did not expect integer for %q", test.text)
132 }
133 if test.isUint {
134 if !n.IsUint {
135 t.Errorf("expected unsigned integer for %q", test.text)
136 }
137 if n.Uint64 != test.uint64 {
138 t.Errorf("uint64 for %q should be %d Is %d", test.text, test.uint64, n.Uint64)
139 }
140 } else if n.IsUint {
141 t.Errorf("did not expect unsigned integer for %q", test.text)
142 }
143 if test.isFloat {
144 if !n.IsFloat {
145 t.Errorf("expected float for %q", test.text)
146 }
147 if n.Float64 != test.float64 {
148 t.Errorf("float64 for %q should be %g Is %g", test.text, test.float64, n.Float64)
149 }
150 } else if n.IsFloat {
151 t.Errorf("did not expect float for %q", test.text)
152 }
153 if test.isComplex {
154 if !n.IsComplex {
155 t.Errorf("expected complex for %q", test.text)
156 }
157 if n.Complex128 != test.complex128 {
158 t.Errorf("complex128 for %q should be %g Is %g", test.text, test.complex128, n.Complex128)
159 }
160 } else if n.IsComplex {
161 t.Errorf("did not expect complex for %q", test.text)
162 }
163 }
164 }
165
166 type parseTest struct {
167 name string
168 input string
169 ok bool
170 result string
171 }
172
173 const (
174 noError = true
175 hasError = false
176 )
177
178 var parseTests = []parseTest{
179 {"empty", "", noError,
180 ``},
181 {"comment", "{{/*\n\n\n*/}}", noError,
182 ``},
183 {"spaces", " \t\n", noError,
184 `" \t\n"`},
185 {"text", "some text", noError,
186 `"some text"`},
187 {"emptyAction", "{{}}", hasError,
188 `{{}}`},
189 {"field", "{{.X}}", noError,
190 `{{.X}}`},
191 {"simple command", "{{printf}}", noError,
192 `{{printf}}`},
193 {"$ invocation", "{{$}}", noError,
194 "{{$}}"},
195 {"variable invocation", "{{with $x := 3}}{{$x 23}}{{end}}", noError,
196 "{{with $x := 3}}{{$x 23}}{{end}}"},
197 {"variable with fields", "{{$.I}}", noError,
198 "{{$.I}}"},
199 {"multi-word command", "{{printf `%d` 23}}", noError,
200 "{{printf `%d` 23}}"},
201 {"pipeline", "{{.X|.Y}}", noError,
202 `{{.X | .Y}}`},
203 {"pipeline with decl", "{{$x := .X|.Y}}", noError,
204 `{{$x := .X | .Y}}`},
205 {"nested pipeline", "{{.X (.Y .Z) (.A | .B .C) (.E)}}", noError,
206 `{{.X (.Y .Z) (.A | .B .C) (.E)}}`},
207 {"field applied to parentheses", "{{(.Y .Z).Field}}", noError,
208 `{{(.Y .Z).Field}}`},
209 {"simple if", "{{if .X}}hello{{end}}", noError,
210 `{{if .X}}"hello"{{end}}`},
211 {"if with else", "{{if .X}}true{{else}}false{{end}}", noError,
212 `{{if .X}}"true"{{else}}"false"{{end}}`},
213 {"if with else if", "{{if .X}}true{{else if .Y}}false{{end}}", noError,
214 `{{if .X}}"true"{{else}}{{if .Y}}"false"{{end}}{{end}}`},
215 {"if else chain", "+{{if .X}}X{{else if .Y}}Y{{else if .Z}}Z{{end}}+", noError,
216 `"+"{{if .X}}"X"{{else}}{{if .Y}}"Y"{{else}}{{if .Z}}"Z"{{end}}{{end}}{{end}}"+"`},
217 {"simple range", "{{range .X}}hello{{end}}", noError,
218 `{{range .X}}"hello"{{end}}`},
219 {"chained field range", "{{range .X.Y.Z}}hello{{end}}", noError,
220 `{{range .X.Y.Z}}"hello"{{end}}`},
221 {"nested range", "{{range .X}}hello{{range .Y}}goodbye{{end}}{{end}}", noError,
222 `{{range .X}}"hello"{{range .Y}}"goodbye"{{end}}{{end}}`},
223 {"range with else", "{{range .X}}true{{else}}false{{end}}", noError,
224 `{{range .X}}"true"{{else}}"false"{{end}}`},
225 {"range over pipeline", "{{range .X|.M}}true{{else}}false{{end}}", noError,
226 `{{range .X | .M}}"true"{{else}}"false"{{end}}`},
227 {"range []int", "{{range .SI}}{{.}}{{end}}", noError,
228 `{{range .SI}}{{.}}{{end}}`},
229 {"range 1 var", "{{range $x := .SI}}{{.}}{{end}}", noError,
230 `{{range $x := .SI}}{{.}}{{end}}`},
231 {"range 2 vars", "{{range $x, $y := .SI}}{{.}}{{end}}", noError,
232 `{{range $x, $y := .SI}}{{.}}{{end}}`},
233 {"range with break", "{{range .SI}}{{.}}{{break}}{{end}}", noError,
234 `{{range .SI}}{{.}}{{break}}{{end}}`},
235 {"range with continue", "{{range .SI}}{{.}}{{continue}}{{end}}", noError,
236 `{{range .SI}}{{.}}{{continue}}{{end}}`},
237 {"constants", "{{range .SI 1 -3.2i true false 'a' nil}}{{end}}", noError,
238 `{{range .SI 1 -3.2i true false 'a' nil}}{{end}}`},
239 {"template", "{{template `x`}}", noError,
240 `{{template "x"}}`},
241 {"template with arg", "{{template `x` .Y}}", noError,
242 `{{template "x" .Y}}`},
243 {"with", "{{with .X}}hello{{end}}", noError,
244 `{{with .X}}"hello"{{end}}`},
245 {"with with else", "{{with .X}}hello{{else}}goodbye{{end}}", noError,
246 `{{with .X}}"hello"{{else}}"goodbye"{{end}}`},
247
248 {"trim left", "x \r\n\t{{- 3}}", noError, `"x"{{3}}`},
249 {"trim right", "{{3 -}}\n\n\ty", noError, `{{3}}"y"`},
250 {"trim left and right", "x \r\n\t{{- 3 -}}\n\n\ty", noError, `"x"{{3}}"y"`},
251 {"trim with extra spaces", "x\n{{- 3 -}}\ny", noError, `"x"{{3}}"y"`},
252 {"comment trim left", "x \r\n\t{{- /* hi */}}", noError, `"x"`},
253 {"comment trim right", "{{/* hi */ -}}\n\n\ty", noError, `"y"`},
254 {"comment trim left and right", "x \r\n\t{{- /* */ -}}\n\n\ty", noError, `"x""y"`},
255 {"block definition", `{{block "foo" .}}hello{{end}}`, noError,
256 `{{template "foo" .}}`},
257
258 {"newline in assignment", "{{ $x \n := \n 1 \n }}", noError, "{{$x := 1}}"},
259 {"newline in empty action", "{{\n}}", hasError, "{{\n}}"},
260 {"newline in pipeline", "{{\n\"x\"\n|\nprintf\n}}", noError, `{{"x" | printf}}`},
261 {"newline in comment", "{{/*\nhello\n*/}}", noError, ""},
262 {"newline in comment", "{{-\n/*\nhello\n*/\n-}}", noError, ""},
263 {"spaces around continue", "{{range .SI}}{{.}}{{ continue }}{{end}}", noError,
264 `{{range .SI}}{{.}}{{continue}}{{end}}`},
265 {"spaces around break", "{{range .SI}}{{.}}{{ break }}{{end}}", noError,
266 `{{range .SI}}{{.}}{{break}}{{end}}`},
267
268
269 {"unclosed action", "hello{{range", hasError, ""},
270 {"unmatched end", "{{end}}", hasError, ""},
271 {"unmatched else", "{{else}}", hasError, ""},
272 {"unmatched else after if", "{{if .X}}hello{{end}}{{else}}", hasError, ""},
273 {"multiple else", "{{if .X}}1{{else}}2{{else}}3{{end}}", hasError, ""},
274 {"missing end", "hello{{range .x}}", hasError, ""},
275 {"missing end after else", "hello{{range .x}}{{else}}", hasError, ""},
276 {"undefined function", "hello{{undefined}}", hasError, ""},
277 {"undefined variable", "{{$x}}", hasError, ""},
278 {"variable undefined after end", "{{with $x := 4}}{{end}}{{$x}}", hasError, ""},
279 {"variable undefined in template", "{{template $v}}", hasError, ""},
280 {"declare with field", "{{with $x.Y := 4}}{{end}}", hasError, ""},
281 {"template with field ref", "{{template .X}}", hasError, ""},
282 {"template with var", "{{template $v}}", hasError, ""},
283 {"invalid punctuation", "{{printf 3, 4}}", hasError, ""},
284 {"multidecl outside range", "{{with $v, $u := 3}}{{end}}", hasError, ""},
285 {"too many decls in range", "{{range $u, $v, $w := 3}}{{end}}", hasError, ""},
286 {"dot applied to parentheses", "{{printf (printf .).}}", hasError, ""},
287 {"adjacent args", "{{printf 3`x`}}", hasError, ""},
288 {"adjacent args with .", "{{printf `x`.}}", hasError, ""},
289 {"extra end after if", "{{if .X}}a{{else if .Y}}b{{end}}{{end}}", hasError, ""},
290 {"break outside range", "{{range .}}{{end}} {{break}}", hasError, ""},
291 {"continue outside range", "{{range .}}{{end}} {{continue}}", hasError, ""},
292 {"break in range else", "{{range .}}{{else}}{{break}}{{end}}", hasError, ""},
293 {"continue in range else", "{{range .}}{{else}}{{continue}}{{end}}", hasError, ""},
294
295 {"bug0a", "{{$x := 0}}{{$x}}", noError, "{{$x := 0}}{{$x}}"},
296 {"bug0b", "{{$x += 1}}{{$x}}", hasError, ""},
297 {"bug0c", "{{$x ! 2}}{{$x}}", hasError, ""},
298 {"bug0d", "{{$x % 3}}{{$x}}", hasError, ""},
299
300 {"bug0e", "{{range $x := $y := 3}}{{end}}", hasError, ""},
301
302 {"bug1a", "{{$x:=.}}{{$x!2}}", hasError, ""},
303 {"bug1b", "{{$x:=.}}{{$x+2}}", hasError, ""},
304 {"bug1c", "{{$x:=.}}{{$x +2}}", noError, "{{$x := .}}{{$x +2}}"},
305
306 {"dot after integer", "{{1.E}}", hasError, ""},
307 {"dot after float", "{{0.1.E}}", hasError, ""},
308 {"dot after boolean", "{{true.E}}", hasError, ""},
309 {"dot after char", "{{'a'.any}}", hasError, ""},
310 {"dot after string", `{{"hello".guys}}`, hasError, ""},
311 {"dot after dot", "{{..E}}", hasError, ""},
312 {"dot after nil", "{{nil.E}}", hasError, ""},
313
314 {"wrong pipeline dot", "{{12|.}}", hasError, ""},
315 {"wrong pipeline number", "{{.|12|printf}}", hasError, ""},
316 {"wrong pipeline string", "{{.|printf|\"error\"}}", hasError, ""},
317 {"wrong pipeline char", "{{12|printf|'e'}}", hasError, ""},
318 {"wrong pipeline boolean", "{{.|true}}", hasError, ""},
319 {"wrong pipeline nil", "{{'c'|nil}}", hasError, ""},
320 {"empty pipeline", `{{printf "%d" ( ) }}`, hasError, ""},
321
322 {"block definition", `{{block "foo"}}hello{{end}}`, hasError, ""},
323 }
324
325 var builtins = map[string]any{
326 "printf": fmt.Sprintf,
327 "contains": strings.Contains,
328 }
329
330 func testParse(doCopy bool, t *testing.T) {
331 textFormat = "%q"
332 defer func() { textFormat = "%s" }()
333 for _, test := range parseTests {
334 tmpl, err := New(test.name).Parse(test.input, "", "", make(map[string]*Tree), builtins)
335 switch {
336 case err == nil && !test.ok:
337 t.Errorf("%q: expected error; got none", test.name)
338 continue
339 case err != nil && test.ok:
340 t.Errorf("%q: unexpected error: %v", test.name, err)
341 continue
342 case err != nil && !test.ok:
343
344 if *debug {
345 fmt.Printf("%s: %s\n\t%s\n", test.name, test.input, err)
346 }
347 continue
348 }
349 var result string
350 if doCopy {
351 result = tmpl.Root.Copy().String()
352 } else {
353 result = tmpl.Root.String()
354 }
355 if result != test.result {
356 t.Errorf("%s=(%q): got\n\t%v\nexpected\n\t%v", test.name, test.input, result, test.result)
357 }
358 }
359 }
360
361 func TestParse(t *testing.T) {
362 testParse(false, t)
363 }
364
365
366 func TestParseCopy(t *testing.T) {
367 testParse(true, t)
368 }
369
370 func TestParseWithComments(t *testing.T) {
371 textFormat = "%q"
372 defer func() { textFormat = "%s" }()
373 tests := [...]parseTest{
374 {"comment", "{{/*\n\n\n*/}}", noError, "{{/*\n\n\n*/}}"},
375 {"comment trim left", "x \r\n\t{{- /* hi */}}", noError, `"x"{{/* hi */}}`},
376 {"comment trim right", "{{/* hi */ -}}\n\n\ty", noError, `{{/* hi */}}"y"`},
377 {"comment trim left and right", "x \r\n\t{{- /* */ -}}\n\n\ty", noError, `"x"{{/* */}}"y"`},
378 }
379 for _, test := range tests {
380 t.Run(test.name, func(t *testing.T) {
381 tr := New(test.name)
382 tr.Mode = ParseComments
383 tmpl, err := tr.Parse(test.input, "", "", make(map[string]*Tree))
384 if err != nil {
385 t.Errorf("%q: expected error; got none", test.name)
386 }
387 if result := tmpl.Root.String(); result != test.result {
388 t.Errorf("%s=(%q): got\n\t%v\nexpected\n\t%v", test.name, test.input, result, test.result)
389 }
390 })
391 }
392 }
393
394 func TestKeywordsAndFuncs(t *testing.T) {
395
396
397
398 textFormat = "%q"
399 defer func() { textFormat = "%s" }()
400
401 inp := `{{range .X}}{{break 20}}{{end}}`
402 {
403
404
405 var funcsWithKeywordFunc = map[string]any{
406 "break": func(in any) any { return in },
407 }
408 tmpl, err := New("").Parse(inp, "", "", make(map[string]*Tree), funcsWithKeywordFunc)
409 if err != nil || tmpl == nil {
410 t.Errorf("with break func: unexpected error: %v", err)
411 }
412 }
413
414 {
415
416
417 tmpl, err := New("").Parse(inp, "", "", make(map[string]*Tree), make(map[string]any))
418 if err == nil || tmpl != nil {
419 t.Errorf("without break func: expected error; got none")
420 }
421 }
422 }
423
424 func TestSkipFuncCheck(t *testing.T) {
425 oldTextFormat := textFormat
426 textFormat = "%q"
427 defer func() { textFormat = oldTextFormat }()
428 tr := New("skip func check")
429 tr.Mode = SkipFuncCheck
430 tmpl, err := tr.Parse("{{fn 1 2}}", "", "", make(map[string]*Tree))
431 if err != nil {
432 t.Fatalf("unexpected error: %v", err)
433 }
434 expected := "{{fn 1 2}}"
435 if result := tmpl.Root.String(); result != expected {
436 t.Errorf("got\n\t%v\nexpected\n\t%v", result, expected)
437 }
438 }
439
440 type isEmptyTest struct {
441 name string
442 input string
443 empty bool
444 }
445
446 var isEmptyTests = []isEmptyTest{
447 {"empty", ``, true},
448 {"nonempty", `hello`, false},
449 {"spaces only", " \t\n \t\n", true},
450 {"comment only", "{{/* comment */}}", true},
451 {"definition", `{{define "x"}}something{{end}}`, true},
452 {"definitions and space", "{{define `x`}}something{{end}}\n\n{{define `y`}}something{{end}}\n\n", true},
453 {"definitions and text", "{{define `x`}}something{{end}}\nx\n{{define `y`}}something{{end}}\ny\n", false},
454 {"definition and action", "{{define `x`}}something{{end}}{{if 3}}foo{{end}}", false},
455 }
456
457 func TestIsEmpty(t *testing.T) {
458 if !IsEmptyTree(nil) {
459 t.Errorf("nil tree is not empty")
460 }
461 for _, test := range isEmptyTests {
462 tree, err := New("root").Parse(test.input, "", "", make(map[string]*Tree), nil)
463 if err != nil {
464 t.Errorf("%q: unexpected error: %v", test.name, err)
465 continue
466 }
467 if empty := IsEmptyTree(tree.Root); empty != test.empty {
468 t.Errorf("%q: expected %t got %t", test.name, test.empty, empty)
469 }
470 }
471 }
472
473 func TestErrorContextWithTreeCopy(t *testing.T) {
474 tree, err := New("root").Parse("{{if true}}{{end}}", "", "", make(map[string]*Tree), nil)
475 if err != nil {
476 t.Fatalf("unexpected tree parse failure: %v", err)
477 }
478 treeCopy := tree.Copy()
479 wantLocation, wantContext := tree.ErrorContext(tree.Root.Nodes[0])
480 gotLocation, gotContext := treeCopy.ErrorContext(treeCopy.Root.Nodes[0])
481 if wantLocation != gotLocation {
482 t.Errorf("wrong error location want %q got %q", wantLocation, gotLocation)
483 }
484 if wantContext != gotContext {
485 t.Errorf("wrong error location want %q got %q", wantContext, gotContext)
486 }
487 }
488
489
490 var errorTests = []parseTest{
491
492 {"unclosed1",
493 "line1\n{{",
494 hasError, `unclosed1:2: unclosed action`},
495 {"unclosed2",
496 "line1\n{{define `x`}}line2\n{{",
497 hasError, `unclosed2:3: unclosed action`},
498 {"unclosed3",
499 "line1\n{{\"x\"\n\"y\"\n",
500 hasError, `unclosed3:4: unclosed action started at unclosed3:2`},
501 {"unclosed4",
502 "{{\n\n\n\n\n",
503 hasError, `unclosed4:6: unclosed action started at unclosed4:1`},
504 {"var1",
505 "line1\n{{\nx\n}}",
506 hasError, `var1:3: function "x" not defined`},
507
508 {"function",
509 "{{foo}}",
510 hasError, `function "foo" not defined`},
511 {"comment1",
512 "{{/*}}",
513 hasError, `comment1:1: unclosed comment`},
514 {"comment2",
515 "{{/*\nhello\n}}",
516 hasError, `comment2:1: unclosed comment`},
517 {"lparen",
518 "{{.X (1 2 3}}",
519 hasError, `unclosed left paren`},
520 {"rparen",
521 "{{.X 1 2 3 ) }}",
522 hasError, "unexpected right paren"},
523 {"rparen2",
524 "{{(.X 1 2 3",
525 hasError, `unclosed action`},
526 {"space",
527 "{{`x`3}}",
528 hasError, `in operand`},
529 {"idchar",
530 "{{a#}}",
531 hasError, `'#'`},
532 {"charconst",
533 "{{'a}}",
534 hasError, `unterminated character constant`},
535 {"stringconst",
536 `{{"a}}`,
537 hasError, `unterminated quoted string`},
538 {"rawstringconst",
539 "{{`a}}",
540 hasError, `unterminated raw quoted string`},
541 {"number",
542 "{{0xi}}",
543 hasError, `number syntax`},
544 {"multidefine",
545 "{{define `a`}}a{{end}}{{define `a`}}b{{end}}",
546 hasError, `multiple definition of template`},
547 {"eof",
548 "{{range .X}}",
549 hasError, `unexpected EOF`},
550 {"variable",
551
552 "{{$x := 23}}{{with $x.y := 3}}{{$x 23}}{{end}}",
553 hasError, `unexpected ":="`},
554 {"multidecl",
555 "{{$a,$b,$c := 23}}",
556 hasError, `too many declarations`},
557 {"undefvar",
558 "{{$a}}",
559 hasError, `undefined variable`},
560 {"wrongdot",
561 "{{true.any}}",
562 hasError, `unexpected . after term`},
563 {"wrongpipeline",
564 "{{12|false}}",
565 hasError, `non executable command in pipeline`},
566 {"emptypipeline",
567 `{{ ( ) }}`,
568 hasError, `missing value for parenthesized pipeline`},
569 {"multilinerawstring",
570 "{{ $v := `\n` }} {{",
571 hasError, `multilinerawstring:2: unclosed action`},
572 {"rangeundefvar",
573 "{{range $k}}{{end}}",
574 hasError, `undefined variable`},
575 {"rangeundefvars",
576 "{{range $k, $v}}{{end}}",
577 hasError, `undefined variable`},
578 {"rangemissingvalue1",
579 "{{range $k,}}{{end}}",
580 hasError, `missing value for range`},
581 {"rangemissingvalue2",
582 "{{range $k, $v := }}{{end}}",
583 hasError, `missing value for range`},
584 {"rangenotvariable1",
585 "{{range $k, .}}{{end}}",
586 hasError, `range can only initialize variables`},
587 {"rangenotvariable2",
588 "{{range $k, 123 := .}}{{end}}",
589 hasError, `range can only initialize variables`},
590 }
591
592 func TestErrors(t *testing.T) {
593 for _, test := range errorTests {
594 t.Run(test.name, func(t *testing.T) {
595 _, err := New(test.name).Parse(test.input, "", "", make(map[string]*Tree))
596 if err == nil {
597 t.Fatalf("expected error %q, got nil", test.result)
598 }
599 if !strings.Contains(err.Error(), test.result) {
600 t.Fatalf("error %q does not contain %q", err, test.result)
601 }
602 })
603 }
604 }
605
606 func TestBlock(t *testing.T) {
607 const (
608 input = `a{{block "inner" .}}bar{{.}}baz{{end}}b`
609 outer = `a{{template "inner" .}}b`
610 inner = `bar{{.}}baz`
611 )
612 treeSet := make(map[string]*Tree)
613 tmpl, err := New("outer").Parse(input, "", "", treeSet, nil)
614 if err != nil {
615 t.Fatal(err)
616 }
617 if g, w := tmpl.Root.String(), outer; g != w {
618 t.Errorf("outer template = %q, want %q", g, w)
619 }
620 inTmpl := treeSet["inner"]
621 if inTmpl == nil {
622 t.Fatal("block did not define template")
623 }
624 if g, w := inTmpl.Root.String(), inner; g != w {
625 t.Errorf("inner template = %q, want %q", g, w)
626 }
627 }
628
629 func TestLineNum(t *testing.T) {
630
631 const count = 3
632 text := strings.Repeat("{{printf 1234}}\n", count)
633 tree, err := New("bench").Parse(text, "", "", make(map[string]*Tree), builtins)
634 if err != nil {
635 t.Fatal(err)
636 }
637
638
639 nodes := tree.Root.Nodes
640 for i := 0; i < len(nodes); i += 2 {
641 line := 1 + i/2
642
643 action := nodes[i].(*ActionNode)
644 if action.Line != line {
645 t.Errorf("line %d: action is line %d", line, action.Line)
646 }
647 pipe := action.Pipe
648 if pipe.Line != line {
649 t.Errorf("line %d: pipe is line %d", line, pipe.Line)
650 }
651 }
652 }
653
654 func BenchmarkParseLarge(b *testing.B) {
655 text := strings.Repeat("{{1234}}\n", 10000)
656 for i := 0; i < b.N; i++ {
657 _, err := New("bench").Parse(text, "", "", make(map[string]*Tree), builtins)
658 if err != nil {
659 b.Fatal(err)
660 }
661 }
662 }
663
664 var sinkv, sinkl string
665
666 func BenchmarkVariableString(b *testing.B) {
667 v := &VariableNode{
668 Ident: []string{"$", "A", "BB", "CCC", "THIS_IS_THE_VARIABLE_BEING_PROCESSED"},
669 }
670 b.ResetTimer()
671 b.ReportAllocs()
672 for i := 0; i < b.N; i++ {
673 sinkv = v.String()
674 }
675 if sinkv == "" {
676 b.Fatal("Benchmark was not run")
677 }
678 }
679
680 func BenchmarkListString(b *testing.B) {
681 text := `
682 {{(printf .Field1.Field2.Field3).Value}}
683 {{$x := (printf .Field1.Field2.Field3).Value}}
684 {{$y := (printf $x.Field1.Field2.Field3).Value}}
685 {{$z := $y.Field1.Field2.Field3}}
686 {{if contains $y $z}}
687 {{printf "%q" $y}}
688 {{else}}
689 {{printf "%q" $x}}
690 {{end}}
691 {{with $z.Field1 | contains "boring"}}
692 {{printf "%q" . | printf "%s"}}
693 {{else}}
694 {{printf "%d %d %d" 11 11 11}}
695 {{printf "%d %d %s" 22 22 $x.Field1.Field2.Field3 | printf "%s"}}
696 {{printf "%v" (contains $z.Field1.Field2 $y)}}
697 {{end}}
698 `
699 tree, err := New("bench").Parse(text, "", "", make(map[string]*Tree), builtins)
700 if err != nil {
701 b.Fatal(err)
702 }
703 b.ResetTimer()
704 b.ReportAllocs()
705 for i := 0; i < b.N; i++ {
706 sinkl = tree.Root.String()
707 }
708 if sinkl == "" {
709 b.Fatal("Benchmark was not run")
710 }
711 }
712
View as plain text