DBUnit的疑惑

        在单元测试中测试DAO,我们通常都需要初始化一些数据,这时使用DBUnit就比较方便,只需要在XML文件中写就可以了。但比较让我困惑的是DBUnit初始化数据是直接提交到数据库的,这样如果我们需要在执行完测试后将数据库恢复成原始状态就显的异常的困难。尤其在使用Spring的AbstractTransactionalDataSourceSpringContextTests做测试时,我们不需要显示的设置事务回滚,所有的测试代码执行完后就自动回滚了。找了很多资料也没发现DBUnit有这样的功能。请问谁有好的办法没有?另外如果使用DBUnit的时候完全可以只初始化一次数据,不必每次在setUp的时候执行一遍。请问这样的做法是否妥当?

我的代码:

java 代码
  1. /**  
  2.  * 由DBUnit初始化数据.  
  3.  *   
  4.  * @param dataName 数据配置文件名称  
  5.  *   
  6.  * @throws Exception  
  7.  */  
  8. protected void initData(String dataName) throws Exception   
  9. {          
  10.     IDatabaseConnection connection = null;   
  11.     IDataSet dataSet = null;   
  12.     this.jdbcTemplate.getDataSource().getConnection().setAutoCommit(false);   
  13.     connection =  new DatabaseConnection(this.jdbcTemplate.getDataSource().getConnection());          
  14.     logger.info("连接的数据库为:"+this.jdbcTemplate.getDataSource().getConnection().getCatalog());   
  15.     ClassPathResource cpr = new ClassPathResource(dataName+"-data.xml");             
  16.     dataSet = new XmlDataSet(cpr.getInputStream());                
  17.     InsertIdentityOperation.CLEAN_INSERT.execute(connection, dataSet);   
  18. }  

猜你喜欢

转载自littcai.iteye.com/blog/69234