ColumnSensingReplacementDataSetLoader.java

package net.petrikainulainen.springdata.jpa.web;

import com.github.springtestdbunit.dataset.FlatXmlDataSetLoader;
import org.dbunit.dataset.IDataSet;
import org.dbunit.dataset.ReplacementDataSet;
import org.springframework.core.io.Resource;
/**
 * This class is a custom DbUnit data set loader that support flat XML data sets. This data set loader
 * adds support for the extra features:
 * <ul>
 * <li>You can use the column sensing feature of DbUnit.</li>
 * <li>You can specify that a column's value is null by using the string [null].</li>
 * </ul>
 * @author Petri Kainulainen
 */
public class ColumnSensingReplacementDataSetLoader extends FlatXmlDataSetLoader {

    @Override
    protected IDataSet createDataSet(Resource resource) throws Exception {
        return createReplacementDataSet(super.createDataSet(resource));
    }
    private ReplacementDataSet createReplacementDataSet(IDataSet dataSet) {
        ReplacementDataSet replacementDataSet = new ReplacementDataSet(dataSet);
        replacementDataSet.addReplacementObject("[null]", null);
        return replacementDataSet;
    }
}