SpringDatabaseDataSourceConnection.java

import java.sql.Connection;
import java.sql.SQLException;
import javax.sql.DataSource;
import org.dbunit.database.DatabaseDataSourceConnection;
import org.springframework.jdbc.datasource.DataSourceUtils;

public class SpringDatabaseDataSourceConnection extends DatabaseDataSourceConnection {
    private final DataSource dataSource;

    public SpringDatabaseDataSourceConnection(DataSource dataSource) throws SQLException {
        super(dataSource);
        this.dataSource = dataSource;
    }

    public Connection getConnection() throws SQLException {
        Connection conn = DataSourceUtils.getConnection(dataSource);
        return new SpringConnection(dataSource, conn);
    }
}