Skip to content

Commit 79cb474

Browse files
committed
bugfix: insert test case
1 parent 025955b commit 79cb474

File tree

2 files changed

+44
-0
lines changed

2 files changed

+44
-0
lines changed

sqlparser/seata-sqlparser-druid/src/test/java/io/seata/sqlparser/druid/MySQLInsertRecognizerTest.java

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -174,4 +174,26 @@ public void testGetInsertRows() {
174174
public String getDbType() {
175175
return JdbcConstants.MYSQL;
176176
}
177+
178+
@Test
179+
public void testGetInsertColumns_2() {
180+
String sql = "insert into t(`id`, `no`, `name`, `age`) values (1, 'no001', 'aaa', '20')";
181+
List<SQLStatement> asts = SQLUtils.parseStatements(sql, JdbcConstants.MYSQL);
182+
MySQLInsertRecognizer recognizer = new MySQLInsertRecognizer(sql, asts.get(0));
183+
List<String> insertColumns = recognizer.getInsertColumns();
184+
for (String insertColumn : insertColumns) {
185+
Assertions.assertFalse(insertColumn.contains("`"));
186+
}
187+
}
188+
189+
@Test
190+
public void testGetInsertColumnsIsSimplified() {
191+
String sql = "insert into t(`id`, `no`, `name`, `age`) values (1, 'no001', 'aaa', '20')";
192+
List<SQLStatement> asts = SQLUtils.parseStatements(sql, JdbcConstants.MYSQL);
193+
MySQLInsertRecognizer recognizer = new MySQLInsertRecognizer(sql, asts.get(0));
194+
List<String> insertColumns = recognizer.getInsertColumnsIsSimplified();
195+
for (String insertColumn : insertColumns) {
196+
Assertions.assertFalse(insertColumn.contains("`"));
197+
}
198+
}
177199
}

sqlparser/seata-sqlparser-druid/src/test/java/io/seata/sqlparser/druid/MySQLUpdateRecognizerTest.java

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -354,4 +354,26 @@ public void testGetTableAlias() {
354354
public String getDbType() {
355355
return JdbcConstants.MYSQL;
356356
}
357+
358+
@Test
359+
public void testGetUpdateColumns_2() {
360+
String sql = "update t set `a` = 1, `b` = 2, `c` = 3";
361+
List<SQLStatement> asts = SQLUtils.parseStatements(sql, JdbcConstants.MYSQL);
362+
MySQLUpdateRecognizer recognizer = new MySQLUpdateRecognizer(sql, asts.get(0));
363+
List<String> updateColumns = recognizer.getUpdateColumns();
364+
for (String updateColumn : updateColumns) {
365+
Assertions.assertFalse(updateColumn.contains("`"));
366+
}
367+
}
368+
369+
@Test
370+
public void testGetUpdateColumnsIsSimplified() {
371+
String sql = "update t set `a` = 1, `b` = 2, `c` = 3";
372+
List<SQLStatement> asts = SQLUtils.parseStatements(sql, JdbcConstants.MYSQL);
373+
MySQLUpdateRecognizer recognizer = new MySQLUpdateRecognizer(sql, asts.get(0));
374+
List<String> updateColumns = recognizer.getUpdateColumnsIsSimplified();
375+
for (String updateColumn : updateColumns) {
376+
Assertions.assertFalse(updateColumn.contains("`"));
377+
}
378+
}
357379
}

0 commit comments

Comments
 (0)