CreateDropTableTest.java

package org.nutz.dao.test.normal;

import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.SQLException;

import org.junit.Assert;

import org.junit.Test;
import org.nutz.dao.Sqls;
import org.nutz.dao.sql.Sql;
import org.nutz.dao.sql.SqlCallback;
import org.nutz.dao.test.DaoCase;

public class CreateDropTableTest extends DaoCase {

    private static String DESC_TABLE_SQL = "select column_name, is_nullable, column_default, extra from information_schema.`columns` where table_name = @tableName";

    @Test
    public void createTableTimestampFieldDefaultNullInMySQL() throws Exception {
        boolean isMySql = dao.meta().isMySql();
        // 这个仅仅测试MySQL数据库
        if (isMySql) {
            dao.create(TableWithTimestampInMySql.class, true);
            Sql descTable = Sqls.create(DESC_TABLE_SQL);
            descTable.params().set("tableName", "t_ts");
            descTable.setCallback(new SqlCallback() {
                public Object invoke(Connection conn, ResultSet rs, Sql sql) throws SQLException {
                    boolean ts1 = false;
                    boolean ts2 = false;
                    boolean ts3 = false;
                    boolean ts4 = false;
                    while (rs.next()) {
                        String columnName = rs.getString(1);
                        String isNullable = rs.getString(2);
                        String defaultValue = rs.getString(3);
                        String extra = rs.getString(4);
                        if ("ts1".equals(columnName)) {
                            if ((null == defaultValue || "NULL".equalsIgnoreCase(defaultValue))
                                && "YES".equalsIgnoreCase(isNullable)) {
                                ts1 = true;
                            }
                        }
                        if ("ts2".equals(columnName)) {
                            if (null != defaultValue
                                && "0000-00-00 00:00:00".equals(defaultValue)
                                && "YES".equalsIgnoreCase(isNullable)) {
                                ts2 = true;
                            }
                        }
                        if ("ts3".equals(columnName)) {
                            if (null != defaultValue
                                && "0000-00-00 00:00:00".equals(defaultValue)
                                && "NO".equalsIgnoreCase(isNullable)) {
                                ts3 = true;
                            }
                        }
                        if ("ts4".equals(columnName)) {
                            if (null != defaultValue
                                && "CURRENT_TIMESTAMP".equalsIgnoreCase(defaultValue)
                                && "NO".equalsIgnoreCase(isNullable)
                                && "ON UPDATE CURRENT_TIMESTAMP".equalsIgnoreCase(extra)) {
                                ts4 = true;
                            }
                        }
                    }
                    return ts1 && ts2 && ts3 && ts4;
                }
            });
            dao.execute(descTable);
            boolean isTrue = descTable.getObject(Boolean.class);
            Assert.assertTrue(isTrue);
        }
    }

    @Test
    public void createDropTableInOracle() throws Exception {
        boolean isOracle = dao.meta().isOracle();
        if (isOracle) {
            dao.create(CatAutoId.class, true);
            Assert.assertTrue(dao.drop(CatAutoId.class));

            dao.create(CatNotAutoId.class, true);
            Assert.assertTrue(dao.drop(CatNotAutoId.class));
        }
    }
}