github.com/euank/go@v0.0.0-20160829210321-495514729181/src/cmd/compile/internal/gc/lex_test.go (about)

     1  // Copyright 2016 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 gc
     6  
     7  import "testing"
     8  
     9  func eq(a, b []string) bool {
    10  	if len(a) != len(b) {
    11  		return false
    12  	}
    13  	for i := 0; i < len(a); i++ {
    14  		if a[i] != b[i] {
    15  			return false
    16  		}
    17  	}
    18  	return true
    19  }
    20  
    21  func TestPragmaFields(t *testing.T) {
    22  
    23  	var tests = []struct {
    24  		in   string
    25  		want []string
    26  	}{
    27  		{"", []string{}},
    28  		{" \t ", []string{}},
    29  		{`""""`, []string{`""`, `""`}},
    30  		{"  a'b'c  ", []string{"a'b'c"}},
    31  		{"1 2 3 4", []string{"1", "2", "3", "4"}},
    32  		{"\n☺\t☹\n", []string{"☺", "☹"}},
    33  		{`"1 2 "  3  " 4 5"`, []string{`"1 2 "`, `3`, `" 4 5"`}},
    34  		{`"1""2 3""4"`, []string{`"1"`, `"2 3"`, `"4"`}},
    35  		{`12"34"`, []string{`12`, `"34"`}},
    36  		{`12"34 `, []string{`12`}},
    37  	}
    38  
    39  	for _, tt := range tests {
    40  		got := pragmaFields(tt.in)
    41  		if !eq(got, tt.want) {
    42  			t.Errorf("pragmaFields(%q) = %v; want %v", tt.in, got, tt.want)
    43  			continue
    44  		}
    45  	}
    46  }
    47  
    48  func TestPragcgo(t *testing.T) {
    49  
    50  	var tests = []struct {
    51  		in   string
    52  		want string
    53  	}{
    54  		{`go:cgo_export_dynamic local`, "cgo_export_dynamic local\n"},
    55  		{`go:cgo_export_dynamic local remote`, "cgo_export_dynamic local remote\n"},
    56  		{`go:cgo_export_dynamic local' remote'`, "cgo_export_dynamic 'local''' 'remote'''\n"},
    57  		{`go:cgo_export_static local`, "cgo_export_static local\n"},
    58  		{`go:cgo_export_static local remote`, "cgo_export_static local remote\n"},
    59  		{`go:cgo_export_static local' remote'`, "cgo_export_static 'local''' 'remote'''\n"},
    60  		{`go:cgo_import_dynamic local`, "cgo_import_dynamic local\n"},
    61  		{`go:cgo_import_dynamic local remote`, "cgo_import_dynamic local remote\n"},
    62  		{`go:cgo_import_dynamic local remote "library"`, "cgo_import_dynamic local remote library\n"},
    63  		{`go:cgo_import_dynamic local' remote' "lib rary"`, "cgo_import_dynamic 'local''' 'remote''' 'lib rary'\n"},
    64  		{`go:cgo_import_static local`, "cgo_import_static local\n"},
    65  		{`go:cgo_import_static local'`, "cgo_import_static 'local'''\n"},
    66  		{`go:cgo_dynamic_linker "/path/"`, "cgo_dynamic_linker /path/\n"},
    67  		{`go:cgo_dynamic_linker "/p ath/"`, "cgo_dynamic_linker '/p ath/'\n"},
    68  		{`go:cgo_ldflag "arg"`, "cgo_ldflag arg\n"},
    69  		{`go:cgo_ldflag "a rg"`, "cgo_ldflag 'a rg'\n"},
    70  	}
    71  
    72  	for _, tt := range tests {
    73  		got := pragcgo(tt.in)
    74  		if got != tt.want {
    75  			t.Errorf("pragcgo(%q) = %q; want %q", tt.in, got, tt.want)
    76  			continue
    77  		}
    78  	}
    79  }