1 // Copyright 2011 The Go Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style 3 // license that can be found in the LICENSE file. 4 5 package ast 6 7 import ( 8 "bytes" 9 "strings" 10 "testing" 11 ) 12 13 var tests = []struct { 14 x interface{} // x is printed as s 15 s string 16 }{ 17 // basic types 18 {nil, "0 nil"}, 19 {true, "0 true"}, 20 {42, "0 42"}, 21 {3.14, "0 3.14"}, 22 {1 + 2.718i, "0 (1+2.718i)"}, 23 {"foobar", "0 \"foobar\""}, 24 25 // maps 26 {map[Expr]string{}, `0 map[ast.Expr]string (len = 0) {}`}, 27 {map[string]int{"a": 1}, 28 `0 map[string]int (len = 1) { 29 1 . "a": 1 30 2 }`}, 31 32 // pointers 33 {new(int), "0 *0"}, 34 35 // arrays 36 {[0]int{}, `0 [0]int {}`}, 37 {[3]int{1, 2, 3}, 38 `0 [3]int { 39 1 . 0: 1 40 2 . 1: 2 41 3 . 2: 3 42 4 }`}, 43 {[...]int{42}, 44 `0 [1]int { 45 1 . 0: 42 46 2 }`}, 47 48 // slices 49 {[]int{}, `0 []int (len = 0) {}`}, 50 {[]int{1, 2, 3}, 51 `0 []int (len = 3) { 52 1 . 0: 1 53 2 . 1: 2 54 3 . 2: 3 55 4 }`}, 56 57 // structs 58 {struct{}{}, `0 struct {} {}`}, 59 {struct{ x int }{007}, `0 struct { x int } {}`}, 60 {struct{ X, y int }{42, 991}, 61 `0 struct { X int; y int } { 62 1 . X: 42 63 2 }`}, 64 {struct{ X, Y int }{42, 991}, 65 `0 struct { X int; Y int } { 66 1 . X: 42 67 2 . Y: 991 68 3 }`}, 69 } 70 71 // Split s into lines, trim whitespace from all lines, and return 72 // the concatenated non-empty lines. 73 func trim(s string) string { 74 lines := strings.Split(s, "\n") 75 i := 0 76 for _, line := range lines { 77 line = strings.TrimSpace(line) 78 if line != "" { 79 lines[i] = line 80 i++ 81 } 82 } 83 return strings.Join(lines[0:i], "\n") 84 } 85 86 func TestPrint(t *testing.T) { 87 var buf bytes.Buffer 88 for _, test := range tests { 89 buf.Reset() 90 if err := Fprint(&buf, nil, test.x, nil); err != nil { 91 t.Errorf("Fprint failed: %s", err) 92 } 93 if s, ts := trim(buf.String()), trim(test.s); s != ts { 94 t.Errorf("got:\n%s\nexpected:\n%s\n", s, ts) 95 } 96 } 97 } 98
