Fix test case

apache · Nov 2, 2023 · 3ba231d · 3ba231d
1 parent b33f866
commit 3ba231d
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/...java/org/apache/shardingsphere/data/pipeline/mysql/ingest/MySQLIncrementalDumperTest.java b/...java/org/apache/shardingsphere/data/pipeline/mysql/ingest/MySQLIncrementalDumperTest.java
@@ -55,11 +55,11 @@
 import java.sql.SQLException;
 import java.sql.Statement;
 import java.sql.Types;
+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;
 
@@ -137,7 +137,7 @@ void assertWriteRowsEventWithCustomColumns() throws ReflectiveOperationException
         assertWriteRowsEvent0(mockTargetTableColumnsMap(), 1);
     }
 
-    private void assertWriteRowsEvent0(final Map<LogicTableName, Set<ColumnName>> targetTableColumnsMap, final int expectedColumnCount) throws ReflectiveOperationException {
+    private void assertWriteRowsEvent0(final Map<LogicTableName, Collection<ColumnName>> targetTableColumnsMap, final int expectedColumnCount) throws ReflectiveOperationException {
         dumperConfig.setTargetTableColumnsMap(targetTableColumnsMap);
         WriteRowsEvent rowsEvent = new WriteRowsEvent();
         rowsEvent.setDatabaseName("");
@@ -151,7 +151,7 @@ private void assertWriteRowsEvent0(final Map<LogicTableName, Set<ColumnName>> ta
         assertThat(((DataRecord) actual.get(0)).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")));
     }
 
@@ -165,7 +165,7 @@ void assertUpdateRowsEventWithCustomColumns() throws ReflectiveOperationExceptio
         assertUpdateRowsEvent0(mockTargetTableColumnsMap(), 1);
     }
 
-    private void assertUpdateRowsEvent0(final Map<LogicTableName, Set<ColumnName>> targetTableColumnsMap, final int expectedColumnCount) throws ReflectiveOperationException {
+    private void assertUpdateRowsEvent0(final Map<LogicTableName, Collection<ColumnName>> targetTableColumnsMap, final int expectedColumnCount) throws ReflectiveOperationException {
         dumperConfig.setTargetTableColumnsMap(targetTableColumnsMap);
         UpdateRowsEvent rowsEvent = new UpdateRowsEvent();
         rowsEvent.setDatabaseName("test");
@@ -190,7 +190,7 @@ void assertDeleteRowsEventWithCustomColumns() throws ReflectiveOperationExceptio
         assertDeleteRowsEvent0(mockTargetTableColumnsMap(), 1);
     }
 
-    private void assertDeleteRowsEvent0(final Map<LogicTableName, Set<ColumnName>> targetTableColumnsMap, final int expectedColumnCount) throws ReflectiveOperationException {
+    private void assertDeleteRowsEvent0(final Map<LogicTableName, Collection<ColumnName>> targetTableColumnsMap, final int expectedColumnCount) throws ReflectiveOperationException {
         dumperConfig.setTargetTableColumnsMap(targetTableColumnsMap);
         DeleteRowsEvent rowsEvent = new DeleteRowsEvent();
         rowsEvent.setDatabaseName("");