TestCase class we need to extend DatabaseTestCase class. This class provides two abstract methods "getConnection()" and "getDataSet()".

IDatabaseConnection getConnection()
throws Exception
protected IDataSet getDataSet() throws Exception.

Because of its being an abstract class we need to implement these two methods:

TestDbUnit.java :

...........................
...........................
// Provide a connection to the database
protected IDatabaseConnection
getConnection() throws Exception{
Class driverClass = Class.forName
("com.mysql.jdbc.Driver");
Connection jdbcConnection =
DriverManager.getConnection
("jdbc:mysql://localhost:3306/hrapptest",
"root", "root");
return new
DatabaseConnection(jdbcConnection);
}
// Load the data which will be inserted for
the test
protected IDataSet getDataSet() throws
Exception{ loadedDataSet =
new FlatXmlDataSet(this.getClass()
.getClassLoader().getResourceAsStream
("input.xml"));
return loadedDataSet;
}
............................
............................

4. Writing Test :

Now, write test to check that the data has been loaded in TestDbUnit.java file:

............................
............................
public void testCheckLoginDataLoaded()
throws Exception{
assertNotNull(loadedDataSet);
int rowCount = loadedDataSet.getTable
(TABLE_LOGIN).getRowCount();
assertEquals(2, rowCount);
}
............................
............................

Combining all of the above functionalities
into one TestDbUnit.java file, you will find it as follows :

TestDbUnit.java :package test;

import java.sql.Connection;
import java.sql.DriverManager;
import org.dbunit.DatabaseTestCase;
import org.dbunit.database.DatabaseConnection;
import org.dbunit.database.IDatabaseConnection;
import org.dbunit.dataset.IDataSet;
import org.dbunit.dataset.xml.FlatXmlDataSet;
public class TestDbUnit extends
DatabaseTestCase{
public static final String TABLE_LOGIN =