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
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
|
package sqlutil
import (
"reflect"
"testing"
)
func TestSplitStatements(t *testing.T) {
tests := []struct {
name string
input string
expected []string
}{
{
name: "Simple single statement",
input: "SELECT * FROM users",
expected: []string{"SELECT * FROM users"},
},
{
name: "Multiple statements",
input: "SELECT * FROM users; SELECT * FROM orders;",
expected: []string{"SELECT * FROM users", "SELECT * FROM orders"},
},
{
name: "Semicolon in single quotes",
input: "SELECT 'hello;world' FROM users; SELECT * FROM orders;",
expected: []string{"SELECT 'hello;world' FROM users", "SELECT * FROM orders"},
},
{
name: "Semicolon in double quotes",
input: `SELECT "column;name" FROM users; SELECT * FROM orders;`,
expected: []string{`SELECT "column;name" FROM users`, "SELECT * FROM orders"},
},
{
name: "Escaped quotes in strings",
input: `SELECT 'don''t split; here' FROM users; SELECT * FROM orders;`,
expected: []string{`SELECT 'don''t split; here' FROM users`, "SELECT * FROM orders"},
},
{
name: "Escaped quotes in identifiers",
input: `SELECT "column""name" FROM users; SELECT * FROM orders;`,
expected: []string{`SELECT "column""name" FROM users`, "SELECT * FROM orders"},
},
{
name: "Single line comment",
input: "SELECT * FROM users; -- This is a comment\nSELECT * FROM orders;",
expected: []string{"SELECT * FROM users", "SELECT * FROM orders"},
},
{
name: "Single line comment with semicolon",
input: "SELECT * FROM users; -- Comment with; semicolon\nSELECT * FROM orders;",
expected: []string{"SELECT * FROM users", "SELECT * FROM orders"},
},
{
name: "Multi-line comment",
input: "SELECT * FROM users; /* Multi-line\ncomment */ SELECT * FROM orders;",
expected: []string{"SELECT * FROM users", "SELECT * FROM orders"},
},
{
name: "Multi-line comment with semicolon",
input: "SELECT * FROM users; /* Comment with; semicolon */ SELECT * FROM orders;",
expected: []string{"SELECT * FROM users", "SELECT * FROM orders"},
},
{
name: "Complex mixed case",
input: `SELECT 'test;string', "quoted;id" FROM users; -- Comment; here
/* Another; comment */
INSERT INTO users VALUES ('name''s value', "id""field");`,
expected: []string{
`SELECT 'test;string', "quoted;id" FROM users`,
`INSERT INTO users VALUES ('name''s value', "id""field")`,
},
},
{
name: "Empty statements filtered",
input: "SELECT * FROM users;;; SELECT * FROM orders;",
expected: []string{"SELECT * FROM users", "SELECT * FROM orders"},
},
{
name: "Whitespace handling",
input: " SELECT * FROM users ; SELECT * FROM orders ; ",
expected: []string{"SELECT * FROM users", "SELECT * FROM orders"},
},
{
name: "Single statement without semicolon",
input: "SELECT * FROM users",
expected: []string{"SELECT * FROM users"},
},
{
name: "Empty query",
input: "",
expected: []string{},
},
{
name: "Only whitespace",
input: " \n\t ",
expected: []string{},
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
result := SplitStatements(tt.input)
if !reflect.DeepEqual(result, tt.expected) {
t.Errorf("SplitStatements() = %v, expected %v", result, tt.expected)
}
})
}
}
func TestSplitStatements_EdgeCases(t *testing.T) {
tests := []struct {
name string
input string
expected []string
}{
{
name: "Nested comments are not supported but handled gracefully",
input: "SELECT * FROM users; /* Outer /* inner */ comment */ SELECT * FROM orders;",
expected: []string{"SELECT * FROM users", "comment */ SELECT * FROM orders"},
},
{
name: "Unterminated string (malformed SQL)",
input: "SELECT 'unterminated string; SELECT * FROM orders;",
expected: []string{"SELECT 'unterminated string; SELECT * FROM orders;"},
},
{
name: "Unterminated comment (malformed SQL)",
input: "SELECT * FROM users; /* unterminated comment",
expected: []string{"SELECT * FROM users"},
},
{
name: "Multiple semicolons in quotes",
input: "SELECT ';;;' FROM users; SELECT ';;;' FROM orders;",
expected: []string{"SELECT ';;;' FROM users", "SELECT ';;;' FROM orders"},
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
result := SplitStatements(tt.input)
if !reflect.DeepEqual(result, tt.expected) {
t.Errorf("SplitStatements() = %v, expected %v", result, tt.expected)
}
})
}
}
|
