JAVA 数据库测试框架DBUnit之Hibernate篇

1.准备数据库,这里我使用MYSQL数据库，创建表格 student,如下所示：

SET FOREIGN_KEY_CHECKS=0;
-- ----------------------------
-- Table structure for `student`
-- ----------------------------
DROP TABLE IF EXISTS `student`;
CREATE TABLE `student` (
  `id` char(4) NOT NULL,
  `name` char(50) DEFAULT NULL,
  `sex` char(1) DEFAULT NULL,
  `birthday` date DEFAULT NULL,
  PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8;

2.在上面的表中插入如下数据：

INSERT INTO`student` VALUES ('0011', 'haha', 'f', '1984-11-2');
INSERT INTO `student` VALUES ('0044', 'lulu', 'f', '1984-11-2');
INSERT INTO `student` VALUES ('0066', 'jie', 'm', '1984-11-2');
INSERT INTO `student` VALUES ('0077', 'minggxu9', 'm', '1984-11-2');

3.准备好数据库以后，然后配置Hibernate,以及DBUnit的支持。我这里还没有用Maven进行管理，只是将它们下载下来，并导入至CLASSPATH中而已。
Hibernate的下载地址为：
http://sourceforge.net/projects/hibernate/files/hibernate3/3.1.2/hibernate-3.1.2.zip/download

DBUnit的下载地址为：
http://sourceforge.net/projects/dbunit/files/

4.编写一个Hibernate例子。这里使用上面提拱的student表做以下两个操作：
1）查询所有学生
2）添加一个学生

实现步骤如下所示：
4.1 书写hibernate.cfg.xml文件，将它保存至src/main/resources源文件夹下面，它的内容如下所示：

<?xml version='1.0'encoding='UTF-8'?>
<!DOCTYPE hibernate-configuration PUBLIC
          "-//Hibernate/HibernateConfiguration DTD 3.0//EN"
         "http://hibernate.sourceforge.net/hibernate-configuration-3.0.dtd">
<!-- Generated by MyEclipse Hibernate Tools. -->
<hibernate-configuration>
        <session-factory>
                <!-- JDBC驱动程序 -->
                <propertyname="connection.driver_class">com.mysql.jdbc.Driver</property>
                <!-- JDBCURL -->
               <propertyname="connection.url">               jdbc:mysql://localhost:3306/atdDB?useUnicode=true&characterEncoding=UTF8</property>
                <!-- 数据库用户名 -->
                <propertyname="connection.username">root</property>
                <!-- 数据库密码 -->
                <propertyname="connection.password">123456</property>
                <!-- sql的方言，不同db的sql略有不同   方言区分大小写-->
                <propertyname="dialect">org.hibernate.dialect.MySQLDialect</property>
                <!-- 后台显示sql语句，主要是调试用 -->
                <propertyname="show_sql">true</property>

         <!-- 映射文件 --> 
       <mapping resource="Student.hbm.xml"/>

        </session-factory>
</hibernate-configuration>

注意： 上面的红色高亮显示，表示是要引入一个student.hbm.xml自定义映射文件，它的细节如4.3所示。

4.2 编写POJO类（Student .java），也即JAVABEAN,相当于DDD中的CTO数据传输对象。

package po;

public class Student {

        private String id;
        private String name;
        private String sex;
        private String birthday;

        public String getId() {
                return id;
        }

        public void setId(String id) {
                this.id = id;
        }

        public String getName() {
                return name;
        }

        public void setName(String name) {
                this.name =name;
        }

        public String getSex() {
                return sex;
        }

        public void setSex(String sex) {
                this.sex =sex;
        }

        public String getBirthday() {
                returnbirthday;
        }

        public void setBirthday(String birthday){
                this.birthday= birthday;
        }

}

4.3 编写映射文件Stuend.hbm.xml，之前的例子中使用的是JPA方式--泛型注入的方式，我这里写了个映射文件，可以看到Hibernate的数据库表格中的映射情况，如下所示：

<!DOCTYPE hibernate-mapping 
    PUBLIC "-//Hibernate/Hibernate Mapping DTD//EN" 
   "http://hibernate.sourceforge.net/hibernate-mapping-2.0.dtd">

<hibernate-mapping>
       <class name="po.Student"table="student">
                <idname="id" type="string" unsaved-value="null">
                       <generator class="assigned" />
                </id>
                <propertyname="name"  type="string"not-null="true"/>
                <propertyname="sex"  type="string"not-null="true"/>
                <propertyname="birthday"  />
        </class>
</hibernate-mapping>

注意：这个映射文件表示它与po层的JAVABEAN(Student.java)配合一起使用的。它们之间的字段和属性一一对应，
<generator class="assigned" />表示的是id为主键的。


4.3书写测试测子--HibernateTest.java，就是上面的两个操作（查询全部和插入功能）:

packagecom.db.oper;

import java.util.List;
import java.util.ListIterator;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.hibernate.HibernateException;
import org.hibernate.Session;
import org.hibernate.SessionFactory;
import org.hibernate.Transaction;
import org.hibernate.cfg.Configuration;

import po.Student;

public class HibernateTest {
        private static Log log =LogFactory.getLog(HibernateTest.class);

        public void find() {
               SessionFactory sessionFactory;
                try {
                       sessionFactory = newConfiguration().configure()
                                      .buildSessionFactory();

                       Session session = sessionFactory.openSession();

                       List students = (List)session.createQuery("from Student").list();
                        //也可以使用原生态的SQL,我这里写了一个类SQL语句

注：也可以使用原生态的SQL,我这里写了一个类SQL语句  ，而插入操作也比较简单，即将student对象持久化到session中即可。而这个session是单例模式，只允许同一个时间出现一个。

5.使用DBUnit持久化测试框架

      dbunit的原理是：它会把数据库表里的数据和一个xml文件里表示的数据关联起来。也就是说数据库表里的数据可以导出到一个对应的xml里，同时也可以将一个xml里的数据导入到数据库表里。是相互转换的。

数据库表里的数据<===>xml文件

5.1 一个简单的测试例子。TestApp.java的内容如下所示：

 

package com.db.oper;

import java.io.FileOutputStream;
import java.sql.Connection;
import java.sql.DriverManager;

import org.dbunit.database.DatabaseConnection;
import org.dbunit.database.IDatabaseConnection;
import org.dbunit.database.QueryDataSet;
import org.dbunit.dataset.xml.FlatXmlDataSet;

public class TestApp {

        public static void main(String[] args) throwsException {
                
               Class.forName("com.mysql.jdbc.Driver");
                Connectionconn = DriverManager
                              .getConnection(
                                             "jdbc:mysql://localhost:3306/atdDB?useUnicode=true&characterEncoding=UTF8",
                                              "root","123456");

               IDatabaseConnection connection = new DatabaseConnection(conn);
                QueryDataSetdataSet = new QueryDataSet(connection);
                
                // 将整个person表里的数据导出到 xml文件里
               dataSet.addTable("student");
                // 将users表里符合条件的数据导出到xml文件里
                // 导出到student.xml文件里
               FlatXmlDataSet.write(dataSet, new FileOutputStream("student.xml"));
        }

}

导出的student.xml文件的内容如下所示：

<?xml version='1.0'encoding='UTF-8'?>
<dataset>
  <student id="0001" name="mingg"sex="m" birthday="2011-11-06"/>
  <student id="0002" name="jie"sex="f" birthday="2010-10-01"/>
  <student id="004" name="dsad"sex="f" birthday="2011-12-14"/>
  <student id="0088" name="wangdo"sex="m" birthday="2008-12-16"/>
</dataset>

里面的内容同数据库表中的内容是一一对应的。如果我会想测试查询中的数据是否同数据库中的内容一致，我们只要对比这个XML文件即可，不用再去查看数据库表和表中的内容了。

下面我们做一个测试类，里面包括表，表结构，以及表中内容的测试。它的内容如下所示：

package com.db.test;

import java.io.File;
import java.io.FileInputStream;

import org.dbunit.Assertion;
import org.dbunit.DBTestCase;
import org.dbunit.PropertiesBasedJdbcDatabaseTester;
import org.dbunit.dataset.IDataSet;
import org.dbunit.dataset.ITable;
import org.dbunit.dataset.SortedTable;
import org.dbunit.dataset.filter.DefaultColumnFilter;
import org.dbunit.dataset.xml.FlatXmlDataSet;
import org.dbunit.operation.DatabaseOperation;
public class SimpleTest extends DBTestCase {

        public SimpleTest()
        {
//                super();
               System.setProperty(
                              PropertiesBasedJdbcDatabaseTester.DBUNIT_DRIVER_CLASS,
                              "com.mysql.jdbc.Driver");
                System.setProperty(
                              PropertiesBasedJdbcDatabaseTester.DBUNIT_CONNECTION_URL,                         

              "jdbc:mysql://localhost:3306/atdDB?useUnicode=true&characterEncoding=UTF8");
               System.setProperty(PropertiesBasedJdbcDatabaseTester.DBUNIT_USERNAME,
                              "root");
               System.setProperty(PropertiesBasedJdbcDatabaseTester.DBUNIT_PASSWORD,
                              "123456");
        }
        
        /**
         * 给定数据集
         */
        @Override
        protected IDataSet getDataSet() throwsException {
               System.out.println("init...");
                return newFlatXmlDataSet(new FileInputStream("student.xml"));
        }
        
        public void test1() throws Exception{
                IDataSetdbDataSet = getConnection().createDataSet();
                ITabledbTable = dbDataSet.getTable("student");
                
                IDataSetxmlDataSet = new FlatXmlDataSet(new FileInputStream("student.xml"));
                ITablexmlTable = xmlDataSet.getTable("student");
                
               Assertion.assertEquals(xmlTable, dbTable);
        }
        /**
         * 表结构和student.xml结构不一致
         */
        public void test2() throws Exception{
                IDataSetdbDataSet = getConnection().createDataSet();
                ITabledbTable = dbDataSet.getTable("student");
                
                IDataSetxmlDataSet = new FlatXmlDataSet(new FileInputStream("student.xml"));
                ITablexmlTable = xmlDataSet.getTable("student");
   

             dbTable =  DefaultColumnFilter.includedColumnsTable(dbTable,xmlDataSet.getTableMetaData("student").getColumns());
         
               Assertion.assertEquals(xmlTable, dbTable);
        }
        /**
         * 只比较name
         */
        public void test3() throws Exception{
                IDataSetdbDataSet = getConnection().createDataSet();
                ITabledbTable = dbDataSet.getTable("student");
                
                IDataSetxmlDataSet = new FlatXmlDataSet(new FileInputStream("student.xml"));
                ITablexmlTable = xmlDataSet.getTable("student");
                
                dbTable =DefaultColumnFilter.excludedColumnsTable(dbTable, newString[]{"id","name"});
                xmlTable =DefaultColumnFilter.excludedColumnsTable(dbTable, newString[]{"name"});
                
               Assertion.assertEquals(xmlTable, dbTable);
        }
        /**
         * 表里数据和student.xml里的数据的顺序不一致
         */
        public void test4() throws Exception{
                IDataSetdbDataSet = getConnection().createDataSet();
                ITabledbTable = dbDataSet.getTable("student");
                
                IDataSet xmlDataSet= new FlatXmlDataSet(new FileInputStream("student.xml"));
                ITablexmlTable = xmlDataSet.getTable("student");
                
                SortedTabledbSortedTable = new SortedTable(dbTable,new String[]{"id"});
                //按数据库里表结构字段类型来进行排序，如果是int则按int排序，不是依照字符串的形式排序
               dbSortedTable.setUseComparable(true);
                
                SortedTablexmlSortedTable = new SortedTable(xmlTable,new String[]{"id"});
                //按数据库里表结构字段类型来进行排序，如果是int则按int排序，不是依照字符串的形式排序
               xmlSortedTable.setUseComparable(true);
                
               Assertion.assertEquals(xmlSortedTable, dbSortedTable);
        }
        
        /**
         * 对student表插入一条记录做测试
         * @throws Exception
         */
//        public void test5() throws Exception{
//
//
//               db.save(users);
//
//                IDataSetdataSet = getConnection().createDataSet();
//                ITableactualTable = dataSet.getTable("users");
//
//        IDataSet dataSet2 = newFlatXmlDataSet(new File("expected.xml"));
//                ITableexpectedTable = dataSet2.getTable("users");
//
//        Assertion.assertEquals(expectedTable,actualTable);
//        }
        
        /**
         * 在把xml里的数据导入到db里之前，需要对数据库里的表的数据做的操作。
         */
        protected DatabaseOperationgetSetUpOperation() throws Exception
    {
               System.out.println("setup...");
        return DatabaseOperation.CLEAN_INSERT;
    }
        /**
         * 测试执行完以后，需要对数据库里的表里的数据做的操作
         */
    protected DatabaseOperation getTearDownOperation() throwsException
    {
           System.out.println("teardown...");
        return DatabaseOperation.DELETE_ALL;
    }

}

特别注意注释掉的部分，下面我重写一下：
HibernateTest.java部分：

public voidinsert(Student student){
                SessionFactory sessionFactory;
                try {
                sessionFactory = new Configuration().configure()
                .buildSessionFactory();
                
                Session session = sessionFactory.openSession();
                Transaction tx = session.beginTransaction();
                session.save(student);
                tx.commit();
                session.close();
                sessionFactory.close();
                
                 System.out.println("插入数据OK!请在MySQL查看结果！");
                 }catch (HibernateException e) {
                 //TODO Auto-generated catch block
                e.printStackTrace();
                 }
         }

SimpleTest.java重写部分：

/**
         * 对Users表插入一条记录做测试
         * @throws Exception
         */
        public void test5() throws Exception{


                Studentstudent = new Student();
       student.setBirthday("1988-11-02");
        student.setId("1111");
        student.setName("hong");
        student.setSex("f");
        
        HibernateTest ht = new HibernateTest();
        ht.insert(student);
        
        
                
                IDataSetdataSet = getConnection().createDataSet();
                ITableactualTable = dataSet.getTable("student");

        IDataSet dataSet2 = newFlatXmlDataSet(new File("expected.xml"));
                ITableexpectedTable = dataSet2.getTable("student");

        Assertion.assertEquals(expectedTable,actualTable);
        }
        

而expected.xml部分如下所示：

<?xmlversion='1.0' encoding='UTF-8'?>
<dataset>
  <student id="0001" name="mingg"sex="m" birthday="2011-11-06"/>
  <student id="0002" name="jie"sex="f" birthday="2010-10-01"/>
  <student id="004" name="dsad"sex="f" birthday="2011-12-14"/>
  <student id="0088" name="wangdo"sex="m" birthday="2008-12-16"/>
   <student id="1111" name="hong"sex="f" birthday="1988-11-02"/>
</dataset>

右键点击SimpleTest.java类，运行结果如下所示：

下面的五个测试方法全部通过。
（完，待续.................................................................）

我在一个内部论坛也写过的