com.github.hippoom.xunittemplate.springtestdbunit.dataset.CreateTemplateDataSet Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spring-test-dbunit-template Show documentation

An extension to simplify CRUD test with spring-test-dbunit

There is a newer version: 0.1.1

Show newest version

package com.github.hippoom.xunittemplate.springtestdbunit.dataset;

import org.dbunit.dataset.*;

public class CreateTemplateDataSet extends AbstractDataSet {
    private final IDataSet _dataSet;
    private String column;
    private Object value;

    public CreateTemplateDataSet(IDataSet _dataSet, String column, Object value) {
        this._dataSet = _dataSet;
        this.column = column;
        this.value = value;
    }

    @Override
    protected ITableIterator createIterator(boolean reversed) throws DataSetException {
        return reversed ? _dataSet.reverseIterator() : _dataSet.iterator();
    }

    @Override
    public ITable getTable(String tableName) throws DataSetException {
        return new CreateTemplateTable(_dataSet.getTable(tableName), column, value);
    }

}