修正输出格式

git-svn-id: http://code.alibabatech.com/svn/druid/trunk@159 b9813039-fb51-4c41-a8b9-e21c2acb5095

src/test/java/com/alibaba/druid/sql/mysql/bvt/CastFunctionsAndOperatorsTest.java

@@ -112,7 +112,7 @@ public class CastFunctionsAndOperatorsTest extends TestCase {
 
         String text = output(stmtList);
 
-        Assert.assertEquals("SELECT enum_col\nFROM tbl_name ORDER BY CAST(enum_col AS CHAR);", text);
+        Assert.assertEquals("SELECT enum_col\nFROM tbl_name\nORDER BY CAST(enum_col AS CHAR);", text);
     }
 
     public void test_8() throws Exception {

src/test/java/com/alibaba/druid/sql/mysql/bvt/MathematicalFunctionsTest.java

@@ -310,7 +310,7 @@ public class MathematicalFunctionsTest extends TestCase {
 
         String text = output(stmtList);
 
-        Assert.assertEquals("SELECT *\nFROM tbl_name ORDER BY RAND();", text);
+        Assert.assertEquals("SELECT *\nFROM tbl_name\nORDER BY RAND();", text);
     }
 
     public void test_26() throws Exception {

src/test/java/com/alibaba/druid/sql/mysql/bvt/parser/MySqlParserResourceTest.java

@@ -0,0 +1,56 @@
+package com.alibaba.druid.sql.mysql.bvt.parser;
+
+import java.io.InputStream;
+import java.io.InputStreamReader;
+import java.io.Reader;
+import java.util.List;
+
+import junit.framework.Assert;
+import junit.framework.TestCase;
+
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.mysql.parser.MySqlStatementParser;
+import com.alibaba.druid.sql.dialect.mysql.visitor.MySqlOutputVisitor;
+import com.alibaba.druid.util.JdbcUtils;
+
+public class MySqlParserResourceTest extends TestCase {
+
+    public void test_0() throws Exception {
+        exec_test("bvt/parser/mysql-0.txt");
+    }
+    
+    
+    public void exec_test(String resource) throws Exception {
+        System.out.println(resource);
+        InputStream is = null;
+
+        is = Thread.currentThread().getContextClassLoader().getResourceAsStream(resource);
+        Reader reader = new InputStreamReader(is, "UTF-8");
+        String input = JdbcUtils.read(reader);
+        JdbcUtils.close(reader);
+        String[] items = input.split("---------------------------");
+        String sql = items[0].trim();
+        String expect = items[1].trim();
+        
+        MySqlStatementParser parser = new MySqlStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        
+        Assert.assertEquals(1, statementList.size());
+        
+        String text = output(statementList);
+        System.out.println(text);
+        Assert.assertEquals(expect, text.trim());
+       
+    }
+
+    private String output(List<SQLStatement> stmtList) {
+        StringBuilder out = new StringBuilder();
+        MySqlOutputVisitor visitor = new MySqlOutputVisitor(out);
+
+        for (SQLStatement stmt : stmtList) {
+            stmt.accept(visitor);
+        }
+
+        return out.toString();
+    }
+}