dbunit实例小结

最新推荐文章于 2021-02-01 01:42:30 发布

yanebupt

最新推荐文章于 2021-02-01 01:42:30 发布

阅读量5.6k

点赞数

文章标签： exception import dataset 数据库 string jdbc

本文链接：https://blog.csdn.net/yangyan19870319/article/details/6063049

版权

使用dbunit用到的jar包有：

1.添加dbunit-2.3.0.jar

2.添加slf4j-api-1.5.3.jar

3.添加slf4j-jcl-1.5.3.jar

4.添加commons-logging.jar

另外如果没有对应数据库的驱动还需要添加对应数据库驱动jar

准备一个数据库表person

id username password

1 zhangsan 123

2 lisi 456

3 wangwu 789

实例一：导出数据库中的数据到xml文件

package com.test.dbunit;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.SQLException;

public class UsersDB {
public void save(Users users){
   Connection conn=null;
   try{
    Class.forName("com.mysql.jdbc.Driver");
    conn=DriverManager.getConnection("jdbc:mysql://localhost/test",
      "root", "");
    String sql="insert into person(id,username,password) values(?,?,?)";
    PreparedStatement ps =conn.prepareStatement(sql);
    ps.setInt(1, users.getId());
    ps.setString(2,users.getUsername());
    ps.setString(3,users.getPassword());
    ps.executeUpdate();
   }catch(Exception ex){
    ex.printStackTrace();
   }finally
   {
    if(null!= conn)
    {
     try
     {
      conn.close();
     }catch(SQLException e){
      e.printStackTrace();
     }
    }
   }
}

}

dbunit测试类：

package com.test.dbunit;

import org.dbunit.DBTestCase;
import org.dbunit.database.DatabaseConnection;
import org.dbunit.database.IDatabaseConnection;
import org.dbunit.dataset.IDataSet;
import java.io.File;
import java.io.FileInputStream;
import java.sql.Connection;
import java.sql.DriverManager;

import org.dbunit.Assertion;
import org.dbunit.DBTestCase;
import org.dbunit.DatabaseTestCase;
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 Test extends DatabaseTestCase {

    public Test()
    {
//      super();
        System.setProperty(
                PropertiesBasedJdbcDatabaseTester.DBUNIT_DRIVER_CLASS,
                "com.mysql.jdbc.Driver");
        System.setProperty(
                PropertiesBasedJdbcDatabaseTester.DBUNIT_CONNECTION_URL,
                "jdbc:mysql://localhost/test");
        System.setProperty(PropertiesBasedJdbcDatabaseTester.DBUNIT_USERNAME,
                "root");
        System.setProperty(PropertiesBasedJdbcDatabaseTester.DBUNIT_PASSWORD,
                "");
    }

    /**
     * 给定数据集
     */
    @Override
    protected IDataSet getDataSet() throws Exception {
        System.out.println("init...");
        return new FlatXmlDataSet(new FileInputStream("dbunit.xml"));
    }

    public void test1() throws Exception{
        IDataSet dbDataSet = getConnection().createDataSet();
        ITable dbTable = dbDataSet.getTable("person");

        IDataSet xmlDataSet = new FlatXmlDataSet(new FileInputStream("dbunit.xml"));
        ITable xmlTable = xmlDataSet.getTable("person");

        Assertion.assertEquals(xmlTable, dbTable);
    }

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

        Users users = new Users();
        users.setId(9);
        users.setUsername("hello");
        users.setPassword("world");

        db.save(users);

        IDataSet dataSet = getConnection().createDataSet();
        ITable actualTable = dataSet.getTable("person");

        IDataSet dataSet2 = new FlatXmlDataSet(new File("expect.xml"));
        ITable expectedTable = dataSet2.getTable("person");

        Assertion.assertEquals(expectedTable, actualTable);
    }

    /**
     * 在把xml里的数据导入到db里之前，需要对数据库里的表的数据做的操作。
     */
    protected DatabaseOperation getSetUpOperation() throws Exception
    {
        System.out.println("setup...");
        return DatabaseOperation.CLEAN_INSERT;
    }
    /**
     * 测试执行完以后，需要对数据库里的表里的数据做的操作
     */
    protected DatabaseOperation getTearDownOperation() throws Exception
    {
        System.out.println("teardown...");
        return DatabaseOperation.NONE;
        //清空数据库
       // return DatabaseOperation.DELETE_ALL;
    }

@Override
protected IDatabaseConnection getConnection() throws Exception {
  // TODO Auto-generated method stub
   Class.forName("com.mysql.jdbc.Driver");

      Connection conn = DriverManager.getConnection("jdbc:mysql://localhost/test", "root", "");

      IDatabaseConnection connection = new DatabaseConnection(conn);
  return connection;
}

}

package com.test.dbunit;

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) throws Exception{
        Class.forName("com.mysql.jdbc.Driver");

        Connection conn = DriverManager.getConnection(
                "jdbc:mysql://localhost/test", "root", "");

        IDatabaseConnection connection = new DatabaseConnection(conn);
        QueryDataSet dataSet = new QueryDataSet(connection);
        //将整个person表里的数据导出到 xml文件里
        dataSet.addTable("person");
        //将users表里符合条件的数据导出到xml文件里
        //dataSet.addTable("users","select * from users where id < 4");
        //导出到dbunit.xml文件里
        FlatXmlDataSet.write(dataSet,new FileOutputStream("dbunit.xml"));
    }

}

几点说明：jdbc:mysql://localhost/test中test为对应的数据库名称，dbunit.xml为对应的数据输出文件，导出的文件格式如下

实例三：表操作测试实例

此实例用到的几个类文件内容分别如下：

package com.test.dbunit;

public class Users {
private int id;
private String username;
private String password;

public int getId() {
return id;
}
public void setId(int id) {
this.id = id;
}
public String getUsername() {
return username;
}
public void setUsername(String userName) {
this.username = userName;
}
public String getPassword() {
return password;
}
public void setPassword(String passWord) {
this.password = passWord;
}

}

<?xml version='1.0' encoding='UTF-8'?>
<dataset>
<person id="1" username="zhangsan" password="123"/>
<person id="2" username="lisi" password="456"/>
<person id="3" username="wangwu" password="789"/>
</dataset>

实例二：验证数据库表内容与导出xml文件数据是否一致

package com.test.dbunit;

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.operation.DatabaseOperation;
import junit.framework.TestCase;
import org.dbunit.dataset.xml.FlatXmlDataSet;
/*
* 测试表数据是否一致
*/
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/test");
        System.setProperty(PropertiesBasedJdbcDatabaseTester.DBUNIT_USERNAME,"root");
        System.setProperty(PropertiesBasedJdbcDatabaseTester.DBUNIT_PASSWORD,"");
    }

    /**
     * 给定数据集
     */
    @Override
    protected IDataSet getDataSet() throws Exception {
        System.out.println("init...");
        return new FlatXmlDataSet(new FileInputStream("dbunit.xml"));
    }

    public void test1() throws Exception{
        IDataSet dbDataSet = getConnection().createDataSet();
        ITable dbTable = dbDataSet.getTable("person");

        IDataSet xmlDataSet = new FlatXmlDataSet(new FileInputStream("dbunit.xml"));
        ITable xmlTable = xmlDataSet.getTable("person");

        Assertion.assertEquals(xmlTable, dbTable);
    }
}