-
Notifications
You must be signed in to change notification settings - Fork 77
Expand file tree
/
Copy pathtestTSampleClause.java
More file actions
85 lines (71 loc) · 4.33 KB
/
Copy pathtestTSampleClause.java
File metadata and controls
85 lines (71 loc) · 4.33 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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
package test;
/*
* Date: 2010-10-18
* Time: 16:11:32
*/
import junit.framework.TestCase;
import gudusoft.gsqlparser.TGSqlParser;
import gudusoft.gsqlparser.EDbVendor;
import gudusoft.gsqlparser.nodes.TSampleClause;
import gudusoft.gsqlparser.nodes.TWhenClauseItem;
import gudusoft.gsqlparser.stmt.TSelectSqlStatement;
public class testTSampleClause extends TestCase {
public void test1(){
TGSqlParser sqlparser = new TGSqlParser(EDbVendor.dbvteradata);
sqlparser.sqltext = "SELECT customer_id, age, income, marital_status, SAMPLEID\n" +
"FROM customer_table\n" +
"SAMPLE 0.6, 0.25, 0.15;";
assertTrue(sqlparser.parse() == 0);
TSelectSqlStatement select = (TSelectSqlStatement)sqlparser.sqlstatements.get(0);
TSampleClause sampleClause = select.getSampleClause();
assertTrue(sampleClause.getCount_fraction_description_list().size() == 3);
assertTrue(sampleClause.getCount_fraction_description_list().getConstant(0).toString().equalsIgnoreCase("0.6"));
assertTrue(sampleClause.getCount_fraction_description_list().getConstant(1).toString().equalsIgnoreCase("0.25"));
assertTrue(sampleClause.getCount_fraction_description_list().getConstant(2).toString().equalsIgnoreCase("0.15"));
}
public void test2(){
TGSqlParser sqlparser = new TGSqlParser(EDbVendor.dbvteradata);
sqlparser.sqltext = "SELECT city, state, sampleid\n" +
"FROM stores\n" +
"SAMPLE RANDOMIZED ALLOCATION\n" +
"WHEN state = 'WI' THEN 0.25, 0.5\n" +
"WHEN state = 'CA' THEN 0.25, 0.25\n" +
"END\n" +
"ORDER BY 3;";
assertTrue(sqlparser.parse() == 0);
TSelectSqlStatement select = (TSelectSqlStatement)sqlparser.sqlstatements.get(0);
TSampleClause sampleClause = select.getSampleClause();
assertTrue(sampleClause.getWhenClauseItems().size() == 2);
TWhenClauseItem item0 = sampleClause.getWhenClauseItems().getWhenClauseItem(0);
TWhenClauseItem item1 = sampleClause.getWhenClauseItems().getWhenClauseItem(1);
assertTrue(item0.getComparison_expr().toString().equalsIgnoreCase("state = 'WI'"));
assertTrue(item0.getCount_fraction_description_list().getConstant(0).toString().equalsIgnoreCase("0.25"));
assertTrue(item0.getCount_fraction_description_list().getConstant(1).toString().equalsIgnoreCase("0.5"));
assertTrue(item1.getComparison_expr().toString().equalsIgnoreCase("state = 'CA'"));
assertTrue(item1.getCount_fraction_description_list().getConstant(0).toString().equalsIgnoreCase("0.25"));
assertTrue(item1.getCount_fraction_description_list().getConstant(1).toString().equalsIgnoreCase("0.25"));
//assertTrue(sampleClause.getCount_fraction_description_list().size() == 3);
//assertTrue(sampleClause.getCount_fraction_description_list().getConstant(0).toString().equalsIgnoreCase("0.6"));
//assertTrue(sampleClause.getCount_fraction_description_list().getConstant(1).toString().equalsIgnoreCase("0.25"));
//assertTrue(sampleClause.getCount_fraction_description_list().getConstant(2).toString().equalsIgnoreCase("0.15"));
}
public void test3(){
TGSqlParser sqlparser = new TGSqlParser(EDbVendor.dbvteradata);
sqlparser.sqltext = "SELECT city, state, sampleid\n" +
"FROM stores\n" +
"SAMPLE WITH REPLACEMENT RANDOMIZED ALLOCATION\n" +
"WHEN state = 'WI' THEN 3\n" +
"ELSE 2\n" +
"END\n" +
"ORDER BY 3;";
assertTrue(sqlparser.parse() == 0);
TSelectSqlStatement select = (TSelectSqlStatement)sqlparser.sqlstatements.get(0);
TSampleClause sampleClause = select.getSampleClause();
assertTrue(sampleClause.getWhenClauseItems().size() == 1);
TWhenClauseItem item0 = sampleClause.getWhenClauseItems().getWhenClauseItem(0);
assertTrue(item0.getComparison_expr().toString().equalsIgnoreCase("state = 'WI'"));
assertTrue(item0.getCount_fraction_description_list().getConstant(0).toString().equalsIgnoreCase("3"));
assertTrue(sampleClause.getCount_fraction_description_list().size() == 1);
assertTrue(sampleClause.getCount_fraction_description_list().getConstant(0).toString().equalsIgnoreCase("2"));
}
}