improve wall filter

2023-03-04 19:17:16 +00:00 · 2013-07-30 22:37:11 +08:00
parent 63574c4702
commit 30fcdc8a4e
4 changed files with 216 additions and 79 deletions
@@ -0,0 +1,51 @@
+package com.alibaba.druid.bvt.filter.wall;
+
+import junit.framework.TestCase;
+
+import org.junit.Assert;
+
+import com.alibaba.druid.wall.WallSqlStat;
+import com.alibaba.druid.wall.WallTableStat;
+import com.alibaba.druid.wall.spi.MySqlWallProvider;
+
+
+public class WallProviderTest extends TestCase {
+    public void test_getSqlStat() throws Exception {
+        String whiteSql_1 = "select * from t1 where fid = 1";
+        String whiteSql_2 = "select * from t1 where fid = 2";
+        String blackSql_1 = "select * from t2 where fid = 1 OR 1 = 1";
+        String blackSql_2 = "select * from t2 where fid = 2 OR 1 = 1";
+        
+        MySqlWallProvider provider = new MySqlWallProvider();
+        
+        Assert.assertTrue(provider.checkValid(whiteSql_1));
+        Assert.assertTrue(provider.checkValid(whiteSql_2));
+        Assert.assertEquals(1, provider.getSqlList().size());
+        
+        Assert.assertFalse(provider.checkValid(blackSql_1));
+        Assert.assertFalse(provider.checkValid(blackSql_2));
+        Assert.assertEquals(2, provider.getSqlList().size());
+        
+        Assert.assertNotNull(provider.getSqlStat(whiteSql_1));
+        Assert.assertNotNull(provider.getSqlStat(blackSql_1));
+        
+        Assert.assertSame(provider.getSqlStat(blackSql_1), provider.getSqlStat(blackSql_2));
+        
+        WallSqlStat whiteStat_1 = provider.getSqlStat(whiteSql_1);
+        WallSqlStat whiteStat_2 = provider.getSqlStat(whiteSql_2);
+        Assert.assertSame(whiteStat_1, whiteStat_2);
+        provider.addFetchRowCount(whiteStat_1, 3);
+        WallTableStat wallTableStat = provider.getTableStat("t1");
+        Assert.assertNotNull(wallTableStat);
+        Assert.assertEquals(3, wallTableStat.getFetchRowCount());
+        
+        for (int i = 1000; i < 1000 * 2; ++i) {
+            String sql_x = "select * from t1 where fid = " + i;
+            Assert.assertTrue(provider.checkValid(sql_x));
+            Assert.assertSame(whiteStat_1, provider.getSqlStat(sql_x));
+        }
+        
+        Assert.assertEquals(500, provider.getWhiteList().size());
+        Assert.assertEquals(2, provider.getSqlList().size());
+    }
+}