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 TestSkipFuncCheck(t *testing.T) {
395 oldTextFormat := textFormat
396 textFormat = "%q"
397 defer func() { textFormat = oldTextFormat }()
398 tr := New("skip func check")
399 tr.Mode = SkipFuncCheck
400 tmpl, err := tr.Parse("{{fn 1 2}}", "", "", make(map[string]*Tree))
401 if err != nil {
402 t.Fatalf("unexpected error: %v", err)
403 }
404 expected := "{{fn 1 2}}"
405 if result := tmpl.Root.String(); result != expected {
406 t.Errorf("got\n\t%v\nexpected\n\t%v", result, expected)
407 }
408 }
409
410 type isEmptyTest struct {
411 name string
412 input string
413 empty bool
414 }
415
416 var isEmptyTests = []isEmptyTest{
417 {"empty", ``, true},
418 {"nonempty", `hello`, false},
419 {"spaces only", " \t\n \t\n", true},
420 {"comment only", "{{/* comment */}}", true},
421 {"definition", `{{define "x"}}something{{end}}`, true},
422 {"definitions and space", "{{define `x`}}something{{end}}\n\n{{define `y`}}something{{end}}\n\n", true},
423 {"definitions and text", "{{define `x`}}something{{end}}\nx\n{{define `y`}}something{{end}}\ny\n", false},
424 {"definition and action", "{{define `x`}}something{{end}}{{if 3}}foo{{end}}", false},
425 }
426
427 func TestIsEmpty(t *testing.T) {
428 if !IsEmptyTree(nil) {
429 t.Errorf("nil tree is not empty")
430 }
431 for _, test := range isEmptyTests {
432 tree, err := New("root").Parse(test.input, "", "", make(map[string]*Tree), nil)
433 if err != nil {
434 t.Errorf("%q: unexpected error: %v", test.name, err)
435 continue
436 }
437 if empty := IsEmptyTree(tree.Root); empty != test.empty {
438 t.Errorf("%q: expected %t got %t", test.name, test.empty, empty)
439 }
440 }
441 }
442
443 func TestErrorContextWithTreeCopy(t *testing.T) {
444 tree, err := New("root").Parse("{{if true}}{{end}}", "", "", make(map[string]*Tree), nil)
445 if err != nil {
446 t.Fatalf("unexpected tree parse failure: %v", err)
447 }
448 treeCopy := tree.Copy()
449 wantLocation, wantContext := tree.ErrorContext(tree.Root.Nodes[0])
450 gotLocation, gotContext := treeCopy.ErrorContext(treeCopy.Root.Nodes[0])
451 if wantLocation != gotLocation {
452 t.Errorf("wrong error location want %q got %q", wantLocation, gotLocation)
453 }
454 if wantContext != gotContext {
455 t.Errorf("wrong error location want %q got %q", wantContext, gotContext)
456 }
457 }
458
459
460 var errorTests = []parseTest{
461
462 {"unclosed1",
463 "line1\n{{",
464 hasError, `unclosed1:2: unclosed action`},
465 {"unclosed2",
466 "line1\n{{define `x`}}line2\n{{",
467 hasError, `unclosed2:3: unclosed action`},
468 {"unclosed3",
469 "line1\n{{\"x\"\n\"y\"\n",
470 hasError, `unclosed3:4: unclosed action started at unclosed3:2`},
471 {"unclosed4",
472 "{{\n\n\n\n\n",
473 hasError, `unclosed4:6: unclosed action started at unclosed4:1`},
474 {"var1",
475 "line1\n{{\nx\n}}",
476 hasError, `var1:3: function "x" not defined`},
477
478 {"function",
479 "{{foo}}",
480 hasError, `function "foo" not defined`},
481 {"comment1",
482 "{{/*}}",
483 hasError, `comment1:1: unclosed comment`},
484 {"comment2",
485 "{{/*\nhello\n}}",
486 hasError, `comment2:1: unclosed comment`},
487 {"lparen",
488 "{{.X (1 2 3}}",
489 hasError, `unclosed left paren`},
490 {"rparen",
491 "{{.X 1 2 3 ) }}",
492 hasError, `unexpected ")" in command`},
493 {"rparen2",
494 "{{(.X 1 2 3",
495 hasError, `unclosed action`},
496 {"space",
497 "{{`x`3}}",
498 hasError, `in operand`},
499 {"idchar",
500 "{{a#}}",
501 hasError, `'#'`},
502 {"charconst",
503 "{{'a}}",
504 hasError, `unterminated character constant`},
505 {"stringconst",
506 `{{"a}}`,
507 hasError, `unterminated quoted string`},
508 {"rawstringconst",
509 "{{`a}}",
510 hasError, `unterminated raw quoted string`},
511 {"number",
512 "{{0xi}}",
513 hasError, `number syntax`},
514 {"multidefine",
515 "{{define `a`}}a{{end}}{{define `a`}}b{{end}}",
516 hasError, `multiple definition of template`},
517 {"eof",
518 "{{range .X}}",
519 hasError, `unexpected EOF`},
520 {"variable",
521
522 "{{$x := 23}}{{with $x.y := 3}}{{$x 23}}{{end}}",
523 hasError, `unexpected ":="`},
524 {"multidecl",
525 "{{$a,$b,$c := 23}}",
526 hasError, `too many declarations`},
527 {"undefvar",
528 "{{$a}}",
529 hasError, `undefined variable`},
530 {"wrongdot",
531 "{{true.any}}",
532 hasError, `unexpected . after term`},
533 {"wrongpipeline",
534 "{{12|false}}",
535 hasError, `non executable command in pipeline`},
536 {"emptypipeline",
537 `{{ ( ) }}`,
538 hasError, `missing value for parenthesized pipeline`},
539 {"multilinerawstring",
540 "{{ $v := `\n` }} {{",
541 hasError, `multilinerawstring:2: unclosed action`},
542 {"rangeundefvar",
543 "{{range $k}}{{end}}",
544 hasError, `undefined variable`},
545 {"rangeundefvars",
546 "{{range $k, $v}}{{end}}",
547 hasError, `undefined variable`},
548 {"rangemissingvalue1",
549 "{{range $k,}}{{end}}",
550 hasError, `missing value for range`},
551 {"rangemissingvalue2",
552 "{{range $k, $v := }}{{end}}",
553 hasError, `missing value for range`},
554 {"rangenotvariable1",
555 "{{range $k, .}}{{end}}",
556 hasError, `range can only initialize variables`},
557 {"rangenotvariable2",
558 "{{range $k, 123 := .}}{{end}}",
559 hasError, `range can only initialize variables`},
560 }
561
562 func TestErrors(t *testing.T) {
563 for _, test := range errorTests {
564 t.Run(test.name, func(t *testing.T) {
565 _, err := New(test.name).Parse(test.input, "", "", make(map[string]*Tree))
566 if err == nil {
567 t.Fatalf("expected error %q, got nil", test.result)
568 }
569 if !strings.Contains(err.Error(), test.result) {
570 t.Fatalf("error %q does not contain %q", err, test.result)
571 }
572 })
573 }
574 }
575
576 func TestBlock(t *testing.T) {
577 const (
578 input = `a{{block "inner" .}}bar{{.}}baz{{end}}b`
579 outer = `a{{template "inner" .}}b`
580 inner = `bar{{.}}baz`
581 )
582 treeSet := make(map[string]*Tree)
583 tmpl, err := New("outer").Parse(input, "", "", treeSet, nil)
584 if err != nil {
585 t.Fatal(err)
586 }
587 if g, w := tmpl.Root.String(), outer; g != w {
588 t.Errorf("outer template = %q, want %q", g, w)
589 }
590 inTmpl := treeSet["inner"]
591 if inTmpl == nil {
592 t.Fatal("block did not define template")
593 }
594 if g, w := inTmpl.Root.String(), inner; g != w {
595 t.Errorf("inner template = %q, want %q", g, w)
596 }
597 }
598
599 func TestLineNum(t *testing.T) {
600 const count = 100
601 text := strings.Repeat("{{printf 1234}}\n", count)
602 tree, err := New("bench").Parse(text, "", "", make(map[string]*Tree), builtins)
603 if err != nil {
604 t.Fatal(err)
605 }
606
607
608 nodes := tree.Root.Nodes
609 for i := 0; i < len(nodes); i += 2 {
610 line := 1 + i/2
611
612 action := nodes[i].(*ActionNode)
613 if action.Line != line {
614 t.Fatalf("line %d: action is line %d", line, action.Line)
615 }
616 pipe := action.Pipe
617 if pipe.Line != line {
618 t.Fatalf("line %d: pipe is line %d", line, pipe.Line)
619 }
620 }
621 }
622
623 func BenchmarkParseLarge(b *testing.B) {
624 text := strings.Repeat("{{1234}}\n", 10000)
625 for i := 0; i < b.N; i++ {
626 _, err := New("bench").Parse(text, "", "", make(map[string]*Tree), builtins)
627 if err != nil {
628 b.Fatal(err)
629 }
630 }
631 }
632
633 var sinkv, sinkl string
634
635 func BenchmarkVariableString(b *testing.B) {
636 v := &VariableNode{
637 Ident: []string{"$", "A", "BB", "CCC", "THIS_IS_THE_VARIABLE_BEING_PROCESSED"},
638 }
639 b.ResetTimer()
640 b.ReportAllocs()
641 for i := 0; i < b.N; i++ {
642 sinkv = v.String()
643 }
644 if sinkv == "" {
645 b.Fatal("Benchmark was not run")
646 }
647 }
648
649 func BenchmarkListString(b *testing.B) {
650 text := `
651 {{(printf .Field1.Field2.Field3).Value}}
652 {{$x := (printf .Field1.Field2.Field3).Value}}
653 {{$y := (printf $x.Field1.Field2.Field3).Value}}
654 {{$z := $y.Field1.Field2.Field3}}
655 {{if contains $y $z}}
656 {{printf "%q" $y}}
657 {{else}}
658 {{printf "%q" $x}}
659 {{end}}
660 {{with $z.Field1 | contains "boring"}}
661 {{printf "%q" . | printf "%s"}}
662 {{else}}
663 {{printf "%d %d %d" 11 11 11}}
664 {{printf "%d %d %s" 22 22 $x.Field1.Field2.Field3 | printf "%s"}}
665 {{printf "%v" (contains $z.Field1.Field2 $y)}}
666 {{end}}
667 `
668 tree, err := New("bench").Parse(text, "", "", make(map[string]*Tree), builtins)
669 if err != nil {
670 b.Fatal(err)
671 }
672 b.ResetTimer()
673 b.ReportAllocs()
674 for i := 0; i < b.N; i++ {
675 sinkl = tree.Root.String()
676 }
677 if sinkl == "" {
678 b.Fatal("Benchmark was not run")
679 }
680 }
681
View as plain text