12

如何轻松导入/导出 dbunit 可以采用以下格式的数据库数据?

<dataset>
   <tablea cola="" colb="" />
   <tableb colc="" cold="" />
</dataset>

我想找到一种方法从数据库中导出现有数据以进行单元测试。

4

2 回答 2

17

蓝色,这将让您以您想要的格式导出数据。

public class DatabaseExportSample {
    public static void main(String[] args) throws Exception {
        // database connection
        Class driverClass = Class.forName("org.hsqldb.jdbcDriver");
        Connection jdbcConnection = DriverManager.getConnection(
                "jdbc:hsqldb:sample", "sa", "");
        IDatabaseConnection connection = new DatabaseConnection(jdbcConnection);

        // partial database export
        QueryDataSet partialDataSet = new QueryDataSet(connection);
        partialDataSet.addTable("FOO", "SELECT * FROM TABLE WHERE COL='VALUE'");
        partialDataSet.addTable("BAR");
        FlatXmlDataSet.write(partialDataSet, new FileOutputStream("partial.xml"));

        // full database export
        IDataSet fullDataSet = connection.createDataSet();
        FlatXmlDataSet.write(fullDataSet, new FileOutputStream("full.xml"));

        // dependent tables database export: export table X and all tables that
        // have a PK which is a FK on X, in the right order for insertion
        String[] depTableNames = 
          TablesDependencyHelper.getAllDependentTables( connection, "X" );
        IDataSet depDataSet = connection.createDataSet( depTableNames );
        FlatXmlDataSet.write(depDataSet, new FileOutputStream("dependents.xml"));
    }
}
于 2013-01-22T02:47:52.997 回答
6

出口已经回答。要完成答案,您可以通过以下方式将数据集导入数据库。 connection是类型IDatabaseConnection。Mark Robinson 的上一个答案(导出)包含如何创建数据库连接的代码。

FlatXmlDataSet dataSet = new FlatXmlDataSet(new FlatXmlProducer(new InputSource(new FileInputStream("dataset.xml"))));
DatabaseOperation.CLEAN_INSERT.execute(connection, dataSet);
于 2013-04-19T10:04:58.487 回答