1 files changed, 41 insertions, 0 deletions
diff --git a/src/test/java/at/ac/tuwien/sepm/assignment/groupphase/util/Helper.java b/src/test/java/at/ac/tuwien/sepm/assignment/groupphase/util/Helper.java
index b808206..618b06f 100644
--- a/src/test/java/at/ac/tuwien/sepm/assignment/groupphase/util/Helper.java
+++ b/src/test/java/at/ac/tuwien/sepm/assignment/groupphase/util/Helper.java
@@ -6,6 +6,12 @@ import java.util.List;
 import javafx.stage.Modality;
 import javafx.stage.Stage;
 import javafx.stage.Window;
+import org.dbunit.database.IDatabaseConnection;
+import org.dbunit.dataset.DataSetException;
+import org.dbunit.dataset.IDataSet;
+import org.dbunit.dataset.ITable;
+import org.dbunit.dataset.filter.DefaultColumnFilter;
+import org.dbunit.util.fileloader.FlatXmlDataFileLoader;
 import org.testfx.api.FxRobotContext;
 
 public class Helper {
@@ -25,4 +31,39 @@ public class Helper {
                         .findFirst()
                         .orElse(null);
     }
+
+    public static ITable getActualFilteredTableData(
+            IDatabaseConnection connection, String tableName, String[] excludedColumns)
+            throws Exception {
+        IDataSet actualDataSet = connection.createDataSet();
+        ITable actualTable = actualDataSet.getTable(tableName);
+        return getFilteredTableData(actualTable, excludedColumns);
+    }
+
+    public static ITable getExpectedFilteredTableData(
+            String tableName, String[] excludedColumns, String expectedXmlDataFileName)
+            throws Exception {
+        IDataSet dataSet = loadDataSet(expectedXmlDataFileName);
+        ITable table = dataSet.getTable(tableName);
+        return getFilteredTableData(table, excludedColumns);
+    }
+
+    public static ITable getActualFilteredQueryTableData(
+            IDatabaseConnection connection,
+            String resultTableName,
+            String sql,
+            String[] excludedColumns)
+            throws Exception {
+        ITable queryData = connection.createQueryTable(resultTableName, sql);
+        return getFilteredTableData(queryData, excludedColumns);
+    }
+
+    private static ITable getFilteredTableData(ITable tableData, String[] excludedColumns)
+            throws DataSetException {
+        return DefaultColumnFilter.excludedColumnsTable(tableData, excludedColumns);
+    }
+
+    private static IDataSet loadDataSet(String fileName) {
+        return new FlatXmlDataFileLoader().load("/" + fileName);
+    }
 }