You can certainly use DBUnit with JUnit 4.x. I've tried to whip up an example using vanilla JDBC leaving the Spring and Hibernate to you. Although, I highly recommend giving this open source utility a try to make your Spring and DBUnit experience much more enjoyable. Note, I haven't contributed to that utility, just used it.
Another DBUnit challenge is that it does not create tables for you, so an in-memory database can give you issues, take a look at this other stackoverflow post for more details there.
I'm using DBUnit 2.4.8, HSQLDB 1.8.0.10, and JUnit 4.10.
import static org.junit.Assert.assertEquals;
import java.io.StringReader;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import org.dbunit.IDatabaseTester;
import org.dbunit.JdbcDatabaseTester;
import org.dbunit.dataset.IDataSet;
import org.dbunit.dataset.xml.FlatXmlDataSetBuilder;
import org.dbunit.operation.DatabaseOperation;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
public class DBUnitTester {
private IDatabaseTester databaseTester;
@Before
public void init() throws Exception {
databaseTester = new JdbcDatabaseTester(org.hsqldb.jdbcDriver.class.getName(), "jdbc:hsqldb:sample", "sa", "");
createTablesSinceDbUnitDoesNot(databaseTester.getConnection().getConnection());
String inputXml =
"<dataset>" +
" <TEST_TABLE COL0=\"row 0 col 0\" " +
" COL1=\"row 0 col 1\"" +
" COL2=\"row 0 col 2\"/> " +
"</dataset>";
IDataSet dataSet = new FlatXmlDataSetBuilder().build(new StringReader(inputXml));
databaseTester.setDataSet(dataSet);
databaseTester.setSetUpOperation(DatabaseOperation.CLEAN_INSERT);
databaseTester.setTearDownOperation(DatabaseOperation.DELETE_ALL);
databaseTester.onSetup();
}
private void createTablesSinceDbUnitDoesNot(Connection connection) throws SQLException {
PreparedStatement statement = connection.prepareStatement("CREATE TABLE TEST_TABLE(COL0 VARCHAR(20), COL1 VARCHAR(20), COL2 VARCHAR(20))");
statement.execute();
statement.close();
}
@Test
public void firstTest() throws SQLException, Exception {
PreparedStatement statement = databaseTester.getConnection().getConnection().prepareStatement("SELECT COL0 FROM TEST_TABLE");
ResultSet rs = statement.executeQuery();
rs.next();
assertEquals("row 0 col 0", rs.getString("COL0"));
}
@After
public void cleanUp() throws Exception {
databaseTester.onTearDown();
}
}