forked from taozhi8833998/node-sql-parser
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathinsert.spec.js
More file actions
211 lines (190 loc) · 8.76 KB
/
insert.spec.js
File metadata and controls
211 lines (190 loc) · 8.76 KB
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
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
const { expect } = require('chai');
const Parser = require('../src/parser').default
describe('insert', () => {
const parser = new Parser();
function getParsedSql(sql, opt) {
const ast = parser.astify(sql, opt);
return parser.sqlify(ast, opt);
}
it('should parse baisc usage', () => {
const { tableList, columnList, ast } = parser.parse('INSERT INTO t (col1, col2) VALUES (1, 2)');
expect(tableList).to.eql(['insert::null::t']);
expect(columnList).to.eql(['insert::t::col1', 'insert::t::col2']);
expect(ast.type).to.be.eql('insert');
expect(ast.table).to.be.eql([ { db: null, table: 't', as: null } ]);
expect(ast.columns).to.be.eql(["col1", "col2"]);
expect(ast.values).to.eql({
type: 'values',
values: [
{
type: 'expr_list',
value: [
{
type: 'number',
value: 1
},
{
type: 'number',
value: 2
}
],
prefix: null
}
]
});
});
it('should support parse insert with multiple rows', () => {
const sql = 'INSERT INTO t1 values("1223", "name"), ("1224", "name2")'
const ast = parser.astify(sql)
const backSQL = parser.sqlify(ast)
expect(backSQL).to.be.equal('INSERT INTO `t1` VALUES ("1223","name"), ("1224","name2")')
})
it("should throw error if column count doesn't match value count", () => {
const sql = 'INSERT INTO t1 (col1, col2, col3) values("1223", "name"), ("1224", "name2")'
const fun = parser.astify.bind(parser, sql)
expect(fun).to.throw("column count doesn't match value count at row 1")
})
it('should support parse insert from select', () => {
const sql = 'INSERT INTO t1(col_a, col_b) select col_a, col_b from t2'
const ast = parser.astify(sql)
const backSQL = parser.sqlify(ast)
expect(backSQL).to.be.equal("INSERT INTO `t1` (col_a, col_b) SELECT `col_a`, `col_b` FROM `t2`")
})
it('should parse insert and select', () => {
const sql = 'INSERT INTO t1 values("1223", "name") ; SELECT * FROM t'
const [sqla, sqlb] = sql.split(';')
const astFirstSQL = parser.astify(sqla.trim())
const astSecondSQL = parser.astify(sqlb.trim())
const { tableList, columnList, ast } = parser.parse(sql)
expect(tableList).to.eql(['insert::null::t1', 'select::null::t'])
expect(columnList).to.eql(['insert::t1::(.*)', 'select::null::(.*)'])
expect(ast).to.have.lengthOf(2)
expect(ast[0]).to.eql(astFirstSQL)
expect(ast[1]).to.eql(astSecondSQL)
})
it('should parser no columns', () => {
const ast = {
"type": "insert",
"table": [
{
"db": null,
"table": "t",
"as": null
}
],
"values": {
type: 'values',
values: [
{
"type": "expr_list",
"value": [
{
"type": "number",
"value": 1
},
{
"type": "number",
"value": 2
}
]
}]
}
}
expect(parser.sqlify(ast)).to.be.eql('INSERT INTO `t` VALUES (1,2)')
ast.columns = ['col1', 'col2']
expect(parser.sqlify(ast)).to.be.eql('INSERT INTO `t` (col1, col2) VALUES (1,2)')
})
it('should support big number', () => {
const bigNumberList = ['3511161156327326047123', '23.3e+12323243434']
for (const bigNumber of bigNumberList) {
const sql = `INSERT INTO t1(id) VALUES(${bigNumber})`
const ast = parser.astify(sql)
expect(ast.values).to.be.eql({
type: 'values',
values: [
{
type: 'expr_list',
value: [
{
type: 'bigint',
value: bigNumber
}
],
prefix: null
}
]
})
expect(parser.sqlify(ast)).to.equal('INSERT INTO `t1` (id) VALUES (' + bigNumber + ')')
}
})
it('should support parse hive sql insert from select', () => {
const sql = 'INSERT overwrite table account select col_a, col_b from t2'
const ast = parser.astify(sql, { database: 'hive' })
const backSQL = parser.sqlify(ast)
expect(backSQL).to.be.equal("INSERT OVERWRITE TABLE `account` SELECT `col_a`, `col_b` FROM `t2`")
})
it('should support parse insert partition', () => {
const sql = 'INSERT into account partition(date, id) select col_a, col_b from t2'
const ast = parser.astify(sql)
const backSQL = parser.sqlify(ast)
expect(backSQL).to.be.equal("INSERT INTO `account` PARTITION(`date`, `id`) SELECT `col_a`, `col_b` FROM `t2`")
})
it('should support parse insert partition', () => {
const sql = 'INSERT into account partition(date, id) (id, name) values(123, "test"), (124, "test2")'
const ast = parser.astify(sql)
const backSQL = parser.sqlify(ast)
expect(backSQL).to.be.equal('INSERT INTO `account` PARTITION(`date`, `id`) (id, name) VALUES (123,"test"), (124,"test2")')
})
it('should support parse insert on duplicate key update', () => {
const sql = 'INSERT into account partition(date, id) (id, name) values(123, "test"), (124, "test2") on duplicate key update id = 123, name = "test"'
const ast = parser.astify(sql)
const backSQL = parser.sqlify(ast)
expect(backSQL).to.be.equal('INSERT INTO `account` PARTITION(`date`, `id`) (id, name) VALUES (123,"test"), (124,"test2") ON DUPLICATE KEY UPDATE `id` = 123, `name` = "test"')
expect(parser.sqlify(parser.astify(`INSERT INTO user (id, name, age) VALUES (1, 'user1', 50) ON DUPLICATE KEY UPDATE name = VALUES(name), age = VALUES(age)`))).to.be.equal("INSERT INTO `user` (id, name, age) VALUES (1,'user1',50) ON DUPLICATE KEY UPDATE `name` = VALUES(`name`), `age` = VALUES(`age`)")
})
it('should support parse insert set', () => {
const sql = 'INSERT into account partition(date, id) set id = 234, name="my-name" on duplicate key update id = 123, name = "test"'
const ast = parser.astify(sql)
const backSQL = parser.sqlify(ast)
expect(backSQL).to.be.equal('INSERT INTO `account` PARTITION(`date`, `id`) SET `id` = 234, `name` = "my-name" ON DUPLICATE KEY UPDATE `id` = 123, `name` = "test"')
})
it('should support parse insert partition expr', () => {
const sql = 'INSERT into account partition(date = 20191218, id = 2) (id, name) values(123, "test"), (124, "test2")'
const ast = parser.astify(sql)
const backSQL = parser.sqlify(ast)
expect(backSQL).to.be.equal('INSERT INTO `account` PARTITION(`date` = 20191218, `id` = 2) (id, name) VALUES (123,"test"), (124,"test2")')
})
it('should support parse insert partition for hive', () => {
const sql = 'INSERT overwrite table account partition(date, id) select * from tmp'
const ast = parser.astify(sql, { database: 'hive' })
const backSQL = parser.sqlify(ast)
expect(backSQL).to.be.equal("INSERT OVERWRITE TABLE `account` PARTITION(`date`, `id`) SELECT * FROM `tmp`")
})
it('should support parse pg insert returning', () => {
const sql = 'INSERT into account (date, id) values("2019-12-23", 123) returning id'
const opt = { database: 'postgresql' }
const ast = parser.astify(sql, opt)
const backSQL = parser.sqlify(ast, opt)
expect(backSQL).to.be.equal('INSERT INTO "account" (date, id) VALUES ("2019-12-23",123) RETURNING id')
expect(parser.sqlify(parser.astify(`INSERT INTO account (date, id) VALUES ("2019-12-23",123) RETURNING *`, opt), opt)).to.be.equal('INSERT INTO "account" (date, id) VALUES ("2019-12-23",123) RETURNING *')
})
it('should support insert hex value', () => {
// TODO: reserve original quote
expect(parser.sqlify(parser.astify(`INSERT INTO \`t\`
(\`a\`, \`b\`) VALUES
(X'AD', 0x123BF)`))).to.be.equal("INSERT INTO `t` (a, b) VALUES (X'AD',0x123BF)")
})
it('should support replace into', () => {
const sql = "REPLACE INTO test (test_column1, test_column2) VALUES ('testvalue1', 'testvalue2')"
expect(getParsedSql(sql)).to.be.equal("REPLACE INTO `test` (test_column1, test_column2) VALUES ('testvalue1','testvalue2')")
})
describe('support ascii pnCtrl single-char', () => {
it('should support ascii pnCtrl', () => {
// 0～31及127(共33个)是控制字符或通信专用字符 \0-\x1F和\x7f in ascii, ETX ascii code is 0x03
const sql = `INSERT INTO posts (content) VALUES('\\'s ')`
const ast = parser.astify(sql)
const backSQL = parser.sqlify(ast)
expect(backSQL).to.be.equal("INSERT INTO `posts` (content) VALUES ('\\\'s ')")
})
})
});