diff --git a/src/test/java/com/alibaba/druid/bvt/filter/wall/MySqlWallTest43.java b/src/test/java/com/alibaba/druid/bvt/filter/wall/MySqlWallTest43.java index 2bd2723367..aeac2c5801 100644 --- a/src/test/java/com/alibaba/druid/bvt/filter/wall/MySqlWallTest43.java +++ b/src/test/java/com/alibaba/druid/bvt/filter/wall/MySqlWallTest43.java @@ -36,19 +36,40 @@ public class MySqlWallTest43 extends TestCase { Assert.assertFalse(provider.checkValid(// "SELECT COUNT(1) AS count FROM `team` " + // - "WHERE `team_type` = 'normal' AND 1 = 1 AND `city_id` IN (0,10)")); + "WHERE `team_type` = 'normal' AND 1 = 1 AND `city_id` IN (0,10)")); Assert.assertEquals(1, provider.getTableStats().size()); } - + public void test_true() throws Exception { WallProvider provider = new MySqlWallProvider(); provider.getConfig().setConditionAndAlwayTrueAllow(true); - + Assert.assertTrue(provider.checkValid(// - "SELECT COUNT(1) AS count FROM `team` " + // + "SELECT COUNT(1) AS count FROM `team` " + // "WHERE `team_type` = 'normal' AND 1 = 1 AND `city_id` IN (0,10)")); - + + Assert.assertEquals(1, provider.getTableStats().size()); + } + + public void test_false2() throws Exception { + WallProvider provider = new MySqlWallProvider(); + + Assert.assertFalse(provider.checkValid(// + "SELECT COUNT(1) AS count FROM `team` " + // + "WHERE `team_type` = 'normal' AND 1 = 2 AND `city_id` IN (0,10)")); + + Assert.assertEquals(1, provider.getTableStats().size()); + } + + public void test_true2() throws Exception { + WallProvider provider = new MySqlWallProvider(); + provider.getConfig().setConditionAndAlwayFalseAllow(true); + + Assert.assertTrue(provider.checkValid(// + "SELECT COUNT(1) AS count FROM `team` " + // + "WHERE `team_type` = 'normal' AND 1 = 2 AND `city_id` IN (0,10)")); + Assert.assertEquals(1, provider.getTableStats().size()); } }