import org.dbunit.IDatabaseTester;
import org.dbunit.database.DatabaseConfig;
import org.dbunit.database.IDatabaseConnection;
-import org.dbunit.dataset.CompositeDataSet;
-import org.dbunit.dataset.DataSetException;
-import org.dbunit.dataset.IDataSet;
-import org.dbunit.dataset.ITable;
-import org.dbunit.dataset.ReplacementDataSet;
+import org.dbunit.dataset.*;
import org.dbunit.dataset.filter.DefaultColumnFilter;
import org.dbunit.dataset.xml.FlatXmlDataSet;
import org.junit.Assert;
connection.close();
}
} catch (SQLException e) {
+ // ignore
}
}
}
}
+ protected void checkTable(String testName, String table, String... columns) {
+ IDatabaseConnection connection = null;
+ try {
+ connection = createConnection();
+
+ IDataSet dataSet = connection.createDataSet();
+ IDataSet expectedDataSet = getExpectedData(testName);
+ ITable filteredTable = DefaultColumnFilter.includedColumnsTable(dataSet.getTable(table), columns);
+ ITable filteredExpectedTable = DefaultColumnFilter.includedColumnsTable(expectedDataSet.getTable(table), columns);
+ Assertion.assertEquals(filteredExpectedTable, filteredTable);
+ } catch (DatabaseUnitException e) {
+ fail(e.getMessage());
+ } catch (SQLException e) {
+ throw translateException("Error while checking results", e);
+ } finally {
+ closeQuietly(connection);
+ }
+ }
+
protected void assertEmptyTables(String... emptyTables) {
IDatabaseConnection connection = null;
try {
}
}
+ protected void checkTable(String testName, String table, String... columns) {
+ IDatabaseConnection connection = null;
+ try {
+ connection = createConnection();
+
+ IDataSet dataSet = connection.createDataSet();
+ IDataSet expectedDataSet = getExpectedData(testName);
+ ITable filteredTable = DefaultColumnFilter.includedColumnsTable(dataSet.getTable(table), columns);
+ ITable filteredExpectedTable = DefaultColumnFilter.includedColumnsTable(expectedDataSet.getTable(table), columns);
+ Assertion.assertEquals(filteredExpectedTable, filteredTable);
+
+ } catch (DatabaseUnitException e) {
+ fail(e.getMessage());
+ } catch (SQLException e) {
+ throw translateException("Error while checking results", e);
+ } finally {
+ closeQuietly(connection);
+ }
+ }
+
private IDatabaseConnection createConnection() {
try {
IDatabaseConnection connection = databaseTester.getConnection();