-
Notifications
You must be signed in to change notification settings - Fork 77
Expand file tree
/
Copy pathtestPlpgsql_block.java
More file actions
80 lines (69 loc) · 3.8 KB
/
Copy pathtestPlpgsql_block.java
File metadata and controls
80 lines (69 loc) · 3.8 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
package test.postgresql;
/*
* Date: 13-12-4
*/
import gudusoft.gsqlparser.*;
import gudusoft.gsqlparser.nodes.TVarDeclStmt;
import gudusoft.gsqlparser.stmt.TAssignStmt;
import gudusoft.gsqlparser.stmt.TCommonBlock;
import gudusoft.gsqlparser.stmt.TRaiseStmt;
import gudusoft.gsqlparser.stmt.postgresql.TPostgresqlCreateFunction;
import junit.framework.TestCase;
public class testPlpgsql_block extends TestCase {
public void test1(){
TGSqlParser sqlparser = new TGSqlParser(EDbVendor.dbvpostgresql);
sqlparser.sqltext = "CREATE FUNCTION somefunc() RETURNS integer AS $$\n" +
"<< outerblock >>\n" +
"DECLARE\n" +
" quantity integer := 30;\n" +
"BEGIN\n" +
" RAISE NOTICE 'Quantity here is %', quantity; -- Prints 30\n" +
" quantity := 50;\n" +
" --\n" +
" -- Create a subblock\n" +
" --\n" +
" DECLARE\n" +
" quantity integer := 80;\n" +
" BEGIN\n" +
" RAISE NOTICE 'Quantity here is %', quantity; -- Prints 80\n" +
" RAISE NOTICE 'Outer quantity here is %', outerblock.quantity; -- Prints 50\n" +
" END;\n" +
"\n" +
" RAISE NOTICE 'Quantity here is %', quantity; -- Prints 50\n" +
"\n" +
" RETURN quantity;\n" +
"END;\n" +
"$$ LANGUAGE plpgsql;";
assertTrue(sqlparser.parse() == 0);
TPostgresqlCreateFunction createFunction = (TPostgresqlCreateFunction)sqlparser.sqlstatements.get(0);
assertTrue(createFunction.getReturnDataType().getDataType() == EDataType.int_t);
assertTrue(createFunction.getBodyStatements().size() == 5);
TCustomSqlStatement stmt = createFunction.getBodyStatements().get(0);
assertTrue(stmt.sqlstatementtype == ESqlStatementType.sstl_raisestmt);
TRaiseStmt raiseStmt = (TRaiseStmt)stmt;
assertTrue(raiseStmt.getRaiseLevel() == ERaiseLevel.notice);
assertTrue(raiseStmt.getFormatString().toString().equalsIgnoreCase("'Quantity here is %'"));
assertTrue(raiseStmt.getExprList().getExpression(0).toString().equalsIgnoreCase("quantity"));
stmt = createFunction.getBodyStatements().get(1);
assertTrue(stmt.sqlstatementtype == ESqlStatementType.sst_assignstmt);
TAssignStmt assignStmt = (TAssignStmt)stmt;
assertTrue(assignStmt.getLeft().toString().equalsIgnoreCase("quantity"));
assertTrue(assignStmt.getExpression().toString().equalsIgnoreCase("50"));
stmt = createFunction.getBodyStatements().get(2);
assertTrue(stmt.sqlstatementtype == ESqlStatementType.sst_block_with_label);
TCommonBlock block = (TCommonBlock)stmt;
assertTrue(block.getDeclareStatements().size() == 1);
TVarDeclStmt declStmt = (TVarDeclStmt)block.getDeclareStatements().get(0);
assertTrue(declStmt.getDeclareType() == EDeclareType.variable);
assertTrue(declStmt.getElementName().toString().equalsIgnoreCase("quantity"));
assertTrue(declStmt.getDataType().getDataType() == EDataType.int_t);
assertTrue(declStmt.getDefaultValue().toString().equalsIgnoreCase("80"));
assertTrue(block.getBodyStatements().size() == 2);
stmt = block.getBodyStatements().get(1);
assertTrue(stmt.sqlstatementtype == ESqlStatementType.sstl_raisestmt);
raiseStmt = (TRaiseStmt)stmt;
assertTrue(raiseStmt.getRaiseLevel() == ERaiseLevel.notice);
assertTrue(raiseStmt.getFormatString().toString().equalsIgnoreCase("'Outer quantity here is %'"));
assertTrue(raiseStmt.getExprList().getExpression(0).toString().equalsIgnoreCase("outerblock.quantity"));
}
}