1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
// Copyright 2017 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package batch
import (
"fmt"
"testing"
)
func TestBatchSplit(t *testing.T) {
type testItem struct {
Sql string
Expect []string
}
list := []testItem{
testItem{
Sql: `use DB
go
select 1
go
select 2
`,
Expect: []string{`use DB
`, `
select 1
`, `
select 2
`,
},
},
testItem{
Sql: `go
use DB go
`,
Expect: []string{`
use DB go
`,
},
},
testItem{
Sql: `select 'It''s go time'
go
select top 1 1`,
Expect: []string{`select 'It''s go time'
`, `
select top 1 1`,
},
},
testItem{
Sql: `select 1 /* go */
go
select top 1 1`,
Expect: []string{`select 1 /* go */
`, `
select top 1 1`,
},
},
testItem{
Sql: `select 1 -- go
go
select top 1 1`,
Expect: []string{`select 1 -- go
`, `
select top 1 1`,
},
},
testItem{Sql: `"0'"`, Expect: []string{`"0'"`}},
testItem{Sql: "0'", Expect: []string{"0'"}},
testItem{Sql: "--", Expect: []string{"--"}},
testItem{Sql: "GO", Expect: nil},
testItem{Sql: "/*", Expect: []string{"/*"}},
testItem{Sql: "gO\x01\x00O550655490663051008\n", Expect: []string{"\n"}},
testItem{Sql: "select 1;\nGO 2\nselect 2;", Expect: []string{"select 1;\n", "select 1;\n", "\nselect 2;"}},
testItem{Sql: "select 'hi\\\n-hello';", Expect: []string{"select 'hi-hello';"}},
testItem{Sql: "select 'hi\\\r\n-hello';", Expect: []string{"select 'hi-hello';"}},
testItem{Sql: "select 'hi\\\r-hello';", Expect: []string{"select 'hi-hello';"}},
testItem{Sql: "select 'hi\\\n\nhello';", Expect: []string{"select 'hi\nhello';"}},
}
index := -1
for i := range list {
if index >= 0 && index != i {
continue
}
sqltext := list[i].Sql
t.Run(fmt.Sprintf("index-%d", i), func(t *testing.T) {
ss := Split(sqltext, "go")
if len(ss) != len(list[i].Expect) {
t.Errorf("Test Item index %d; expect %d items, got %d %q", i, len(list[i].Expect), len(ss), ss)
return
}
for j := 0; j < len(ss); j++ {
if ss[j] != list[i].Expect[j] {
t.Errorf("Test Item index %d, batch index %d; expect <%s>, got <%s>", i, j, list[i].Expect[j], ss[j])
}
}
})
}
}
func TestHasPrefixFold(t *testing.T) {
list := []struct {
s, pre string
is bool
}{
{"h", "H", true},
{"h", "K", false},
{"go 5\n", "go", true},
}
for _, item := range list {
is := hasPrefixFold(item.s, item.pre)
if is != item.is {
t.Error("want (%q, %q)=%t got %t", item.s, item.pre, item.is, is)
}
}
}
|
