DB2FunctionTestSetup.java

package org.geotools.data.db2;

import org.geotools.jdbc.JDBCDataStore;
import org.geotools.jdbc.SQLDialect;

/**
 * 
 *
 * @source $URL: http://svn.osgeo.org/geotools/trunk/modules/plugin/jdbc/jdbc-db2/src/test/java/org/geotools/data/db2/DB2FunctionTestSetup.java $
 */
public class DB2FunctionTestSetup extends DB2TestSetup {
    
    protected void setUpDataStore(JDBCDataStore dataStore) {
        super.setUpDataStore(dataStore);
        
        // the unit tests needs function encoding enabled to actually test that
        SQLDialect dialect = dataStore.getSQLDialect();
        if (dialect instanceof DB2SQLDialectBasic)
            ((DB2SQLDialectBasic) dialect).setFunctionEncodingEnabled(true);
        if (dialect instanceof DB2SQLDialectPrepared)
            ((DB2SQLDialectPrepared) dialect).setFunctionEncodingEnabled(true);
    }

}