在数据库代码测试中，一般情况使用2种方案：
一是使用mock objects;
二是使用dbunit。

mock objects基于物理隔离层的概念，将涉及到数据库操作的代码，全用虚拟对象代替。这种方案，对业务领域里的代码来讲是可行的，也比较方便，但对于数据库操作层，此方案无用武之地，因为我们必须实实在在地与数据库打交道。

而在数据库测试中，因为我们力求将每个testcase中众多的测试方法完全隔离起来，不会因为一个测试方法因测试增加、删除功能而影响到另一个测试方法，这样，在每一个测试之前，数据库的状态是否稳定，甚至是完全不变，就显得很重要了。而这点，正是数据库测试的难点。

dbunit解决了这个问题。其原理很简单，就是在每个测试方法之前后，通过增删一些固定的记录，保持了数据库的固定状态，由此，我们可以在每个测试方法中自由地增删记录，而不用担心会影响到别的测试方法。

但dbunit也有一个问题，即它不能删除非空的外键记录。举例来说，假设“员工”表中有一非空字段为“部门编号”，引用了“部门”表的id, 只要“员工”表存在任一记录，“部门”表将不能被删除，强行删除将出现违犯约束(constraint violation)的异常。当然，如果必要，我们可以将数据库的约束条件改为连锁删除，这样，一旦我们删除一名员工记录，其所在的部门记录也将从“部门”表中删除。而此又会导致“员工”表中所有该部门的员工全被删除。这是绝对不允许的。当然，作为测试，我们可以先删除“员工”表，再删除“部门”表。

但有时，某些表自己引用自己，如“组织”表中有一“上级组织编号”字段，是自己“组织编号”的外键，即，此字段引用了本表中其他记录的“组织编号”。此时，我们必须先将这些引用了其他记录的“组织编号”的记录先删除，才能删除此表中的其他记录。而dbunit在实现上，只是用了一个简单的"delete from ..."的sql语句，不能解决这个问题。

dbunit的原理是如此简单，我们完全可以设计的“dbunit”，通过多重循环语句，干脆利落地删除自引用的整表。我们的“dbunit”，可以命名为“sqlrunner”。

package com.sarkuya.util.database;

import java.sql.connection;
import java.sql.drivermanager;
import java.sql.resultset;
import java.sql.sqlexception;
import java.sql.statement;

public class sqlrunner {
    
    static {
        try {
            class.forname("org.hsqldb.jdbcdriver");
        } catch (classnotfoundexception ex) {
            ex.printstacktrace();
        }
    }

    public static void executeupdate(string sql) {
        connection conn;
        statement stmt;
        
        try {
            conn = drivermanager.getconnection("jdbc:hsqldb:mem:testingdb", "sa", "");
            stmt = conn.createstatement();
            
            stmt.executeupdate(sql);
            
            stmt.close();
            conn.close();
        } catch (sqlexception ex) {
            ex.printstacktrace();
        }
        
    }
    
