Allow filtering on integration test print dataset
authorAndreas Müller <a.mueller@bgbm.org>
Tue, 3 Jul 2012 19:50:38 +0000 (19:50 +0000)
committerAndreas Müller <a.mueller@bgbm.org>
Tue, 3 Jul 2012 19:50:38 +0000 (19:50 +0000)
cdmlib-persistence/src/test/java/eu/etaxonomy/cdm/test/integration/CdmIntegrationTest.java

index 000a3dd869d84ff666a99f5894a924d51021fef2..fe994edf4b58c1de9743e783bc4583821178ee7d 100644 (file)
@@ -38,6 +38,8 @@ import org.dbunit.database.DatabaseDataSet;
 import org.dbunit.database.IDatabaseConnection;\r
 import org.dbunit.dataset.IDataSet;\r
 import org.dbunit.dataset.ITableIterator;\r
+import org.dbunit.dataset.filter.ExcludeTableFilter;\r
+import org.dbunit.dataset.filter.ITableFilterSimple;\r
 import org.dbunit.dataset.xml.FlatDtdDataSet;\r
 import org.dbunit.dataset.xml.FlatXmlDataSet;\r
 import org.dbunit.dataset.xml.XmlDataSetWriter;\r
@@ -168,13 +170,21 @@ public abstract class CdmIntegrationTest extends UnitilsJUnit4 {
      * @see FlatFullXmlWriter\r
      */\r
     public void printDataSetWithNull(OutputStream out) {\r
-        IDatabaseConnection connection = null;\r
-\r
+       printDataSetWithNull(out, null);\r
+    }\r
+    \r
+    public void printDataSetWithNull(OutputStream out, ITableFilterSimple filter) {\r
+        if (filter == null){\r
+               filter = new ExcludeTableFilter();\r
+        }\r
+       IDatabaseConnection connection = null;\r
+        \r
+        \r
         try {\r
             connection = getConnection();\r
 \r
 //                     IDataSet dataSet = connection.createDataSet();\r
-            DatabaseDataSet dataSet = new DatabaseDataSet(connection, false);\r
+            DatabaseDataSet dataSet = new DatabaseDataSet(connection, false, filter);\r
 \r
             FlatFullXmlWriter writer = new FlatFullXmlWriter(out);\r
             writer.write(dataSet);\r