Class InsertTest

src/test/java/com/sql/parse/statement/druid/InsertTest.java:17–190

@date 2019年4月30日上午09:58:02 @desc Insert测试类

 * @desc Insert测试类
 */
public class InsertTest {
    private static final Logger logger = LogManager.getLogger(InsertTest.class);

    private String table;
    private String column1;
    private String column2;
    private String column3;
    private String value1;
    private String value2;
    private String value3;
    private String simpleSql;
    private String simpleSql2;
    private String simpleSql3;

    @Before
    public void before() {
        // TODO Auto-generated method stub
        table = "table1";
        column1 = "column1";
        column2 = "column2";
        column3 = "column3";
        value1 = "1";
        value2 = "'2'";
        value3 = "3";
        simpleSql = "insert into " + table + " values (1)";
        simpleSql2 = "insert into " + table + "(pk) values (1)";
        simpleSql3 = "insert into " + table + " select * from table2";
    }

    @Test
    public void testParse() {
        try {
            Insert insert = Insert.parse(simpleSql);
            Assert.assertEquals(
                    SqlFormatter.format(insert.getSql()),
                    SqlFormatter.format("INSERT INTO table1 VALUES (1)")
            );
            Assert.assertEquals(
                    SqlFormatter.format(insert.getSql(false)),
                    SqlFormatter.format("INSERT INTO table1 VALUES (1)")
            );
        } catch (Exception e) {
            Assert.fail(e.getMessage());
        }
    }

    /**
     * @desc 简单的新增：INSERT INTO table VALUES (1, '2', 3)
     */
    @Test
    public void testInsertWithValueColumn() {
        try {
            Insert insert = Insert.parse(simpleSql);
            insert.addValueColumn(value1).addValueColumn(value2).addValueColumn(value3);
            Assert.assertEquals(
                    SqlFormatter.format(insert.getSql()),
                    SqlFormatter.format("INSERT INTO table1 VALUES (1, 1, '2', 3)")
            );