    public static boolean isundeletableforselfreference (string 表名, string 字段名) {
        connection conn;
        statement stmt;
        boolean result = true;
        
        try {
            conn = drivermanager.getconnection("jdbc:hsqldb:mem:testingdb", "sa", "");
            stmt = conn.createstatement();
            
            resultset rs = stmt.executequery("select count(*) from " + 表名 + " where " + 字段名 + " is not null");
            
            rs.next();
            
            if (rs.getint(1) != 0) {
                result = true;
            }
            else {
                result = false;
            }

            rs.close();
            stmt.close();
            conn.close();
        } catch (sqlexception ex) {
            ex.printstacktrace();
        }
        
        return result;
    }

可以看出，我们使用了jdbc的sql语句，而不是hibernate语句。hibernate的粉丝们可能大为不满，为何不使用hiberante? 别急，hibernate的语句将被大量地应用于实际测试当中。但是根据测试先行的原则，任何一个基于hiberante的语句都必须先测试再使用。而我们的这个“dbunit”是运行在实际测试之前，无法经过测试。当然，我们可以先假定这段hiberante代码正确无误，然后再实际测试它。这种方法也有一个缺点，因为测试代码常常会因为重构而发生改变，当测试代码改变时，这个“dbunit”也将被迫发生改变。而用jdbc的sql语句，可保持这段代码相对独立，不至于连诛九族。

executeupdate()将执行“insert”、“delete”语句。重点在于isundeletableforselfreference()方法。此方法在某个表的某个字段非空时，会返回false，告诉我们，此表中尚有被引用的记录存在，从而不能删除此表。尽管只有两个方法，但对于我们的“dbunit”来讲，已经足够了。

在testcase的setup()中，我们利用其executeupdate来增加一些必须的记录。

protected void setup() throws exception {
        sqlrunner.executeupdate("insert into 组织分类 values(1, '教育系统')");
        sqlrunner.executeupdate("insert into 组织分类 values(2, '商贸系统')");
        sqlrunner.executeupdate("insert into 组织分类 values(3, '供应商家')");
        sqlrunner.executeupdate("insert into 组织分类 values(4, '政府')");
        
        sqlrunner.executeupdate("insert into 组织 values(1, '中国贸易部', '北京三环路558号', 2, null)");
        sqlrunner.executeupdate("insert into 组织 values(2, '北京贸易厅', '北京四环路8号', 2, 1)");
        sqlrunner.executeupdate("insert into 组织 values(3, '河北高科技技术服务有限公司', '石家庄市白龙路23号', 3, null)");
        sqlrunner.executeupdate("insert into 组织 values(4, '四川珠宝有限公司', '成都市蓝天路56号', 3, null)");
        sqlrunner.executeupdate("insert into 组织 values(5, '北京昌平贸易局', '北京五环路18号', 2, 2)");
        
        sqlrunner.executeupdate("insert into 部门 values(1, '财务科', 2)");
        sqlrunner.executeupdate("insert into 部门 values(2, '市场部', 2)");
        sqlrunner.executeupdate("insert into 部门 values(3, '人事部', 2)");
    }

其中，“组织”表的结构为：

编号(bigint)，名称(varchar)，地址(varchar)，组织分类编号(bigint)，上级组织编号(bigint)

“部门”表的结构为：

编号(bigint)，名称(varchar)，地址(varchar)，组织编号(bigint)

在“组织”表中，编号为5的记录引用了2的记录，2的记录引用了1的记录。


而在teardown()中，我们配合isundeletableforselfreference()来删除相应记录。

 protected void teardown() throws exception {
        sqlrunner.executeupdate("delete from 部门");
        while (sqlrunner.isundeletableforselfreference("组织", "上级组织编号")) {
            sqlrunner.executeupdate("delete from 组织 where 上级组织编号 is not null and 编号 not in (select 上级组织编号 from 组织 where 上级组织编号 is not null)");
        }
        sqlrunner.executeupdate("delete from 组织");
        sqlrunner.executeupdate("delete from 组织分类");
    }

因为“部门”引用“组织”，“组织”引用“组织分类”，因此我们必须依序删除“部门”、“组织”及“组织分类”。难点在于while语句，其人工语义是，只要“组织”表中存在引用了其他记录的“编号”的记录，就会返回true，就先将这些引用的记录删除；只要“组织”表中不再有被引用的记录了，我们可以安全地用“delete from 组织”删除它们。

而在测试代码中，在任何一个测试方法中，我们可以直接使用如下语句：

assertequals(5, 组织service.get组织数量());

对于数据库测试代码来讲，速度是摆在第一位的，因此我们选择了hsqldb的内存数据库方式。这种方式不能永久保存记录，但只有测试期间，数据可用就行了。本人的实际测试代码中，某个testcase，共有28个测试方法，代码将近千行，测试速度不到8秒，基本可以忍受。主要瓶颈在于setup()及teardown()总共运行了28遍。当然，setup()中插入的数据越少，测试速度就越快，但每个测试方法中可能就需要增加一些工作量了。取舍完全在于你自己。

作者:sarkuya(作者的blog:http://blog.matrix.org.cn/page/sarkuya)
原文:http://blog.matrix.org.cn/page/sarkuya?entry=%e8%ae%be%e8%ae%a1%e8%87%aa%e5%b7%b1%e7%9a%84dbunit