增强测试

git-svn-id: http://code.alibabatech.com/svn/druid/trunk@984 b9813039-fb51-4c41-a8b9-e21c2acb5095
2023-03-04 19:17:16 +00:00 · 2011-10-15 10:47:06 +00:00
parent 2f59c50d92
commit 0525a0d229
4 changed files with 156 additions and 3 deletions
@@ -0,0 +1,106 @@
+package com.alibaba.druid.bvt.sql.oracle;
+
+import junit.framework.Assert;
+import junit.framework.TestCase;
+
+import com.alibaba.druid.sql.ast.statement.SQLSelectStatement;
+import com.alibaba.druid.sql.dialect.oracle.parser.OracleStatementParser;
+import com.alibaba.druid.sql.test.TestUtils;
+
+public class OraclePivotClauseTest extends TestCase {
+
+    public void test_0() throws Exception {
+        String sql = "SELECT *\n" + "FROM pivot_table\n"
+                     + "UNPIVOT (yearly_total FOR order_mode IN (store AS 'direct', internet AS 'online'))\n"
+                     + "ORDER BY year, order_mode;";
+
+        String expected = "SELECT *\n" + "FROM pivot_table\n"
+                + "\tUNPIVOT (yearly_total FOR order_mode IN (store AS 'direct', internet AS 'online'))\n"
+                + "ORDER BY YEAR, order_mode;\n";
+
+        OracleStatementParser parser = new OracleStatementParser(sql);
+        SQLSelectStatement stmt = (SQLSelectStatement) parser.parseStatementList().get(0);
+
+        String text = TestUtils.outputOracle(stmt);
+
+        Assert.assertEquals(expected, text);
+
+        System.out.println(text);
+    }
+
+    public void test_pivot() throws Exception {
+        String sql = "SELECT *\n" + "FROM (SELECT EXTRACT(YEAR FROM order_date) year, order_mode, order_total FROM orders)\n" +
+        		"PIVOT (SUM(order_total) FOR order_mode IN ('direct' AS Store, 'online' AS Internet));\n";
+
+        String expected = "SELECT *\n" + "FROM (SELECT EXTRACT(YEAR FROM order_date) AS YEAR, order_mode, order_total\n" +
+        		"\tFROM orders\n" +
+        		"\t)\n" +
+                "\tPIVOT (SUM(order_total) FOR order_mode IN ('direct' AS Store, 'online' AS Internet));\n";
+
+        OracleStatementParser parser = new OracleStatementParser(sql);
+        SQLSelectStatement stmt = (SQLSelectStatement) parser.parseStatementList().get(0);
+
+        String text = TestUtils.outputOracle(stmt);
+
+        Assert.assertEquals(expected, text);
+
+        System.out.println(text);
+    }
+    
+    public void test_pivot_1() throws Exception {
+        String sql = "SELECT *\n" + "FROM (SELECT EXTRACT(YEAR FROM order_date) as year, order_mode, order_total FROM orders)\n" +
+                "PIVOT (SUM(order_total) FOR order_mode IN ('direct' AS Store, 'online' AS Internet));\n";
+
+        String expected = "SELECT *\n" + "FROM (SELECT EXTRACT(YEAR FROM order_date) AS YEAR, order_mode, order_total\n" +
+                "\tFROM orders\n" +
+                "\t)\n" +
+                "\tPIVOT (SUM(order_total) FOR order_mode IN ('direct' AS Store, 'online' AS Internet));\n";
+
+        OracleStatementParser parser = new OracleStatementParser(sql);
+        SQLSelectStatement stmt = (SQLSelectStatement) parser.parseStatementList().get(0);
+
+        String text = TestUtils.outputOracle(stmt);
+
+        Assert.assertEquals(expected, text);
+
+        System.out.println(text);
+    }
+    
+    public void test_pivot_2() throws Exception {
+        String sql = "SELECT *\n" + "FROM (SELECT EXTRACT(YEAR FROM order_date) as day, order_mode, order_total FROM orders)\n" +
+                "PIVOT (SUM(order_total) FOR order_mode IN ('direct' AS Store, 'online' AS Internet));\n";
+
+        String expected = "SELECT *\n" + "FROM (SELECT EXTRACT(YEAR FROM order_date) AS DAY, order_mode, order_total\n" +
+                "\tFROM orders\n" +
+                "\t)\n" +
+                "\tPIVOT (SUM(order_total) FOR order_mode IN ('direct' AS Store, 'online' AS Internet));\n";
+
+        OracleStatementParser parser = new OracleStatementParser(sql);
+        SQLSelectStatement stmt = (SQLSelectStatement) parser.parseStatementList().get(0);
+
+        String text = TestUtils.outputOracle(stmt);
+
+        Assert.assertEquals(expected, text);
+
+        System.out.println(text);
+    }
+    
+    public void test_pivot_3() throws Exception {
+        String sql = "SELECT *\n" + "FROM (SELECT EXTRACT(YEAR FROM order_date) day, order_mode YEAR, order_total FROM orders)\n" +
+                "PIVOT (SUM(order_total) FOR order_mode IN ('direct' AS Store, 'online' AS Internet));\n";
+
+        String expected = "SELECT *\n" + "FROM (SELECT EXTRACT(YEAR FROM order_date) AS DAY, order_mode AS YEAR, order_total\n" +
+                "\tFROM orders\n" +
+                "\t)\n" +
+                "\tPIVOT (SUM(order_total) FOR order_mode IN ('direct' AS Store, 'online' AS Internet));\n";
+
+        OracleStatementParser parser = new OracleStatementParser(sql);
+        SQLSelectStatement stmt = (SQLSelectStatement) parser.parseStatementList().get(0);
+
+        String text = TestUtils.outputOracle(stmt);
+
+        Assert.assertEquals(expected, text);
+
+        System.out.println(text);
+    }
+}