git-svn-id: http://code.alibabatech.com/svn/druid/trunk@129 b9813039-fb51-4c41-a8b9-e21c2acb5095

2023-03-04 19:17:16 +00:00 · 2011-05-19 01:55:04 +00:00
parent b98c5763a4
commit b81d40fa73
1 changed files with 52 additions and 0 deletions
@@ -0,0 +1,52 @@
+package com.alibaba.druid.sql.oracle.bvt;
+
+import java.util.List;
+
+import junit.framework.Assert;
+import junit.framework.TestCase;
+
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.ast.expr.SQLBinaryOpExpr;
+import com.alibaba.druid.sql.ast.expr.SQLBinaryOperator;
+import com.alibaba.druid.sql.ast.expr.SQLIdentifierExpr;
+import com.alibaba.druid.sql.dialect.oracle.ast.stmt.OracleUpdateSetListClause;
+import com.alibaba.druid.sql.dialect.oracle.ast.stmt.OracleUpdateSetListSingleColumnItem;
+import com.alibaba.druid.sql.dialect.oracle.ast.stmt.OracleUpdateStatement;
+import com.alibaba.druid.sql.dialect.oracle.ast.visitor.OracleOutputVisitor;
+import com.alibaba.druid.sql.dialect.oracle.parser.OracleStatementParser;
+
+public class OracleSQLParserTest extends TestCase {
+
+    public void test_0() throws Exception {
+        String sql = "UPDATE employees SET salary = salary * 1.1 WHERE employee_id IN (SELECT employee_id FROM job_history);";
+        OracleStatementParser parser = new OracleStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        
+        Assert.assertEquals(1, statementList.size());
+        
+        OracleUpdateStatement statement = (OracleUpdateStatement) statementList.get(0);
+        Assert.assertEquals(0, statement.getHints().size());
+        OracleUpdateSetListClause setList = (OracleUpdateSetListClause) statement.getSetClause();
+        Assert.assertEquals(1, setList.getItems().size());
+        OracleUpdateSetListSingleColumnItem item = (OracleUpdateSetListSingleColumnItem) setList.getItems().get(0);
+        SQLIdentifierExpr column = (SQLIdentifierExpr) item.getColumn();
+        Assert.assertEquals("salary", column.getName());
+        
+        SQLBinaryOpExpr value = (SQLBinaryOpExpr) item.getValue();
+        Assert.assertEquals(SQLBinaryOperator.Multiply, value.getOperator());
+
+        output(statementList);
+    }
+
+    private void output(List<SQLStatement> stmtList) {
+        StringBuilder out = new StringBuilder();
+        OracleOutputVisitor visitor = new OracleOutputVisitor(out);
+
+        for (SQLStatement stmt : stmtList) {
+            stmt.accept(visitor);
+            visitor.println();
+        }
+
+        System.out.println(out.toString());
+    }
+}