基于Dbunit的数据库层数据验证测试

DbUnit测试基本概念和流程

基于DbUnit的测试的主要接口是IDataSet。IDataSet代表一个或多个表的数据。

可以将数据库模式的全部内容表示为单个IDataSet实例。这些表本身由Itable实例来表示。

IDataSet的实现有很多，每一个都对应一个不同的数据源或加载机制。最常用的几种IDataSet实现为：

FlatXmlDataSet：数据的简单平面文件XML表示

QueryDataSet：用SQL查询获得的数据

DatabaseDataSet：数据库表本身内容的一种表示

XlsDataSet：数据的excel表示

一般而言，使用DbUnit进行单元测试的流程如下：

1根据业务，做好测试用的准备数据和预想结果数据，通常准备成xml格式文件。

2在setUp()方法里边备份数据库中的关联表。

3在setUp()方法里边读入准备数据。

4对测试类的对应测试方法进行实装:执行对象方法，把数据库的实际执行结果和预想结果进行比较。

5在tearDown()方法里边,把数据库还原到测试前状态。

以上内容引用：http://blog.csdn.net/yuezu1026/article/details/4055923

感觉讲的非常通俗易懂。

示例 ：

在 开始之前需要下载 dbunit-2.4.7.jar,slf4j-log4j12-1.7.2,log4j-1.2.17,slf4j-api-1.7.2.jar,mysql-connector-java-5.1.17-bin.jar,testng.jar六个jar包，这是我示例中用到的。

代码 ：

public class TestDbunit  {

public static JdbcDatabaseTester tester;

IDataSet dataSet_Insert ;

IDataSet dataSet_init;

IDataSet dataSet_bak;

@BeforeClass

protected void setUp() throws Exception{

System.out.println("BeforeClass");

IDatabaseConnection conn=tester.getConnection();

String[] depTableNames=new String[]{"ROLES","USERS","USER_ROLES"};

try{

IDataSet utput=conn.createDataSet(depTableNames);

FileOutputStream fos=new FileOutputStream("user-roles-bak.xml");

//备份数据

FlatXmlDataSet.write(output, fos);

//初始化数据

//dataSet_init=new FlatXmlDataSet(new File("user-roles.xml"));

//DatabaseOperation.CLEAN_INSERT.execute(conn, dataSet_init);

}finally{

conn.close();

}

}

@BeforeTest

public void loadDB()throws Exception{

System.out.println("@BeforeTest");

tester=new  JdbcDatabaseTester( "com.mysql.jdbc.Driver" ,

               "jdbc:mysql://localhost:3306/test" ,  "root" , "" );

dataSet_init=new FlatXmlDataSet(new File("user-roles.xml"));

IDatabaseConnection conn=tester.getConnection();

try{

DatabaseOperation.CLEAN_INSERT.execute(conn, dataSet_init);

}finally{

conn.close();

}

}

@Test

public void testInsert()throws Exception{

dataSet_Insert=new FlatXmlDataSet(new File("user_insert.xml"));

IDatabaseConnection conn=tester.getConnection();

try{

DatabaseOperation.INSERT.execute(conn, dataSet_Insert);

assert conn.getRowCount("users")==3;

}finally{

conn.close();

}

}

@Test

public void testUpdate()throws Exception{

IDataSet dataSet_Update=new FlatXmlDataSet(new File("user_update.xml"));

IDatabaseConnection conn=tester.getConnection();

IDataSet expect=new  FlatXmlDataSet(new File("user_update_expected.xml"));

IDataSet act=tester.getConnection().createDataSet();

try{

DatabaseOperation.UPDATE.execute(conn, dataSet_Update);

System.out.println(act.getTable("users").getRowCount());

System.out.println(act.getTable("users").getValue(act.getTable("users").getRowCount()-2,"last_name"));

assert act.getTable("users").getValue(act.getTable("users").getRowCount()-2,"last_name").toString().equals("shifan");

}finally{

conn.close();

}

}

@AfterClass

public void clearDB() throws Exception{

IDatabaseConnection conn=tester.getConnection();

try{

//数据还原

DatabaseOperation.CLEAN_INSERT.execute(conn, dataSet_init);

}finally{

conn.close();

}

}

}