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 {"with with else with", "{{with .X}}hello{{else with .Y}}goodbye{{end}}", noError,
248 `{{with .X}}"hello"{{else}}{{with .Y}}"goodbye"{{end}}{{end}}`},
249 {"with else chain", "{{with .X}}X{{else with .Y}}Y{{else with .Z}}Z{{end}}", noError,
250 `{{with .X}}"X"{{else}}{{with .Y}}"Y"{{else}}{{with .Z}}"Z"{{end}}{{end}}{{end}}`},
251
252 {"trim left", "x \r\n\t{{- 3}}", noError, `"x"{{3}}`},
253 {"trim right", "{{3 -}}\n\n\ty", noError, `{{3}}"y"`},
254 {"trim left and right", "x \r\n\t{{- 3 -}}\n\n\ty", noError, `"x"{{3}}"y"`},
255 {"trim with extra spaces", "x\n{{- 3 -}}\ny", noError, `"x"{{3}}"y"`},
256 {"comment trim left", "x \r\n\t{{- /* hi */}}", noError, `"x"`},
257 {"comment trim right", "{{/* hi */ -}}\n\n\ty", noError, `"y"`},
258 {"comment trim left and right", "x \r\n\t{{- /* */ -}}\n\n\ty", noError, `"x""y"`},
259 {"block definition", `{{block "foo" .}}hello{{end}}`, noError,
260 `{{template "foo" .}}`},
261
262 {"newline in assignment", "{{ $x \n := \n 1 \n }}", noError, "{{$x := 1}}"},
263 {"newline in empty action", "{{\n}}", hasError, "{{\n}}"},
264 {"newline in pipeline", "{{\n\"x\"\n|\nprintf\n}}", noError, `{{"x" | printf}}`},
265 {"newline in comment", "{{/*\nhello\n*/}}", noError, ""},
266 {"newline in comment", "{{-\n/*\nhello\n*/\n-}}", noError, ""},
267 {"spaces around continue", "{{range .SI}}{{.}}{{ continue }}{{end}}", noError,
268 `{{range .SI}}{{.}}{{continue}}{{end}}`},
269 {"spaces around break", "{{range .SI}}{{.}}{{ break }}{{end}}", noError,
270 `{{range .SI}}{{.}}{{break}}{{end}}`},
271
272
273 {"unclosed action", "hello{{range", hasError, ""},
274 {"unmatched end", "{{end}}", hasError, ""},
275 {"unmatched else", "{{else}}", hasError, ""},
276 {"unmatched else after if", "{{if .X}}hello{{end}}{{else}}", hasError, ""},
277 {"multiple else", "{{if .X}}1{{else}}2{{else}}3{{end}}", hasError, ""},
278 {"missing end", "hello{{range .x}}", hasError, ""},
279 {"missing end after else", "hello{{range .x}}{{else}}", hasError, ""},
280 {"undefined function", "hello{{undefined}}", hasError, ""},
281 {"undefined variable", "{{$x}}", hasError, ""},
282 {"variable undefined after end", "{{with $x := 4}}{{end}}{{$x}}", hasError, ""},
283 {"variable undefined in template", "{{template $v}}", hasError, ""},
284 {"declare with field", "{{with $x.Y := 4}}{{end}}", hasError, ""},
285 {"template with field ref", "{{template .X}}", hasError, ""},
286 {"template with var", "{{template $v}}", hasError, ""},
287 {"invalid punctuation", "{{printf 3, 4}}", hasError, ""},
288 {"multidecl outside range", "{{with $v, $u := 3}}{{end}}", hasError, ""},
289 {"too many decls in range", "{{range $u, $v, $w := 3}}{{end}}", hasError, ""},
290 {"dot applied to parentheses", "{{printf (printf .).}}", hasError, ""},
291 {"adjacent args", "{{printf 3`x`}}", hasError, ""},
292 {"adjacent args with .", "{{printf `x`.}}", hasError, ""},
293 {"extra end after if", "{{if .X}}a{{else if .Y}}b{{end}}{{end}}", hasError, ""},
294 {"break outside range", "{{range .}}{{end}} {{break}}", hasError, ""},
295 {"continue outside range", "{{range .}}{{end}} {{continue}}", hasError, ""},
296 {"break in range else", "{{range .}}{{else}}{{break}}{{end}}", hasError, ""},
297 {"continue in range else", "{{range .}}{{else}}{{continue}}{{end}}", hasError, ""},
298
299 {"bug0a", "{{$x := 0}}{{$x}}", noError, "{{$x := 0}}{{$x}}"},
300 {"bug0b", "{{$x += 1}}{{$x}}", hasError, ""},
301 {"bug0c", "{{$x ! 2}}{{$x}}", hasError, ""},
302 {"bug0d", "{{$x % 3}}{{$x}}", hasError, ""},
303
304 {"bug0e", "{{range $x := $y := 3}}{{end}}", hasError, ""},
305
306 {"bug1a", "{{$x:=.}}{{$x!2}}", hasError, ""},
307 {"bug1b", "{{$x:=.}}{{$x+2}}", hasError, ""},
308 {"bug1c", "{{$x:=.}}{{$x +2}}", noError, "{{$x := .}}{{$x +2}}"},
309
310 {"dot after integer", "{{1.E}}", hasError, ""},
311 {"dot after float", "{{0.1.E}}", hasError, ""},
312 {"dot after boolean", "{{true.E}}", hasError, ""},
313 {"dot after char", "{{'a'.any}}", hasError, ""},
314 {"dot after string", `{{"hello".guys}}`, hasError, ""},
315 {"dot after dot", "{{..E}}", hasError, ""},
316 {"dot after nil", "{{nil.E}}", hasError, ""},
317
318 {"wrong pipeline dot", "{{12|.}}", hasError, ""},
319 {"wrong pipeline number", "{{.|12|printf}}", hasError, ""},
320 {"wrong pipeline string", "{{.|printf|\"error\"}}", hasError, ""},
321 {"wrong pipeline char", "{{12|printf|'e'}}", hasError, ""},
322 {"wrong pipeline boolean", "{{.|true}}", hasError, ""},
323 {"wrong pipeline nil", "{{'c'|nil}}", hasError, ""},
324 {"empty pipeline", `{{printf "%d" ( ) }}`, hasError, ""},
325
326 {"block definition", `{{block "foo"}}hello{{end}}`, hasError, ""},
327 }
328
329 var builtins = map[string]any{
330 "printf": fmt.Sprintf,
331 "contains": strings.Contains,
332 }
333
334 func testParse(doCopy bool, t *testing.T) {
335 textFormat = "%q"
336 defer func() { textFormat = "%s" }()
337 for _, test := range parseTests {
338 tmpl, err := New(test.name).Parse(test.input, "", "", make(map[string]*Tree), builtins)
339 switch {
340 case err == nil && !test.ok:
341 t.Errorf("%q: expected error; got none", test.name)
342 continue
343 case err != nil && test.ok:
344 t.Errorf("%q: unexpected error: %v", test.name, err)
345 continue
346 case err != nil && !test.ok:
347
348 if *debug {
349 fmt.Printf("%s: %s\n\t%s\n", test.name, test.input, err)
350 }
351 continue
352 }
353 var result string
354 if doCopy {
355 result = tmpl.Root.Copy().String()
356 } else {
357 result = tmpl.Root.String()
358 }
359 if result != test.result {
360 t.Errorf("%s=(%q): got\n\t%v\nexpected\n\t%v", test.name, test.input, result, test.result)
361 }
362 }
363 }
364
365 func TestParse(t *testing.T) {
366 testParse(false, t)
367 }
368
369
370 func TestParseCopy(t *testing.T) {
371 testParse(true, t)
372 }
373
374 func TestParseWithComments(t *testing.T) {
375 textFormat = "%q"
376 defer func() { textFormat = "%s" }()
377 tests := [...]parseTest{
378 {"comment", "{{/*\n\n\n*/}}", noError, "{{/*\n\n\n*/}}"},
379 {"comment trim left", "x \r\n\t{{- /* hi */}}", noError, `"x"{{/* hi */}}`},
380 {"comment trim right", "{{/* hi */ -}}\n\n\ty", noError, `{{/* hi */}}"y"`},
381 {"comment trim left and right", "x \r\n\t{{- /* */ -}}\n\n\ty", noError, `"x"{{/* */}}"y"`},
382 }
383 for _, test := range tests {
384 t.Run(test.name, func(t *testing.T) {
385 tr := New(test.name)
386 tr.Mode = ParseComments
387 tmpl, err := tr.Parse(test.input, "", "", make(map[string]*Tree))
388 if err != nil {
389 t.Errorf("%q: expected error; got none", test.name)
390 }
391 if result := tmpl.Root.String(); result != test.result {
392 t.Errorf("%s=(%q): got\n\t%v\nexpected\n\t%v", test.name, test.input, result, test.result)
393 }
394 })
395 }
396 }
397
398 func TestKeywordsAndFuncs(t *testing.T) {
399
400
401
402 textFormat = "%q"
403 defer func() { textFormat = "%s" }()
404
405 inp := `{{range .X}}{{break 20}}{{end}}`
406 {
407
408
409 var funcsWithKeywordFunc = map[string]any{
410 "break": func(in any) any { return in },
411 }
412 tmpl, err := New("").Parse(inp, "", "", make(map[string]*Tree), funcsWithKeywordFunc)
413 if err != nil || tmpl == nil {
414 t.Errorf("with break func: unexpected error: %v", err)
415 }
416 }
417
418 {
419
420
421 tmpl, err := New("").Parse(inp, "", "", make(map[string]*Tree), make(map[string]any))
422 if err == nil || tmpl != nil {
423 t.Errorf("without break func: expected error; got none")
424 }
425 }
426 }
427
428 func TestSkipFuncCheck(t *testing.T) {
429 oldTextFormat := textFormat
430 textFormat = "%q"
431 defer func() { textFormat = oldTextFormat }()
432 tr := New("skip func check")
433 tr.Mode = SkipFuncCheck
434 tmpl, err := tr.Parse("{{fn 1 2}}", "", "", make(map[string]*Tree))
435 if err != nil {
436 t.Fatalf("unexpected error: %v", err)
437 }
438 expected := "{{fn 1 2}}"
439 if result := tmpl.Root.String(); result != expected {
440 t.Errorf("got\n\t%v\nexpected\n\t%v", result, expected)
441 }
442 }
443
444 type isEmptyTest struct {
445 name string
446 input string
447 empty bool
448 }
449
450 var isEmptyTests = []isEmptyTest{
451 {"empty", ``, true},
452 {"nonempty", `hello`, false},
453 {"spaces only", " \t\n \t\n", true},
454 {"comment only", "{{/* comment */}}", true},
455 {"definition", `{{define "x"}}something{{end}}`, true},
456 {"definitions and space", "{{define `x`}}something{{end}}\n\n{{define `y`}}something{{end}}\n\n", true},
457 {"definitions and text", "{{define `x`}}something{{end}}\nx\n{{define `y`}}something{{end}}\ny\n", false},
458 {"definition and action", "{{define `x`}}something{{end}}{{if 3}}foo{{end}}", false},
459 }
460
461 func TestIsEmpty(t *testing.T) {
462 if !IsEmptyTree(nil) {
463 t.Errorf("nil tree is not empty")
464 }
465 for _, test := range isEmptyTests {
466 tree, err := New("root").Parse(test.input, "", "", make(map[string]*Tree), nil)
467 if err != nil {
468 t.Errorf("%q: unexpected error: %v", test.name, err)
469 continue
470 }
471 if empty := IsEmptyTree(tree.Root); empty != test.empty {
472 t.Errorf("%q: expected %t got %t", test.name, test.empty, empty)
473 }
474 }
475 }
476
477 func TestErrorContextWithTreeCopy(t *testing.T) {
478 tree, err := New("root").Parse("{{if true}}{{end}}", "", "", make(map[string]*Tree), nil)
479 if err != nil {
480 t.Fatalf("unexpected tree parse failure: %v", err)
481 }
482 treeCopy := tree.Copy()
483 wantLocation, wantContext := tree.ErrorContext(tree.Root.Nodes[0])
484 gotLocation, gotContext := treeCopy.ErrorContext(treeCopy.Root.Nodes[0])
485 if wantLocation != gotLocation {
486 t.Errorf("wrong error location want %q got %q", wantLocation, gotLocation)
487 }
488 if wantContext != gotContext {
489 t.Errorf("wrong error location want %q got %q", wantContext, gotContext)
490 }
491 }
492
493
494 var errorTests = []parseTest{
495
496 {"unclosed1",
497 "line1\n{{",
498 hasError, `unclosed1:2: unclosed action`},
499 {"unclosed2",
500 "line1\n{{define `x`}}line2\n{{",
501 hasError, `unclosed2:3: unclosed action`},
502 {"unclosed3",
503 "line1\n{{\"x\"\n\"y\"\n",
504 hasError, `unclosed3:4: unclosed action started at unclosed3:2`},
505 {"unclosed4",
506 "{{\n\n\n\n\n",
507 hasError, `unclosed4:6: unclosed action started at unclosed4:1`},
508 {"var1",
509 "line1\n{{\nx\n}}",
510 hasError, `var1:3: function "x" not defined`},
511
512 {"function",
513 "{{foo}}",
514 hasError, `function "foo" not defined`},
515 {"comment1",
516 "{{/*}}",
517 hasError, `comment1:1: unclosed comment`},
518 {"comment2",
519 "{{/*\nhello\n}}",
520 hasError, `comment2:1: unclosed comment`},
521 {"lparen",
522 "{{.X (1 2 3}}",
523 hasError, `unclosed left paren`},
524 {"rparen",
525 "{{.X 1 2 3 ) }}",
526 hasError, "unexpected right paren"},
527 {"rparen2",
528 "{{(.X 1 2 3",
529 hasError, `unclosed action`},
530 {"space",
531 "{{`x`3}}",
532 hasError, `in operand`},
533 {"idchar",
534 "{{a#}}",
535 hasError, `'#'`},
536 {"charconst",
537 "{{'a}}",
538 hasError, `unterminated character constant`},
539 {"stringconst",
540 `{{"a}}`,
541 hasError, `unterminated quoted string`},
542 {"rawstringconst",
543 "{{`a}}",
544 hasError, `unterminated raw quoted string`},
545 {"number",
546 "{{0xi}}",
547 hasError, `number syntax`},
548 {"multidefine",
549 "{{define `a`}}a{{end}}{{define `a`}}b{{end}}",
550 hasError, `multiple definition of template`},
551 {"eof",
552 "{{range .X}}",
553 hasError, `unexpected EOF`},
554 {"variable",
555
556 "{{$x := 23}}{{with $x.y := 3}}{{$x 23}}{{end}}",
557 hasError, `unexpected ":="`},
558 {"multidecl",
559 "{{$a,$b,$c := 23}}",
560 hasError, `too many declarations`},
561 {"undefvar",
562 "{{$a}}",
563 hasError, `undefined variable`},
564 {"wrongdot",
565 "{{true.any}}",
566 hasError, `unexpected . after term`},
567 {"wrongpipeline",
568 "{{12|false}}",
569 hasError, `non executable command in pipeline`},
570 {"emptypipeline",
571 `{{ ( ) }}`,
572 hasError, `missing value for parenthesized pipeline`},
573 {"multilinerawstring",
574 "{{ $v := `\n` }} {{",
575 hasError, `multilinerawstring:2: unclosed action`},
576 {"rangeundefvar",
577 "{{range $k}}{{end}}",
578 hasError, `undefined variable`},
579 {"rangeundefvars",
580 "{{range $k, $v}}{{end}}",
581 hasError, `undefined variable`},
582 {"rangemissingvalue1",
583 "{{range $k,}}{{end}}",
584 hasError, `missing value for range`},
585 {"rangemissingvalue2",
586 "{{range $k, $v := }}{{end}}",
587 hasError, `missing value for range`},
588 {"rangenotvariable1",
589 "{{range $k, .}}{{end}}",
590 hasError, `range can only initialize variables`},
591 {"rangenotvariable2",
592 "{{range $k, 123 := .}}{{end}}",
593 hasError, `range can only initialize variables`},
594 }
595
596 func TestErrors(t *testing.T) {
597 for _, test := range errorTests {
598 t.Run(test.name, func(t *testing.T) {
599 _, err := New(test.name).Parse(test.input, "", "", make(map[string]*Tree))
600 if err == nil {
601 t.Fatalf("expected error %q, got nil", test.result)
602 }
603 if !strings.Contains(err.Error(), test.result) {
604 t.Fatalf("error %q does not contain %q", err, test.result)
605 }
606 })
607 }
608 }
609
610 func TestBlock(t *testing.T) {
611 const (
612 input = `a{{block "inner" .}}bar{{.}}baz{{end}}b`
613 outer = `a{{template "inner" .}}b`
614 inner = `bar{{.}}baz`
615 )
616 treeSet := make(map[string]*Tree)
617 tmpl, err := New("outer").Parse(input, "", "", treeSet, nil)
618 if err != nil {
619 t.Fatal(err)
620 }
621 if g, w := tmpl.Root.String(), outer; g != w {
622 t.Errorf("outer template = %q, want %q", g, w)
623 }
624 inTmpl := treeSet["inner"]
625 if inTmpl == nil {
626 t.Fatal("block did not define template")
627 }
628 if g, w := inTmpl.Root.String(), inner; g != w {
629 t.Errorf("inner template = %q, want %q", g, w)
630 }
631 }
632
633 func TestLineNum(t *testing.T) {
634
635 const count = 3
636 text := strings.Repeat("{{printf 1234}}\n", count)
637 tree, err := New("bench").Parse(text, "", "", make(map[string]*Tree), builtins)
638 if err != nil {
639 t.Fatal(err)
640 }
641
642
643 nodes := tree.Root.Nodes
644 for i := 0; i < len(nodes); i += 2 {
645 line := 1 + i/2
646
647 action := nodes[i].(*ActionNode)
648 if action.Line != line {
649 t.Errorf("line %d: action is line %d", line, action.Line)
650 }
651 pipe := action.Pipe
652 if pipe.Line != line {
653 t.Errorf("line %d: pipe is line %d", line, pipe.Line)
654 }
655 }
656 }
657
658 func BenchmarkParseLarge(b *testing.B) {
659 text := strings.Repeat("{{1234}}\n", 10000)
660 for i := 0; i < b.N; i++ {
661 _, err := New("bench").Parse(text, "", "", make(map[string]*Tree), builtins)
662 if err != nil {
663 b.Fatal(err)
664 }
665 }
666 }
667
668 var sinkv, sinkl string
669
670 func BenchmarkVariableString(b *testing.B) {
671 v := &VariableNode{
672 Ident: []string{"$", "A", "BB", "CCC", "THIS_IS_THE_VARIABLE_BEING_PROCESSED"},
673 }
674 b.ResetTimer()
675 b.ReportAllocs()
676 for i := 0; i < b.N; i++ {
677 sinkv = v.String()
678 }
679 if sinkv == "" {
680 b.Fatal("Benchmark was not run")
681 }
682 }
683
684 func BenchmarkListString(b *testing.B) {
685 text := `
686 {{(printf .Field1.Field2.Field3).Value}}
687 {{$x := (printf .Field1.Field2.Field3).Value}}
688 {{$y := (printf $x.Field1.Field2.Field3).Value}}
689 {{$z := $y.Field1.Field2.Field3}}
690 {{if contains $y $z}}
691 {{printf "%q" $y}}
692 {{else}}
693 {{printf "%q" $x}}
694 {{end}}
695 {{with $z.Field1 | contains "boring"}}
696 {{printf "%q" . | printf "%s"}}
697 {{else}}
698 {{printf "%d %d %d" 11 11 11}}
699 {{printf "%d %d %s" 22 22 $x.Field1.Field2.Field3 | printf "%s"}}
700 {{printf "%v" (contains $z.Field1.Field2 $y)}}
701 {{end}}
702 `
703 tree, err := New("bench").Parse(text, "", "", make(map[string]*Tree), builtins)
704 if err != nil {
705 b.Fatal(err)
706 }
707 b.ResetTimer()
708 b.ReportAllocs()
709 for i := 0; i < b.N; i++ {
710 sinkl = tree.Root.String()
711 }
712 if sinkl == "" {
713 b.Fatal("Benchmark was not run")
714 }
715 }
716
View as plain text