From cb8b7222653f3404336d54db4c87957b40a866f3 Mon Sep 17 00:00:00 2001
From: wangjunbo <wangjunbo@qiyi.com>
Date: Thu, 25 Jul 2024 09:35:44 +0800
Subject: [PATCH] optimize unit test

---
 .../mysql/source/MySqlDataSourceFactoryTest.java           | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/flink-cdc-connect/flink-cdc-pipeline-connectors/flink-cdc-pipeline-connector-mysql/src/test/java/org/apache/flink/cdc/connectors/mysql/source/MySqlDataSourceFactoryTest.java b/flink-cdc-connect/flink-cdc-pipeline-connectors/flink-cdc-pipeline-connector-mysql/src/test/java/org/apache/flink/cdc/connectors/mysql/source/MySqlDataSourceFactoryTest.java
index d4f1d8f6700..c7b4424807d 100644
--- a/flink-cdc-connect/flink-cdc-pipeline-connectors/flink-cdc-pipeline-connector-mysql/src/test/java/org/apache/flink/cdc/connectors/mysql/source/MySqlDataSourceFactoryTest.java
+++ b/flink-cdc-connect/flink-cdc-pipeline-connectors/flink-cdc-pipeline-connector-mysql/src/test/java/org/apache/flink/cdc/connectors/mysql/source/MySqlDataSourceFactoryTest.java
@@ -253,13 +253,17 @@ 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()
@@ -267,6 +271,7 @@ public void testAddChunkKeyColumns() {
                         new HashMap<ObjectPath, String>() {
                             {
                                 put(multiMaxTable, "order_id");
+                                put(productsTable, "id");
                             }
                         });
     }