From 9ba1788752d7db26d841c3adc6528097cd7198d2 Mon Sep 17 00:00:00 2001 From: wenshao Date: Sat, 15 Oct 2011 10:18:44 +0000 Subject: [PATCH] sample clause git-svn-id: http://code.alibabatech.com/svn/druid/trunk@982 b9813039-fb51-4c41-a8b9-e21c2acb5095 --- .../oracle/visitor/OracleOutputVisitor.java | 2 +- .../sql/oracle/OracleSampleClauseTest.java | 42 +++++++++++++++++++ 2 files changed, 43 insertions(+), 1 deletion(-) create mode 100644 src/test/java/com/alibaba/druid/bvt/sql/oracle/OracleSampleClauseTest.java diff --git a/src/main/java/com/alibaba/druid/sql/dialect/oracle/visitor/OracleOutputVisitor.java b/src/main/java/com/alibaba/druid/sql/dialect/oracle/visitor/OracleOutputVisitor.java index b42c29cee6..c03b5a77d6 100644 --- a/src/main/java/com/alibaba/druid/sql/dialect/oracle/visitor/OracleOutputVisitor.java +++ b/src/main/java/com/alibaba/druid/sql/dialect/oracle/visitor/OracleOutputVisitor.java @@ -541,7 +541,7 @@ public class OracleOutputVisitor extends SQLASTOutputVisitor implements OracleAS } if (x.getSampleClause() != null) { - print(" "); + println(); x.getSampleClause().accept(this); } diff --git a/src/test/java/com/alibaba/druid/bvt/sql/oracle/OracleSampleClauseTest.java b/src/test/java/com/alibaba/druid/bvt/sql/oracle/OracleSampleClauseTest.java new file mode 100644 index 0000000000..534a829dac --- /dev/null +++ b/src/test/java/com/alibaba/druid/bvt/sql/oracle/OracleSampleClauseTest.java @@ -0,0 +1,42 @@ +package com.alibaba.druid.bvt.sql.oracle; + +import junit.framework.Assert; +import junit.framework.TestCase; + +import com.alibaba.druid.sql.ast.statement.SQLSelectStatement; +import com.alibaba.druid.sql.dialect.oracle.parser.OracleStatementParser; +import com.alibaba.druid.sql.test.TestUtils; + +public class OracleSampleClauseTest extends TestCase { + + public void test_0() throws Exception { + String sql = "SELECT COUNT(*) * 10 FROM orders SAMPLE (10);"; + + String expected = "SELECT COUNT(*) * 10\n" + "FROM orders\n" + "SAMPLE (10);\n"; + + OracleStatementParser parser = new OracleStatementParser(sql); + SQLSelectStatement stmt = (SQLSelectStatement) parser.parseStatementList().get(0); + + String text = TestUtils.outputOracle(stmt); + + Assert.assertEquals(expected, text); + + System.out.println(text); + } + + public void test_1() throws Exception { + String sql = "SELECT COUNT(*) * 10 FROM orders SAMPLE (10) SEED (1);"; + + String expected = "SELECT COUNT(*) * 10\n" + "FROM orders\n" + "SAMPLE (10) SEED (1);\n"; + + OracleStatementParser parser = new OracleStatementParser(sql); + SQLSelectStatement stmt = (SQLSelectStatement) parser.parseStatementList().get(0); + + String text = TestUtils.outputOracle(stmt); + + Assert.assertEquals(expected, text); + + System.out.println(text); + } + +}