Back

DBUnit

DBUnit is a nice tool to set the database in a predefined state. As I could not find a handsome selfcontained example on how to save and restore a database, I composed it from several found snippets. The code also tries to compare the just loaded table with the file it came from. This only seems to work properly if all columns in the file have at least one row with a value. If not, the assertEquals reports a column-count mismatch.

A copy of this file can be found at the dbunit.wikidot site.

package test;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;

import org.dbunit.Assertion;
import org.dbunit.DatabaseUnitException;
import org.dbunit.database.DatabaseConnection;
import org.dbunit.database.DatabaseSequenceFilter;
import org.dbunit.database.IDatabaseConnection;
import org.dbunit.dataset.DataSetException;
import org.dbunit.dataset.FilteredDataSet;
import org.dbunit.dataset.IDataSet;
import org.dbunit.dataset.ITable;
import org.dbunit.dataset.filter.ITableFilter;
import org.dbunit.dataset.xml.FlatXmlDataSet;
import org.dbunit.operation.DatabaseOperation;

import org.junit.Test;

public class DemoDBUnit {
	private static String _testDir        = "testData";
	private static String _dbFile         = "dbDump.xml";
	private static String _driverClass    = "org.apache.derby.jdbc.EmbeddedDriver";
	private static String _jdbcConnection = "jdbc:derby:testdb";
	private static String _testTableName  = "person";

	public static void main(String[] args) throws 	ClassNotFoundException,
							DatabaseUnitException, 
							IOException, 
							SQLException {

		fullDatabaseExport(new File(_testDir, _dbFile));
	}
	
	public static void fullDatabaseExport(File file) throws 	ClassNotFoundException,
									DatabaseUnitException,
									DataSetException,
									IOException, 
									SQLException {
		IDatabaseConnection connection = getConnection();

		ITableFilter filter = new DatabaseSequenceFilter(connection);
		IDataSet dataset    = new FilteredDataSet(filter, connection.createDataSet());
		FlatXmlDataSet.write(dataset, new FileOutputStream(file));
	}
	
	
	@Test
	public void test5_Test() throws ClassNotFoundException, 
					DatabaseUnitException, 
					IOException, 
					SQLException {
		fullDatabaseImport(new File(_testDir, _dbFile));
		ITable actualTable = getConnection().createDataSet().getTable(_testTableName);
	    
	    IDataSet expectedDataSet = new FlatXmlDataSet(new File(_testDir, _dbFile));
	    ITable expectedTable = expectedDataSet.getTable(_testTableName);
	    Assertion.assertEquals(expectedTable, actualTable);
	}
	
	public static void fullDatabaseImport(File file) throws 	ClassNotFoundException, 
									DatabaseUnitException, 
									IOException,
									SQLException {
		IDatabaseConnection connection = getConnection();
		IDataSet dataSet = new FlatXmlDataSet(file, true);

		DatabaseOperation.CLEAN_INSERT.execute(connection, dataSet);
	}
	
	public static IDatabaseConnection getConnection() throws 	ClassNotFoundException, 
									DatabaseUnitException, 
									SQLException {
		// database connection
		Class driverClass = Class.forName(_driverClass);
		Connection jdbcConnection = DriverManager.getConnection(_jdbcConnection);
		return new DatabaseConnection(jdbcConnection);
	}
}

Last updated: 2008-10-17

email