增加针对 BoundSqlInterceptor 的单元测试

Author: abel533

src/test/java/com/github/pagehelper/test/basic/provider/TestBoundSqlInterceptor.java

@@ -0,0 +1,22 @@
+package com.github.pagehelper.test.basic.provider;
+
+import com.github.pagehelper.BoundSqlInterceptor;
+import com.github.pagehelper.util.MetaObjectUtil;
+import org.apache.ibatis.cache.CacheKey;
+import org.apache.ibatis.mapping.BoundSql;
+import org.apache.ibatis.reflection.MetaObject;
+
+public class TestBoundSqlInterceptor implements BoundSqlInterceptor {
+    public static final String COMMENT = "\n /* TestBoundSqlInterceptor */\n";
+
+    @Override
+    public BoundSql boundSql(Type type, BoundSql boundSql, CacheKey cacheKey, Chain chain) {
+        if (type == Type.ORIGINAL) {
+            String sql = boundSql.getSql();
+            MetaObject metaObject = MetaObjectUtil.forObject(boundSql);
+            metaObject.setValue("sql", sql + COMMENT);
+        }
+        return chain.doBoundSql(type, boundSql, cacheKey);
+    }
+
+}

src/test/java/com/github/pagehelper/test/basic/provider/TestProviderInteceptor.java

@@ -24,11 +24,15 @@
 
 package com.github.pagehelper.test.basic.provider;
 
+import com.github.pagehelper.BoundSqlInterceptor;
 import com.github.pagehelper.PageHelper;
 import com.github.pagehelper.mapper.ProviderMethod;
 import com.github.pagehelper.mapper.UserMapper;
 import com.github.pagehelper.util.MybatisInterceptorHelper;
+import org.apache.ibatis.cache.CacheKey;
+import org.apache.ibatis.mapping.BoundSql;
 import org.apache.ibatis.session.SqlSession;
+import org.junit.Assert;
 import org.junit.Test;
 
 import static org.junit.Assert.assertEquals;
@@ -40,7 +44,18 @@ public void testInterceptor() {
         SqlSession sqlSession = MybatisInterceptorHelper.getSqlSession();
         UserMapper userMapper = sqlSession.getMapper(UserMapper.class);
         try {
-            PageHelper.startPage(1, 10);
+            PageHelper.startPage(1, 10).boundSqlInterceptor(new BoundSqlInterceptor() {
+                @Override
+                public BoundSql boundSql(Type type, BoundSql boundSql, CacheKey cacheKey, Chain chain) {
+                    System.out.println("before: " + boundSql.getSql());
+                    BoundSql doBoundSql = chain.doBoundSql(type, boundSql, cacheKey);
+                    System.out.println("after: " + doBoundSql.getSql());
+                    if (type == Type.ORIGINAL) {
+                        Assert.assertTrue(doBoundSql.getSql().contains(TestBoundSqlInterceptor.COMMENT));
+                    }
+                    return doBoundSql;
+                }
+            });
             String str = "飞";
             userMapper.selectSimple(str);
             assertEquals(new ProviderMethod().selectSimple(str), SqlCache.get());

src/test/resources/hsqldb/mybatis-config-interceptor.xml

@@ -47,6 +47,8 @@
             <property name="helperDialect" value="mysql"/>
             <property name="supportMethodsArguments" value="true"/>
             <property name="rowBoundsWithCount" value="true"/>
+            <property name="boundSqlInterceptors"
+                      value="com.github.pagehelper.test.basic.provider.TestBoundSqlInterceptor,com.github.pagehelper.test.basic.provider.TestBoundSqlInterceptor"/>
         </plugin>
     </plugins>
 

src/test/resources/mysql/mybatis-config-interceptor.xml

@@ -47,6 +47,8 @@
             <property name="helperDialect" value="mysql"/>
             <property name="supportMethodsArguments" value="true"/>
             <property name="rowBoundsWithCount" value="true"/>
+            <property name="boundSqlInterceptors"
+                      value="com.github.pagehelper.test.basic.provider.TestBoundSqlInterceptor,com.github.pagehelper.test.basic.provider.TestBoundSqlInterceptor"/>
         </plugin>
     </plugins>