Java IColumnFilter类代码示例

import org.dbunit.dataset.filter.IColumnFilter; //导入依赖的package包/类
/**
 * Configure database.
 *
 * @param config the database-config.
 */
private void setUpDatabaseConfig(DatabaseConfig config) {
    config.setProperty(DatabaseConfig.PROPERTY_BATCH_SIZE, BATCH_SIZE);
    config.setProperty(DatabaseConfig.FEATURE_BATCHED_STATEMENTS,
            Boolean.TRUE);
    config.setProperty(DatabaseConfig.PROPERTY_PRIMARY_KEY_FILTER,
            new IColumnFilter() {

                @Override
                public boolean accept(String tableName, Column column) {
                    if (tableToPrimaryKey.containsKey(tableName)) {
                        return Arrays.asList(
                                tableToPrimaryKey.get(tableName)).contains(
                                column.getColumnName());
                    } else {
                        return column.getColumnName()
                                .equalsIgnoreCase("id");
                    }
                }
            });
}
 

import org.dbunit.dataset.filter.IColumnFilter; //导入依赖的package包/类
public DataSetComparator(final String[] orderBy, final String[] toExclude, final boolean isStrict,
        final Set<Class<? extends IColumnFilter>> columnFilters) {
    this.toExclude = new ColumnsHolder(toExclude, ID_MAPPER);
    this.orderBy = new ColumnsHolder(orderBy, ID_MAPPER);
    this.isStrict = isStrict;
    this.columnFilters = columnFilters;
}
 

import org.dbunit.dataset.filter.IColumnFilter; //导入依赖的package包/类
private ITable applyCustomFilters(final ITable table) throws DataSetException {
    ITable compositeTable = table;
    for (final Class<? extends IColumnFilter> columnFilter : columnFilters) {
        IColumnFilter customColumnFilter;
        try {
            customColumnFilter = columnFilter.newInstance();
        } catch (InstantiationException | IllegalAccessException e) {
            throw new JpaUnitException("Could not instanciate custom column filter", e);
        }
        final FilteredTableMetaData metaData = new FilteredTableMetaData(compositeTable.getTableMetaData(), customColumnFilter);
        compositeTable = new CompositeTable(metaData, compositeTable);
    }
    return compositeTable;
}
 

import org.dbunit.dataset.filter.IColumnFilter; //导入依赖的package包/类
@Test
public void testCurrentDataSetContainsDataAndExpectedDataSetIsEmpty() throws Exception {
    // GIVEN
    final String[] orderBy = new String[] {};
    final String[] toExclude = new String[] {};
    final Set<Class<? extends IColumnFilter>> columnFilters = Collections.emptySet();
    final DataSetComparator comparator = new DataSetComparator(orderBy, toExclude, false, columnFilters);

    when(currentDataSet.getTableNames()).thenReturn(new String[] {
            TABLE_1_NAME
    });
    when(currentDataSet.getTable(any(String.class))).thenAnswer((final InvocationOnMock invocation) -> {
        final String tableName = (String) invocation.getArguments()[0];
        if (tableName == TABLE_1_NAME) {
            return table1;
        } else {
            throw new NoSuchTableException(tableName);
        }
    });

    when(expectedDataSet.getTableNames()).thenReturn(new String[] {});
    when(expectedDataSet.getTable(any(String.class))).thenThrow(new NoSuchTableException());

    // WHEN
    comparator.compare(currentDataSet, expectedDataSet, errorCollector);

    // THEN
    assertThat(errorCollector.amountOfErrors(), equalTo(1));

    try {
        errorCollector.report();
        fail("AssertionError expected");
    } catch (final AssertionError e) {
        assertThat(e.getMessage(), containsString("failed in 1 case"));
        assertThat(e.getMessage(),
                containsString(TABLE_1_NAME + " was expected to be empty, but has <" + TABLE_1_ENTRIES + "> entries"));
    }
}
 

import org.dbunit.dataset.filter.IColumnFilter; //导入依赖的package包/类
@Test
public void testCurrentDataSetIsEmptyAndExpectedDataContainsData() throws Exception {
    // GIVEN
    final String[] orderBy = new String[] {};
    final String[] toExclude = new String[] {};
    final Set<Class<? extends IColumnFilter>> columnFilters = Collections.emptySet();
    final DataSetComparator comparator = new DataSetComparator(orderBy, toExclude, false, columnFilters);

    when(currentDataSet.getTableNames()).thenReturn(new String[] {});
    when(currentDataSet.getTable(any(String.class))).thenThrow(new NoSuchTableException());

    when(expectedDataSet.getTableNames()).thenReturn(new String[] {
            TABLE_1_NAME
    });
    when(expectedDataSet.getTable(any(String.class))).thenAnswer((final InvocationOnMock invocation) -> {
        final String tableName = (String) invocation.getArguments()[0];
        if (tableName == TABLE_1_NAME) {
            return table1;
        } else {
            throw new NoSuchTableException(tableName);
        }
    });

    // WHEN
    comparator.compare(currentDataSet, expectedDataSet, errorCollector);

    // THEN
    assertThat(errorCollector.amountOfErrors(), equalTo(1));

    try {
        errorCollector.report();
        fail("AssertionError expected");
    } catch (final AssertionError e) {
        assertThat(e.getMessage(), containsString("failed in 1 case"));
        assertThat(e.getMessage(), containsString(
                TABLE_1_NAME + " was expected to be present and to contain <" + TABLE_1_ENTRIES + "> entries, but not found"));
    }
}
 

import org.dbunit.dataset.filter.IColumnFilter; //导入依赖的package包/类
private HashSet<Class<? extends IColumnFilter>> getColumnFilter(final ExpectedDataSets expectedDataSets) {
    final Class<? extends IColumnFilter>[] filter = (Class<? extends IColumnFilter>[]) expectedDataSets.filter();
    return filter == null ? new HashSet<>() : new HashSet<>(Arrays.asList(filter));
}