From 845d0a360b7f985c2dfd3bae2b6644eb45b5daec Mon Sep 17 00:00:00 2001 From: wenshao Date: Wed, 27 Jul 2016 10:39:17 +0800 Subject: [PATCH] add testcase for get head comment --- .../druid/bvt/sql/FirstCommentTest.java | 21 +++++++++++++++++++ 1 file changed, 21 insertions(+) create mode 100644 src/test/java/com/alibaba/druid/bvt/sql/FirstCommentTest.java 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)); + } +}