Skip to content

Commit

Permalink
Refactor DumperConfiguration
Browse files Browse the repository at this point in the history
  • Loading branch information
terrymanu committed Nov 2, 2023
1 parent 79ebdaf commit 4a89b05
Showing 1 changed file with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -54,11 +54,11 @@
import java.sql.Statement;
import java.sql.Types;
import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.function.Function;
import java.util.stream.Collectors;

Expand Down Expand Up @@ -131,7 +131,7 @@ void assertWriteRowEventWithCustomColumns() throws ReflectiveOperationException
assertWriteRowEvent0(mockTargetTableColumnsMap(), 1);
}

private void assertWriteRowEvent0(final Map<LogicTableName, Set<ColumnName>> targetTableColumnsMap, final int expectedColumnCount) throws ReflectiveOperationException {
private void assertWriteRowEvent0(final Map<LogicTableName, Collection<ColumnName>> targetTableColumnsMap, final int expectedColumnCount) throws ReflectiveOperationException {
dumperConfig.setTargetTableColumnsMap(targetTableColumnsMap);
WriteRowEvent rowsEvent = new WriteRowEvent();
rowsEvent.setSchemaName("");
Expand All @@ -143,7 +143,7 @@ private void assertWriteRowEvent0(final Map<LogicTableName, Set<ColumnName>> tar
assertThat(actual.getColumnCount(), is(expectedColumnCount));
}

private Map<LogicTableName, Set<ColumnName>> mockTargetTableColumnsMap() {
private Map<LogicTableName, Collection<ColumnName>> mockTargetTableColumnsMap() {
return Collections.singletonMap(new LogicTableName("t_order"), Collections.singleton(new ColumnName("order_id")));
}

Expand Down

0 comments on commit 4a89b05

Please sign in to comment.