package com.alipay.zdal.test.ut.sqlparser.oracle; import java.util.List; import org.junit.Test; import junit.framework.Assert; import com.alipay.zdal.parser.sql.ast.SQLStatement; import com.alipay.zdal.parser.sql.dialect.oracle.parser.OracleStatementParser; import com.alipay.zdal.parser.sql.dialect.oracle.visitor.OracleSchemaStatVisitor; public class OracleBlockTest4 { @Test public void test_0() throws Exception { String sql = "DECLARE" + // " done BOOLEAN;" + // "BEGIN" + // " FOR i IN 1..50 LOOP" + // " IF done THEN" + // " GOTO end_loop;" + // " END IF;" + // " <<end_loop>>" + // " END LOOP;" + // "END;"; OracleStatementParser parser = new OracleStatementParser(sql); List<SQLStatement> statementList = parser.parseStatementList(); SQLStatement statemen = statementList.get(0); Assert.assertEquals(1, statementList.size()); OracleSchemaStatVisitor visitor = new OracleSchemaStatVisitor(); statemen.accept(visitor); Assert.assertEquals(0, visitor.getTables().size()); // Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("departments"))); // Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("employees"))); Assert.assertEquals(0, visitor.getColumns().size()); // Assert.assertTrue(visitor.getColumns().contains(new TableStat.Column("departments", "department_id"))); } }