java mock 数据库_spring-mock + dbutil 用来测试数据库操作

大概流程：1.用dbunit创建初始的测试数据。2.用spring-mock 维护测试过程中的数据会滚，这样可以保证测试后数据库保持原状态。3.用junit架构测试。4.用dbunit销毁初始测试数据。

Java代码 packagecom.test.dbunit.dao;

importjavax.sql.DataSource;

importorg.dbunit.Assertion;

importorg.dbunit.database.DatabaseConnection;

importorg.dbunit.database.IDatabaseConnection;

importorg.dbunit.database.QueryDataSet;

importorg.dbunit.dataset.IDataSet;

importorg.dbunit.dataset.xml.FlatXmlDataSet;

importorg.dbunit.operation.DatabaseOperation;

importorg.junit.Assert;

importorg.junit.Before;

importorg.junit.Test;

importorg.springframework.beans.factory.annotation.Autowired;

importorg.springframework.core.io.ClassPathResource;

importorg.springframework.jdbc.datasource.DataSourceUtils;

importorg.springframework.test.context.ContextConfiguration;

importorg.springframework.test.context.junit4.AbstractTransactionalJUnit4SpringContextTests;

importorg.springframework.test.context.transaction.TransactionConfiguration;

importcom.test.dbunit.entity.User;

@ContextConfiguration(locations = {"classpath:testApplicationContext.xml"})

@TransactionConfiguration(defaultRollback =true)

publicclassUserDaoTestextendsAbstractTransactionalJUnit4SpringContextTests {

@Autowired

privateUserDao userDao;

@Autowired

privateDataSource dataSource;

privateIDatabaseConnection conn;

@Before

publicvoidinitDbunit()throwsException {

conn =newDatabaseConnection(DataSourceUtils.getConnection(dataSource));

}

@Test

publicvoidsaveUser()throwsException {

User user =newUser();

user.setNick("user001");

user.setPassword("password001");

userDao.save(user);

QueryDataSet actual =newQueryDataSet(conn);

actual.addTable("user",

"select * from user where user.nick = 'user001'");

IDataSet expected =newFlatXmlDataSet(newClassPathResource(

"com/taobao/dbunit/dao/user001.xml").getFile());

Assertion.assertEquals(expected, actual);

}

@Test

publicvoidupdateUser()throwsException {

IDataSet origen =newFlatXmlDataSet(newClassPathResource(

"com/taobao/dbunit/dao/user001.xml&