Examples with WallConfig com.alibaba.druid.wall.WallConfig used on opensource projects

Example 76 with WallConfig

use of com.alibaba.druid.wall.WallConfig in project druid by alibaba.

the class BigSqlTest method test_true.

public void test_true() throws Exception {
    String sql = "SELECT c from sbtest where id=0";
    for (int i = 0; i < 10000; i++) {
        sql += " or id=0";
    }
    WallConfig config = new WallConfig();
    Assert.assertTrue(WallUtils.isValidateMySql(sql, config));
}

Example 77 with WallConfig

use of com.alibaba.druid.wall.WallConfig in project druid by alibaba.

the class BitwiseInvertTest method test_false.

public void test_false() throws Exception {
    WallConfig config = new WallConfig();
    config.setConditionOpBitwseAllow(false);
    Assert.assertFalse(//
    WallUtils.isValidateMySql("SELECT * from t where ~2", //
    config));
}

Example 78 with WallConfig

use of com.alibaba.druid.wall.WallConfig in project druid by alibaba.

the class BooleanXorTest method test_true.

public void test_true() throws Exception {
    WallConfig config = new WallConfig();
    config.setConditionOpXorAllow(true);
    Assert.assertTrue(//
    WallUtils.isValidateMySql("SELECT * from t where id = 1 XOR id = 2", //
    config));
}

Example 79 with WallConfig

use of com.alibaba.druid.wall.WallConfig in project druid by alibaba.

the class TenantSelectTest3 method testMySql.

public void testMySql() throws Exception {
    WallConfig config = new WallConfig();
    WallConfig config_callback = new WallConfig();
    config.setTenantTablePattern("*");
    config.setTenantColumn("tenant");
    config_callback.setTenantCallBack(new TenantTestCallBack());
    WallProvider.setTenantValue(123);
    MySqlWallProvider provider = new MySqlWallProvider(config);
    WallCheckResult checkResult = provider.check(sql);
    Assert.assertEquals(0, checkResult.getViolations().size());
    String resultSql = SQLUtils.toSQLString(checkResult.getStatementList(), JdbcConstants.MYSQL);
    Assert.assertEquals(expect_sql, resultSql);
}

Example 80 with WallConfig

use of com.alibaba.druid.wall.WallConfig in project druid by alibaba.

the class TenantSelectTest3 method testMySql2.

public void testMySql2() throws Exception {
    WallConfig config = new WallConfig();
    WallConfig config_callback = new WallConfig();
    config.setTenantTablePattern("*");
    config.setTenantColumn("tenant");
    config_callback.setTenantCallBack(new TenantTestCallBack());
    MySqlWallProvider provider = new MySqlWallProvider(config_callback);
    WallCheckResult checkResult = provider.check(sql);
    Assert.assertEquals(0, checkResult.getViolations().size());
    String resultSql = SQLUtils.toSQLString(checkResult.getStatementList(), JdbcConstants.MYSQL);
    Assert.assertEquals(expect_sql, resultSql);
}