-
Notifications
You must be signed in to change notification settings - Fork 77
Expand file tree
/
Copy pathtestPivot.java
More file actions
68 lines (53 loc) · 3.06 KB
/
Copy pathtestPivot.java
File metadata and controls
68 lines (53 loc) · 3.06 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
package test.oracle;
/*
* Date: 13-1-12
*/
import gudusoft.gsqlparser.EDbVendor;
import gudusoft.gsqlparser.ETableSource;
import gudusoft.gsqlparser.TGSqlParser;
import gudusoft.gsqlparser.nodes.*;
import gudusoft.gsqlparser.stmt.TSelectSqlStatement;
import junit.framework.TestCase;
public class testPivot extends TestCase {
public void testOracle1(){
TGSqlParser sqlparser = new TGSqlParser(EDbVendor.dbvoracle);
sqlparser.sqltext = "SELECT * FROM pivot_table\n" +
" UNPIVOT (yearly_total FOR order_mode IN (store AS 'direct', internet AS 'online'))\n" +
" ORDER BY year, order_mode;";
assertTrue(sqlparser.parse() == 0);
TSelectSqlStatement select = (TSelectSqlStatement)sqlparser.sqlstatements.get(0);
assertTrue(select.getTargetTable().getTableType() == ETableSource.pivoted_table);
TTable table = select.tables.getTable(0);
assertTrue(table.toString().equalsIgnoreCase("pivot_table"));
TPivotedTable pivotedTable = select.getTargetTable().getPivotedTable();
TPivotClause pivotClause = pivotedTable.getPivotClauseList().getElement(0) ;
assertTrue(pivotClause.getType() == TPivotClause.unpivot);
assertTrue(pivotClause.getValueColumnList().getObjectName(0).toString().equalsIgnoreCase("yearly_total"));
assertTrue(pivotClause.getPivotColumnList().getObjectName(0).toString().equalsIgnoreCase("order_mode"));
TUnpivotInClause inClause = pivotClause.getUnpivotInClause();
TUnpivotInClauseItem item0 = inClause.getItems().getElement(0);
assertTrue(item0.getColumn().toString().equalsIgnoreCase("store"));
assertTrue(item0.getConstant().toString().equalsIgnoreCase("'direct'"));
TUnpivotInClauseItem item1 = inClause.getItems().getElement(1);
assertTrue(item1.getColumn().toString().equalsIgnoreCase("internet"));
assertTrue(item1.getConstant().toString().equalsIgnoreCase("'online'"));
}
public void testOracle2(){
TGSqlParser sqlparser = new TGSqlParser(EDbVendor.dbvoracle);
sqlparser.sqltext = "SELECT * FROM orders\n" +
"PIVOT(SUM(order_total) \n" +
"\t\tFOR order_mode IN (SELECT 'direct' AS Store, 'online' AS Internet FROM orders));";
assertTrue(sqlparser.parse() == 0);
TSelectSqlStatement select = (TSelectSqlStatement)sqlparser.sqlstatements.get(0);
assertTrue(select.getTargetTable().getTableType() == ETableSource.pivoted_table);
TTable table = select.tables.getTable(0);
assertTrue(table.toString().equalsIgnoreCase("orders"));
TPivotedTable pivotedTable = select.getTargetTable().getPivotedTable();
TPivotClause pivotClause = pivotedTable.getPivotClauseList().getElement(0);
assertTrue(pivotClause.getType() == TPivotClause.pivot);
TPivotInClause inClause = pivotClause.getPivotInClause();
select = inClause.getSubQuery();
assertTrue(select.getResultColumnList().size() == 2);
assertTrue(select.tables.getTable(0).toString().equalsIgnoreCase("orders"));
}
}