Alawn/Druid
1
0
Fork 0
mirror of https://github.com/alibaba/druid.git synced 2023-03-04 19:17:16 +00:00
Code Issues Releases Wiki Activity

enhance sql parser & add test

Browse Source
This commit is contained in:
温少
2012-02-23 20:58:23 +08:00
parent 2d5c963201 a08c9e8688
commit 9c69b1a1c7
17 changed files with 132 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/test/java/com/alibaba/druid/bvt/sql/mysql/MySqlSelectTest_0.java
+43
View File
@@ -0,0 +1,43 @@
package com.alibaba.druid.bvt.sql.mysql;
import java.util.List;
import junit.framework.Assert;
import com.alibaba.druid.sql.MysqlTest;
import com.alibaba.druid.sql.ast.SQLStatement;
import com.alibaba.druid.sql.dialect.mysql.parser.MySqlStatementParser;
import com.alibaba.druid.sql.dialect.mysql.visitor.MySqlSchemaStatVisitor;
import com.alibaba.druid.stat.TableStat;
import com.alibaba.druid.stat.TableStat.Column;
public class MySqlSelectTest_0 extends MysqlTest {
public void test_0() throws Exception {
String sql = "SELECT CONCAT(last_name,', ',first_name) AS full_name FROM mytable ORDER BY full_name;";
MySqlStatementParser parser = new MySqlStatementParser(sql);
List<SQLStatement> statementList = parser.parseStatementList();
SQLStatement statemen = statementList.get(0);
System.out.println(output(statementList));
Assert.assertEquals(1, statementList.size());
MySqlSchemaStatVisitor visitor = new MySqlSchemaStatVisitor();
statemen.accept(visitor);
System.out.println("Tables : " + visitor.getTables());
System.out.println("fields : " + visitor.getFields());
System.out.println("coditions : " + visitor.getConditions());
Assert.assertEquals(1, visitor.getTables().size());
Assert.assertEquals(3, visitor.getFields().size());
Assert.assertEquals(0, visitor.getConditions().size());
Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("mytable")));
Assert.assertTrue(visitor.getFields().contains(new Column("mytable", "last_name")));
Assert.assertTrue(visitor.getFields().contains(new Column("mytable", "first_name")));
Assert.assertTrue(visitor.getFields().contains(new Column("mytable", "full_name")));
}
}
src/test/java/com/alibaba/druid/bvt/sql/mysql/MySqlSelectTest_1.java
+44
View File
@@ -0,0 +1,44 @@
package com.alibaba.druid.bvt.sql.mysql;
import java.util.List;
import junit.framework.Assert;
import com.alibaba.druid.sql.MysqlTest;
import com.alibaba.druid.sql.ast.SQLStatement;
import com.alibaba.druid.sql.dialect.mysql.parser.MySqlStatementParser;
import com.alibaba.druid.sql.dialect.mysql.visitor.MySqlSchemaStatVisitor;
import com.alibaba.druid.stat.TableStat;
import com.alibaba.druid.stat.TableStat.Column;
public class MySqlSelectTest_1 extends MysqlTest {
public void test_0() throws Exception {
String sql = "SELECT t1.name, t2.salary FROM employee t1, info t2 WHERE t1.name = t2.name;";
MySqlStatementParser parser = new MySqlStatementParser(sql);
List<SQLStatement> statementList = parser.parseStatementList();
SQLStatement statemen = statementList.get(0);
System.out.println(output(statementList));
Assert.assertEquals(1, statementList.size());
MySqlSchemaStatVisitor visitor = new MySqlSchemaStatVisitor();
statemen.accept(visitor);
System.out.println("Tables : " + visitor.getTables());
System.out.println("fields : " + visitor.getFields());
System.out.println("coditions : " + visitor.getConditions());
Assert.assertEquals(2, visitor.getTables().size());
Assert.assertEquals(3, visitor.getFields().size());
Assert.assertEquals(2, visitor.getConditions().size());
Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("employee")));
Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("info")));
Assert.assertTrue(visitor.getFields().contains(new Column("employee", "name")));
Assert.assertTrue(visitor.getFields().contains(new Column("info", "name")));
Assert.assertTrue(visitor.getFields().contains(new Column("info", "salary")));
}
}
src/test/java/com/alibaba/druid/bvt/sql/postgresql/PGDeleteTest5.java
+1
View File
@@ -27,6 +27,7 @@ public class PGDeleteTest5 extends PGTest {
System.out.println("Tables : " + visitor.getTables());
System.out.println("fields : " + visitor.getFields());
System.out.println("coditions : " + visitor.getConditions());
Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("films")));
Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("producers")));
src/test/java/com/alibaba/druid/bvt/sql/postgresql/PGInsertTest0.java
+1
View File
@@ -27,6 +27,7 @@ public class PGInsertTest0 extends PGTest {
System.out.println("Tables : " + visitor.getTables());
System.out.println("fields : " + visitor.getFields());
System.out.println("coditions : " + visitor.getConditions());
Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("films")));
src/test/java/com/alibaba/druid/bvt/sql/postgresql/PGInsertTest1.java
+1
View File
@@ -27,6 +27,7 @@ public class PGInsertTest1 extends PGTest {
System.out.println("Tables : " + visitor.getTables());
System.out.println("fields : " + visitor.getFields());
System.out.println("coditions : " + visitor.getConditions());
Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("films")));
src/test/java/com/alibaba/druid/bvt/sql/postgresql/PGInsertTest3.java
+1
View File
@@ -27,6 +27,7 @@ public class PGInsertTest3 extends PGTest {
System.out.println("Tables : " + visitor.getTables());
System.out.println("fields : " + visitor.getFields());
System.out.println("coditions : " + visitor.getConditions());
Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("films")));
Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("tmp_films")));
src/test/java/com/alibaba/druid/bvt/sql/postgresql/PGInsertTest4.java
+1
View File
@@ -27,6 +27,7 @@ public class PGInsertTest4 extends PGTest {
System.out.println("Tables : " + visitor.getTables());
System.out.println("fields : " + visitor.getFields());
System.out.println("coditions : " + visitor.getConditions());
Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("distributors")));
src/test/java/com/alibaba/druid/bvt/sql/postgresql/PGInsertTest5.java
+1
View File
@@ -32,6 +32,7 @@ public class PGInsertTest5 extends PGTest {
System.out.println("Tables : " + visitor.getTables());
System.out.println("fields : " + visitor.getFields());
System.out.println("coditions : " + visitor.getConditions());
Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("employees_log")));
Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("employees")));
src/test/java/com/alibaba/druid/bvt/sql/postgresql/PGSelectIntoTest.java
+1
View File
@@ -27,6 +27,7 @@ public class PGSelectIntoTest extends PGTest {
System.out.println("Tables : " + visitor.getTables());
System.out.println("fields : " + visitor.getFields());
System.out.println("coditions : " + visitor.getConditions());
Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("films_recent")));
Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("films")));
src/test/java/com/alibaba/druid/bvt/sql/postgresql/PostgresqlResourceTest.java
+1
View File
@@ -49,6 +49,7 @@ public class PostgresqlResourceTest extends PGTest {
System.out.println(sql);
System.out.println("Tables : " + visitor.getTables());
System.out.println("fields : " + visitor.getFields());
System.out.println("coditions : " + visitor.getConditions());
System.out.println();
System.out.println();