@Test
public void testCreateConnection() throws DatabaseUnitException {
    // GIVEN
    final String schema = "foo";

    // WHEN
    final IDatabaseConnection dbConnection = FACTORY.createConnection(connection, schema);

    // THEN
    assertThat(dbConnection, notNullValue());

    final Object typeFactory = dbConnection.getConfig().getProperty(DatabaseConfig.PROPERTY_DATATYPE_FACTORY);
    assertThat(typeFactory, notNullValue());
    assertThat(typeFactory.getClass(), equalTo(MySqlDataTypeFactory.class));

    final Object metadataHandler = dbConnection.getConfig().getProperty(DatabaseConfig.PROPERTY_METADATA_HANDLER);
    assertThat(metadataHandler, notNullValue());
    assertThat(metadataHandler.getClass(), equalTo(MySqlMetadataHandler.class));

    assertThat(dbConnection.getSchema(), equalTo(schema));
}
 

private IMetadataHandler getMetadataHandler(DBType dbType) {
    switch (dbType) {
        case MYSQL:
            return new MySqlMetadataHandler();
        case DB2:
            return new Db2MetadataHandler();
        default:
            return null;
    }
}