-
Notifications
You must be signed in to change notification settings - Fork 77
Expand file tree
/
Copy pathtestSetVariable.java
More file actions
33 lines (27 loc) · 1.44 KB
/
Copy pathtestSetVariable.java
File metadata and controls
33 lines (27 loc) · 1.44 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
package postgresql;
import gudusoft.gsqlparser.EDbVendor;
import gudusoft.gsqlparser.ESetStatementType;
import gudusoft.gsqlparser.TGSqlParser;
import gudusoft.gsqlparser.stmt.TSetStmt;
import junit.framework.TestCase;
public class testSetVariable extends TestCase {
public void test1(){
TGSqlParser sqlparser = new TGSqlParser(EDbVendor.dbvpostgresql);
sqlparser.sqltext = "SET statement_timeout = 0;";
assertTrue(sqlparser.parse() == 0);
TSetStmt setStmt = (TSetStmt)sqlparser.sqlstatements.get(0);
assertTrue(setStmt.getSetStatementType() == ESetStatementType.variable);
assertTrue(setStmt.getVariableName().toString().equalsIgnoreCase("statement_timeout"));
assertTrue(setStmt.getVariableValue().toString().equalsIgnoreCase("0"));
}
public void test2(){
TGSqlParser sqlparser = new TGSqlParser(EDbVendor.dbvpostgresql);
sqlparser.sqltext = "SET search_path = dwd, pg_catalog;";
assertTrue(sqlparser.parse() == 0);
TSetStmt setStmt = (TSetStmt)sqlparser.sqlstatements.get(0);
assertTrue(setStmt.getSetStatementType() == ESetStatementType.variable);
assertTrue(setStmt.getVariableName().toString().equalsIgnoreCase("search_path"));
assertTrue(setStmt.getVariableValueList().getExpression(0).toString().equalsIgnoreCase("dwd"));
assertTrue(setStmt.getVariableValueList().getExpression(1).toString().equalsIgnoreCase("pg_catalog"));
}
}