optimize unit test

apache · Jul 25, 2024 · cb8b722 · cb8b722
1 parent fb67b5c
commit cb8b722
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/...rc/test/java/org/apache/flink/cdc/connectors/mysql/source/MySqlDataSourceFactoryTest.java b/...rc/test/java/org/apache/flink/cdc/connectors/mysql/source/MySqlDataSourceFactoryTest.java
@@ -253,20 +253,25 @@ public void testAddChunkKeyColumns() {
         options.put(TABLES.key(), inventoryDatabase.getDatabaseName() + ".\\.*");
         options.put(
                 SCAN_INCREMENTAL_SNAPSHOT_CHUNK_KEY_COLUMN.key(),
-                inventoryDatabase.getDatabaseName() + ".multi_max_\\.*:order_id");
+                inventoryDatabase.getDatabaseName()
+                        + ".multi_max_\\.*:order_id;"
+                        + inventoryDatabase.getDatabaseName()
+                        + ".products:id;");
         Factory.Context context = new MockContext(Configuration.fromMap(options));
 
         MySqlDataSourceFactory factory = new MySqlDataSourceFactory();
         MySqlDataSource dataSource = (MySqlDataSource) factory.createDataSource(context);
         ObjectPath multiMaxTable =
                 new ObjectPath(inventoryDatabase.getDatabaseName(), "multi_max_table");
+        ObjectPath productsTable = new ObjectPath(inventoryDatabase.getDatabaseName(), "products");
 
         assertThat(dataSource.getSourceConfig().getChunkKeyColumns())
                 .isNotEmpty()
                 .isEqualTo(
                         new HashMap<ObjectPath, String>() {
                             {
                                 put(multiMaxTable, "order_id");
+                                put(productsTable, "id");
                             }
                         });
     }