diff --git a/src/test/java/com/alibaba/druid/bvt/sql/FirstCommentTest.java b/src/test/java/com/alibaba/druid/bvt/sql/FirstCommentTest.java new file mode 100644 index 0000000000..a72381b5a3 --- /dev/null +++ b/src/test/java/com/alibaba/druid/bvt/sql/FirstCommentTest.java @@ -0,0 +1,21 @@ +package com.alibaba.druid.bvt.sql; + +import java.util.List; + +import org.junit.Assert; + +import com.alibaba.druid.sql.dialect.mysql.parser.MySqlLexer; +import com.alibaba.druid.sql.parser.Lexer; + +import junit.framework.TestCase; + +public class FirstCommentTest extends TestCase { + public void test_first_commnet() throws Exception { + String sql = "/*test sql*/select age from user where name = 'xx';"; + Lexer lexer = new MySqlLexer(sql); + lexer.setKeepComments(true); + lexer.nextToken(); + List comments = lexer.readAndResetComments(); + Assert.assertEquals("/*test sql*/", comments.get(0)); + } +}