-
Notifications
You must be signed in to change notification settings - Fork 77
Expand file tree
/
Copy pathtestSet.java
More file actions
67 lines (54 loc) · 2.74 KB
/
Copy pathtestSet.java
File metadata and controls
67 lines (54 loc) · 2.74 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
package hive;
/*
* Date: 13-8-15
*/
import gudusoft.gsqlparser.EDbVendor;
import gudusoft.gsqlparser.EExpressionType;
import gudusoft.gsqlparser.TGSqlParser;
import gudusoft.gsqlparser.nodes.TExpression;
import gudusoft.gsqlparser.nodes.hive.THiveVariable;
import gudusoft.gsqlparser.stmt.hive.THiveSet;
import junit.framework.TestCase;
public class testSet extends TestCase {
public void test1(){
TGSqlParser sqlparser = new TGSqlParser(EDbVendor.dbvhive);
sqlparser.sqltext = "set system:xxx=5;";
assertTrue(sqlparser.parse() == 0);
THiveSet set = (THiveSet)sqlparser.sqlstatements.get(0);
TExpression expression = set.getExpr();
assertTrue(expression.getLeftOperand().toString().equalsIgnoreCase("system:xxx"));
assertTrue(expression.getRightOperand().toString().equalsIgnoreCase("5"));
}
public void test2(){
TGSqlParser sqlparser = new TGSqlParser(EDbVendor.dbvhive);
sqlparser.sqltext = "set system:yyy=${system:xxx};";
assertTrue(sqlparser.parse() == 0);
THiveSet set = (THiveSet)sqlparser.sqlstatements.get(0);
TExpression expression = set.getExpr();
assertTrue(expression.getLeftOperand().toString().equalsIgnoreCase("system:yyy"));
assertTrue(expression.getRightOperand().toString().equalsIgnoreCase("${system:xxx}"));
TExpression right = expression.getRightOperand();
assertTrue(right.getExpressionType() == EExpressionType.hive_variable_t);
THiveVariable v = right.getHive_variable();
assertTrue(v.getVarName().toString().equalsIgnoreCase("system"));
assertTrue(v.getVarProperty().toString().equalsIgnoreCase("xxx"));
}
public void test3(){
TGSqlParser sqlparser = new TGSqlParser(EDbVendor.dbvhive);
sqlparser.sqltext = "set c=${hiveconf:${hiveconf:b}};";
assertTrue(sqlparser.parse() == 0);
THiveSet set = (THiveSet)sqlparser.sqlstatements.get(0);
TExpression expression = set.getExpr();
assertTrue(expression.getLeftOperand().toString().equalsIgnoreCase("c"));
assertTrue(expression.getRightOperand().toString().equalsIgnoreCase("${hiveconf:${hiveconf:b}}"));
TExpression right = expression.getRightOperand();
assertTrue(right.getExpressionType() == EExpressionType.hive_variable_t);
THiveVariable v = right.getHive_variable();
assertTrue(v.getVarName().toString().equalsIgnoreCase("hiveconf"));
assertTrue(v.getVarProperty() == null);
assertTrue(v.getNestedVar().toString().equalsIgnoreCase("${hiveconf:b}"));
v = v.getNestedVar();
assertTrue(v.getVarName().toString().equalsIgnoreCase("hiveconf"));
assertTrue(v.getVarProperty().toString().equalsIgnoreCase("b"));
}
}