package com.alibaba.druid.bvt.sql.mysql.param;
import com.alibaba.druid.sql.ast.SQLStatement;
import com.alibaba.druid.sql.dialect.mysql.visitor.MySqlExportParameterVisitor;
import com.alibaba.druid.sql.parser.SQLParserUtils;
import com.alibaba.druid.sql.parser.SQLStatementParser;
import com.alibaba.druid.sql.visitor.ExportParameterVisitor;
import com.alibaba.druid.sql.visitor.ParameterizedOutputVisitorUtils;
import com.alibaba.druid.util.JdbcConstants;
import junit.framework.TestCase;
import java.util.List;
/**
* Created by wenshao on 16/8/23.
*/
public class MySqlParameterizedOutputVisitorTest_16 extends TestCase {
public void test_for_parameterize() throws Exception {
final String dbType = JdbcConstants.MYSQL;
String sql = "/* 0bfacfa414829200086238910e/0.3// */" +
"insert into `t1` (" +
" `f0`, `f1`, `f2`, `f3`, `f4`, " +
"`f5`, `f6`, `f7`, `f8`, `f9`, " +
"`destination`, `start_standard`, `start_fee`, `add_standard`, `add_fee`, " +
"`region_fee_standard`, `region_fee_add`, `cell_fee`, `way_day`, `version`)" +
" values ( 1, 2, 2, 3, 0, -4, 1, null, '2016-12-28 18:13:28.825', '2016-12-28 18:13:28.825', 1, 1, 0, 1, 0, null, null, null, null, 0)\n";
String psql = ParameterizedOutputVisitorUtils.parameterize(sql, dbType);
assertEquals("INSERT INTO `t1` (`f0`, `f1`, `f2`, `f3`, `f4`\n" +
"\t, `f5`, `f6`, `f7`, `f8`, `f9`\n" +
"\t, `destination`, `start_standard`, `start_fee`, `add_standard`, `add_fee`\n" +
"\t, `region_fee_standard`, `region_fee_add`, `cell_fee`, `way_day`, `version`)\n" +
"VALUES (?, ?, ?, ?, ?\n" +
"\t\t, ?, ?, ?, ?, ?\n" +
"\t\t, ?, ?, ?, ?, ?\n" +
"\t\t, ?, ?, ?, ?, ?)", psql);
SQLStatementParser parser = SQLParserUtils.createSQLStatementParser(sql, dbType);
List<SQLStatement> stmtList = parser.parseStatementList();
StringBuilder out = new StringBuilder();
ExportParameterVisitor visitor = new MySqlExportParameterVisitor(out);
for (SQLStatement stmt : stmtList) {
stmt.accept(visitor);
}
assertEquals(20, visitor.getParameters().size());
}
}