dbunit使用

使用dbunit，可以帮助我们在测试中维护数据，也可以辅助我们的测试。

首先当然是下载dbunit, http://dbunit.sourceforge.net

我测试用的是 MYSQL 5.0 。

建立数据库：
create table test1(
id int not null auto_increment,
user_name varchar(50),
primary key(id)) engine=innodb;


保存数据的xml文件：

<dataset>
    <test1 user_name="tom"/>             
    <test1 user_name="John"/>
    <test1 user_name="Rose"/>  
</dataset>

首先建立一个 JunitTest 的类：

public class Test2 extends TestCase {

	protected void setUp() throws Exception {

         }

	protected void tearDown() throws Exception {

         }

}

我不喜欢继承dbunit的类，所以我们在JunitTest 的类里增加这个变量：

public class Test2 extends TestCase {

	private IDatabaseTester databaseTester;

	protected void setUp() throws Exception {

         }

	protected void tearDown() throws Exception {

         }

}

然后，我们可以该写 setUp() 方法了，无非就是连接数据库，把数据倒入到表里。

	protected void setUp() throws Exception {

		databaseTester = new JdbcDatabaseTester("com.mysql.jdbc.Driver",
	            "jdbc:mysql://127.0.0.1:3306/test", "root", "123");

		IDataSet dataSet = getDataSet();

		databaseTester.setDataSet( dataSet );
		databaseTester.onSetup();


	}

	protected IDataSet getDataSet() throws Exception
    {

        return new FlatXmlDataSet(new FileInputStream(new File("dataset.xml")));
    }

然后是 tearDown 方法

	protected void tearDown() throws Exception
    {

		databaseTester.setTearDownOperation(DatabaseOperation.DELETE_ALL);
        databaseTester.onTearDown();
    }

好了，准备工作完成了，下面开始写测试方法。

	public void test1() throws Exception{

		ITable test1Table = databaseTester.getDataSet().getTable("test1");
		assertEquals(test1Table.getRowCount(), 3);

	}

这个方法就是测试 test1 表有多少条记录

	public void test2() throws Exception{

		ITable test1Table = databaseTester.getConnection().createQueryTable("any_name",
				"select user_name from test1 where user_name='tom'");

		assertTrue(test1Table.getRowCount()==1);

	}

这个方法测试是否有 user_name='tom' 这个记录.


那么自动增加字段怎么办呢？
只要在 tearDown 里增加

databaseTester.setTearDownOperation(DatabaseOperation.TRUNCATE_TABLE);

就可以了！