Search in sources :

Example 1 with MsSqlDataTypeFactory

use of org.dbunit.ext.mssql.MsSqlDataTypeFactory in project sharding-jdbc by shardingjdbc.

the class ShardingJdbcDatabaseTester method getConnection.

@Override
public IDatabaseConnection getConnection() throws Exception {
    IDatabaseConnection result = super.getConnection();
    DatabaseConfig dbConfig = result.getConfig();
    dbConfig.setProperty(DatabaseConfig.FEATURE_CASE_SENSITIVE_TABLE_NAMES, false);
    dbConfig.setProperty(DatabaseConfig.FEATURE_DATATYPE_WARNING, false);
    switch(driverClass) {
        case "org.h2.Driver":
            dbConfig.setProperty(DatabaseConfig.PROPERTY_DATATYPE_FACTORY, new H2DataTypeFactory());
            break;
        case "com.mysql.jdbc.Driver":
            dbConfig.setProperty(DatabaseConfig.PROPERTY_DATATYPE_FACTORY, new MySqlDataTypeFactory());
            break;
        case "org.postgresql.Driver":
            dbConfig.setProperty(DatabaseConfig.PROPERTY_DATATYPE_FACTORY, new PostgresqlDataTypeFactory());
            break;
        case "oracle.jdbc.driver.OracleDriver":
            dbConfig.setProperty(DatabaseConfig.PROPERTY_DATATYPE_FACTORY, new Oracle10DataTypeFactory());
            break;
        case "com.microsoft.sqlserver.jdbc.SQLServerDriver":
            dbConfig.setProperty(DatabaseConfig.PROPERTY_DATATYPE_FACTORY, new MsSqlDataTypeFactory());
            break;
        default:
            throw new ShardingJdbcException("Unsupported JDBC driver '%s'", driverClass);
    }
    return result;
}
Also used : PostgresqlDataTypeFactory(org.dbunit.ext.postgresql.PostgresqlDataTypeFactory) MySqlDataTypeFactory(org.dbunit.ext.mysql.MySqlDataTypeFactory) Oracle10DataTypeFactory(org.dbunit.ext.oracle.Oracle10DataTypeFactory) ShardingJdbcException(io.shardingjdbc.core.exception.ShardingJdbcException) MsSqlDataTypeFactory(org.dbunit.ext.mssql.MsSqlDataTypeFactory) IDatabaseConnection(org.dbunit.database.IDatabaseConnection) H2DataTypeFactory(org.dbunit.ext.h2.H2DataTypeFactory) DatabaseConfig(org.dbunit.database.DatabaseConfig)

Aggregations

ShardingJdbcException (io.shardingjdbc.core.exception.ShardingJdbcException)1 DatabaseConfig (org.dbunit.database.DatabaseConfig)1 IDatabaseConnection (org.dbunit.database.IDatabaseConnection)1 H2DataTypeFactory (org.dbunit.ext.h2.H2DataTypeFactory)1 MsSqlDataTypeFactory (org.dbunit.ext.mssql.MsSqlDataTypeFactory)1 MySqlDataTypeFactory (org.dbunit.ext.mysql.MySqlDataTypeFactory)1 Oracle10DataTypeFactory (org.dbunit.ext.oracle.Oracle10DataTypeFactory)1 PostgresqlDataTypeFactory (org.dbunit.ext.postgresql.PostgresqlDataTypeFactory)1