bug fixed for SchemaStat

src/test/java/com/alibaba/druid/bvt/sql/mysql/visitor/MySqlSchemaStatVisitorTest5.java

@@ -18,7 +18,7 @@ public class MySqlSchemaStatVisitorTest5 extends TestCase {
 				+ "  LEFT JOIN (select * from view_portal_panel) c  ON a.panel_id = c.panel_id   "
 				+ " WHERE     b.user_id = ? and     ((b.is_grid='y' and c.param_name='is_hidden' and c.param_value='false') or      b.is_grid  != 'y') and b.user_id in (select user_id from table1 where id = 1)    ORDER BY    a.col ASC, a.position ASC";
 
-		sql = "select columnName from table1 where id in (select id from table3 where name = ?)";
+//		sql = "select columnName from table1 where id in (select id from table3 where name = ?)";
 		MySqlStatementParser parser = new MySqlStatementParser(sql);
 		List<SQLStatement> statementList = parser.parseStatementList();
 		SQLStatement statemen = statementList.get(0);

src/test/java/com/alibaba/druid/bvt/sql/postgresql/PostgresqlResourceTest.java

@@ -6,15 +6,15 @@ import java.io.Reader;
 import java.util.List;
 
 import junit.framework.Assert;
-import junit.framework.TestCase;
 
+import com.alibaba.druid.sql.PGTest;
 import com.alibaba.druid.sql.ast.SQLStatement;
 import com.alibaba.druid.sql.dialect.postgresql.parser.PGSQLStatementParser;
 import com.alibaba.druid.sql.dialect.postgresql.visitor.PGOutputVisitor;
 import com.alibaba.druid.sql.dialect.postgresql.visitor.PGSchemaStatVisitor;
 import com.alibaba.druid.util.JdbcUtils;
 
-public class PostgresqlResourceTest extends TestCase {
+public class PostgresqlResourceTest extends PGTest {
 
     public void test_0() throws Exception {
         // 13
@@ -70,14 +70,5 @@ public class PostgresqlResourceTest extends TestCase {
         Assert.assertEquals(expect, out.toString());
     }
 
-    private String output(List<SQLStatement> stmtList) {
-        StringBuilder out = new StringBuilder();
-        PGOutputVisitor visitor = new PGOutputVisitor(out);
 
-        for (SQLStatement stmt : stmtList) {
-            stmt.accept(visitor);
-        }
-
-        return out.toString();
-    }
 }

src/test/java/com/alibaba/druid/bvt/sql/postgresql/PostgresqlTest.java

@@ -0,0 +1,37 @@
+package com.alibaba.druid.bvt.sql.postgresql;
+
+import java.util.List;
+
+import junit.framework.Assert;
+
+import com.alibaba.druid.sql.PGTest;
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.postgresql.parser.PGSQLStatementParser;
+import com.alibaba.druid.sql.dialect.postgresql.visitor.PGSchemaStatVisitor;
+import com.alibaba.druid.stat.TableStat;
+import com.alibaba.druid.stat.TableStat.Column;
+
+public class PostgresqlTest extends PGTest {
+	public void test_0() throws Exception {
+		String sql = "SELECT * INTO films_recent FROM films WHERE date_prod >= '2002-01-01';";
+		
+        PGSQLStatementParser parser = new PGSQLStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+        System.out.println(output(statementList));
+
+        Assert.assertEquals(1, statementList.size());
+
+        PGSchemaStatVisitor visitor = new PGSchemaStatVisitor();
+        statemen.accept(visitor);
+        
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getFields());
+        
+        Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("films_recent")));
+        Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("films")));
+        
+        Assert.assertTrue(visitor.getFields().contains(new Column("films", "*")));
+        Assert.assertTrue(visitor.getFields().contains(new Column("films", "date_prod")));
+	}
+}

src/test/java/com/alibaba/druid/sql/PGTest.java

@@ -0,0 +1,21 @@
+package com.alibaba.druid.sql;
+
+import java.util.List;
+
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.postgresql.visitor.PGOutputVisitor;
+
+import junit.framework.TestCase;
+
+public class PGTest extends TestCase {
+	protected String output(List<SQLStatement> stmtList) {
+		StringBuilder out = new StringBuilder();
+		PGOutputVisitor visitor = new PGOutputVisitor(out);
+
+		for (SQLStatement stmt : stmtList) {
+			stmt.accept(visitor);
+		}
+
+		return out.toString();
+	}
+}