参考demo：G:\学习资料\java\java学习文档\java技术文档\Spring文档\xufei-spring-word\xufei-projo-test =》dbunit

dbunit是用来干嘛的呢? 简单的说 dbunit的主要的功能就是用来做DAO层的测试

不是 dbutils(DAO层的解决方案) 、dbunit(专门用来测试DAO层的框架)

导包

         <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
            <version>5.1.6</version>
        </dependency>

        <dependency>
            <groupId>commons-dbutils</groupId>
            <artifactId>commons-dbutils</artifactId>
            <version>1.6</version>
        </dependency>

        <!--DAO层测试的这个框架包-->
        <dependency>
            <groupId>org.dbunit</groupId>
            <artifactId>dbunit</artifactId>
            <version>2.5.3</version>
        </dependency>

        <dependency>
            <groupId>com.alibaba</groupId>
            <artifactId>druid</artifactId>
            <version>1.1.22</version>
        </dependency>

编写DAO代码

public class UserDAO {
    /**
     * 通过id找用户
     *
     * @param id
     * @return
     */
    public User findUserById(int id) throws SQLException {
        User user = queryRunner().query("select * from t_user where uId=?", new BeanHandler<User>(User.class), id);
        return user;
    }
}

编写数据库访问帮助类

public class JdbcUtils {

    private static DruidDataSource dataSource = null;

    static {
        dataSource = new DruidDataSource();
        dataSource.setDriverClassName("com.mysql.jdbc.Driver");
        dataSource.setUrl("jdbc:mysql://127.0.0.1:3369/xx?useSSL=false&serverTimezone=Asia/Shanghai&allowPublicKeyRetrieval=true&nullCatalogMeansCurrent=true&rewriteBatchedStatements=true");
        dataSource.setUsername("root");
        dataSource.setPassword("xxx");
    }

    /**
     * 这个是获取咋们的操作数据库的对象
     *
     * @return
     */
    public static QueryRunner queryRunner() {
        return new QueryRunner(dataSource);
    }

    /**
     * 获取咋们的连接
     *
     * @return
     * @throws SQLException
     */
    public static Connection getConnection() throws SQLException {
        return dataSource.getConnection();
    }


}

编写基类

public class AbstractDbunitTestCase {

    private DatabaseConnection connection;
    private File tempFile;
    private IDataSet testDataSet;    // 测试数据的DataSet对象

    public AbstractDbunitTestCase(Connection connection, String testFileName) throws DatabaseUnitException {
        this.connection =  new DatabaseConnection(connection);
        InputStream inputStream = AbstractDbunitTestCase.class.getClassLoader().getResourceAsStream(testFileName);
        testDataSet = new FlatXmlDataSet(new FlatXmlProducer(new InputSource(inputStream)));
    }

    public AbstractDbunitTestCase(String testFileName) throws DatabaseUnitException {
        InputStream inputStream = AbstractDbunitTestCase.class.getClassLoader().getResourceAsStream(testFileName);
        testDataSet = new FlatXmlDataSet(new FlatXmlProducer(new InputSource(inputStream)));
    }


    /**
     * 设置这个连接池的包裹对象
     *
     * @param conn
     * @throws DatabaseUnitException
     */
    protected void setDatabaseConnection(Connection conn) throws DatabaseUnitException {
        this.connection = new DatabaseConnection(conn);
    }

    /**
     * 这个就是备份多张表
     *
     * @param tabNames
     */
    protected synchronized void backManyTable(String... tabNames) throws Exception {

        QueryDataSet queryDataSet = new QueryDataSet(connection);
        for (String tabName : tabNames) {
            queryDataSet.addTable(tabName);
        }
        // 实施备份 使用到临时文件
        tempFile = File.createTempFile("dbunitTest", ".xml");
        // 实施备份
        if (null != tempFile) {
            FlatXmlDataSet.write(queryDataSet, Files.newOutputStream(tempFile.toPath()));
        }
    }


    /**
     * 备份一张表
     */
    protected void backOneTable(String tabName) throws Exception {
        backManyTable(tabName);
    }


    /**
     * 插入测试数据
     */
    protected void insertTestData() throws DatabaseUnitException, SQLException {
        if (null != connection && null != testDataSet) {
            DatabaseOperation.CLEAN_INSERT.execute(connection, testDataSet);
        } else {
            throw new RuntimeException("参数有误不能插入数据:");
        }
    }

    /**
     * 还原表中的数据
     */
    protected void resumeTable() throws Exception {
        if (null != tempFile) {
            IDataSet dataSet = new FlatXmlDataSet(new FlatXmlProducer(new InputSource(Files.newInputStream(tempFile.toPath()))));
            DatabaseOperation.CLEAN_INSERT.execute(connection, dataSet);
        } else {
            throw new RuntimeException("临时文件都没有怎么还原....");
        }
    }
}

编写测试类

public class TestUserDAO1 extends AbstractDbunitTestCase {

    private UserDAO userDAO;
    private User exUser;

    public TestUserDAO1() throws SQLException, DatabaseUnitException {
        super(JdbcUtils.getConnection(), "testdata.xml");
    }


    @Before
    public void init() throws Exception {
        userDAO = new UserDAO();
        exUser = new User(1, "xff", "123");
        backOneTable("t_user");
        insertTestData();
    }


    @Test
    public void testFindUserById() throws SQLException {
       //进行业务逻辑测试
        User userResult = userDAO.findUserById(1);
        Assert.assertEquals(exUser.getUId(), userResult.getUId());
        Assert.assertEquals(exUser.getUName(), userResult.getUName());
        Assert.assertEquals(exUser.getPassWord(), userResult.getPassWord());
    }


    @After
    public void destory() throws Exception {
        resumeTable();
    }

}