diff --git a/src/test/java/com/alibaba/druid/sql/mysql/bvt/MiscellaneousFunctionsTest.java b/src/test/java/com/alibaba/druid/sql/mysql/bvt/MiscellaneousFunctionsTest.java index d43dbde9b8..cb79f67e33 100644 --- a/src/test/java/com/alibaba/druid/sql/mysql/bvt/MiscellaneousFunctionsTest.java +++ b/src/test/java/com/alibaba/druid/sql/mysql/bvt/MiscellaneousFunctionsTest.java @@ -1,83 +1,78 @@ /* - * Copyright 2011 Alibaba Group. - * - * Licensed under the Apache License, Version 2.0 (the "License"); - * you may not use this file except in compliance with the License. - * You may obtain a copy of the License at - * - * http://www.apache.org/licenses/LICENSE-2.0 - * - * Unless required by applicable law or agreed to in writing, software - * distributed under the License is distributed on an "AS IS" BASIS, - * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. - * See the License for the specific language governing permissions and - * limitations under the License. + * Copyright 2011 Alibaba Group. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this + * file except in compliance with the License. You may obtain a copy of the License at + * http://www.apache.org/licenses/LICENSE-2.0 Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, + * either express or implied. See the License for the specific language governing permissions and limitations under the + * License. */ -package com.alibaba.druid.sql.mysql.bvt; - -import java.util.List; - -import com.alibaba.druid.sql.ast.SQLStatement; -import com.alibaba.druid.sql.dialect.mysql.parser.MySqlStatementParser; -import com.alibaba.druid.sql.dialect.mysql.visitor.MySqlOutputVisitor; -import com.alibaba.druid.sql.parser.SQLStatementParser; - -import junit.framework.Assert; -import junit.framework.TestCase; - -public class MiscellaneousFunctionsTest extends TestCase { - public void test_0() throws Exception { - String sql = "UPDATE t SET i = DEFAULT(i)+1 WHERE id < 100;"; - - SQLStatementParser parser = new MySqlStatementParser(sql); - List stmtList = parser.parseStatementList(); - - String text = output(stmtList); - - Assert.assertEquals("UPDATE t SET i = DEFAULT(i) + 1 WHERE id < 100;", text); - } - - public void test_1() throws Exception { - String sql = "SELECT GET_LOCK('lock1',10);"; - - SQLStatementParser parser = new MySqlStatementParser(sql); - List stmtList = parser.parseStatementList(); - - String text = output(stmtList); - - Assert.assertEquals("SELECT GET_LOCK('lock1', 10);", text); - } - - public void test_2() throws Exception { - String sql = "SELECT INET_ATON('209.207.224.40');"; - - SQLStatementParser parser = new MySqlStatementParser(sql); - List stmtList = parser.parseStatementList(); - - String text = output(stmtList); - - Assert.assertEquals("SELECT INET_ATON('209.207.224.40');", text); - } - - public void test_3() throws Exception { - String sql = "SELECT UUID();"; - - SQLStatementParser parser = new MySqlStatementParser(sql); - List stmtList = parser.parseStatementList(); - - String text = output(stmtList); - - Assert.assertEquals("SELECT UUID();", text); - } - - private String output(List stmtList) { - StringBuilder out = new StringBuilder(); - - for (SQLStatement stmt : stmtList) { - stmt.accept(new MySqlOutputVisitor(out)); - out.append(";"); - } - - return out.toString(); - } -} +package com.alibaba.druid.sql.mysql.bvt; + +import java.util.List; + +import com.alibaba.druid.sql.ast.SQLStatement; +import com.alibaba.druid.sql.dialect.mysql.parser.MySqlStatementParser; +import com.alibaba.druid.sql.dialect.mysql.visitor.MySqlOutputVisitor; +import com.alibaba.druid.sql.parser.SQLStatementParser; + +import junit.framework.Assert; +import junit.framework.TestCase; + +public class MiscellaneousFunctionsTest extends TestCase { + + public void test_0() throws Exception { + String sql = "UPDATE t SET i = DEFAULT(i)+1 WHERE id < 100;"; + + SQLStatementParser parser = new MySqlStatementParser(sql); + List stmtList = parser.parseStatementList(); + + String text = output(stmtList); + + Assert.assertEquals("UPDATE t SET i = DEFAULT(i) + 1 WHERE id < 100;", text); + } + + public void test_1() throws Exception { + String sql = "SELECT GET_LOCK('lock1',10);"; + + SQLStatementParser parser = new MySqlStatementParser(sql); + List stmtList = parser.parseStatementList(); + + String text = output(stmtList); + + Assert.assertEquals("SELECT GET_LOCK('lock1', 10);", text); + } + + public void test_2() throws Exception { + String sql = "SELECT INET_ATON('209.207.224.40');"; + + SQLStatementParser parser = new MySqlStatementParser(sql); + List stmtList = parser.parseStatementList(); + + String text = output(stmtList); + + Assert.assertEquals("SELECT INET_ATON('209.207.224.40');", text); + } + + public void test_3() throws Exception { + String sql = "SELECT UUID();"; + + SQLStatementParser parser = new MySqlStatementParser(sql); + List stmtList = parser.parseStatementList(); + + String text = output(stmtList); + + Assert.assertEquals("SELECT UUID();", text); + } + + private String output(List stmtList) { + StringBuilder out = new StringBuilder(); + MySqlOutputVisitor visitor = new MySqlOutputVisitor(out); + + for (SQLStatement stmt : stmtList) { + stmt.accept(visitor); + out.append(";"); + } + + return out.toString(); + } +}