2013-01-16 189 views
9

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

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

我想找到一种方法,从导出数据库中现有的数据为我的单元测试。

+0

到目前为止您尝试过什么?您是否阅读过dbunit.org上的文档? –

+0

我手动或使用postgresql客户端。只是想知道是否有更好的方法。 @MarkRobinson – blue123

回答

14

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

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")); 
    } 
} 
+2

@ blue123,该摘录来自常见问题解答,其中包含其他信息,您可能会发现它们很有用:http://www.dbunit.org/faq.html#extract – Paul

+0

此复制粘贴的摘录中包含的错误与FAQ – naXa

+0

上面的URL已损坏。使用这一个:http://dbunit.sourceforge.net/faq.html#extract –

6

导出已经回答。 要完成答案,您可以将导入将数据集导入数据库。 connection类型为IDatabaseConnection。 马克罗宾逊先前的答案(出口)包含如何创建数据库连接的代码。

FlatXmlDataSet dataSet = new FlatXmlDataSet(new FlatXmlProducer(new InputSource(new FileInputStream("dataset.xml")))); 
DatabaseOperation.CLEAN_INSERT.execute(connection, dataSet); 
+2

第一行可以简单地为'FlatXmlDataSet dataSet = new FlatXmlDataSet(new File(“dataset.xml”),true);''或'FlatXmlDataSet dataSet = new FlatXmlDataSetBuilder()。build(new File(“dataset.xml”));' –