[cdc] Debezium format completes the test case (#2416)

apache · Nov 30, 2023 · 37f5351 · 37f5351
1 parent 85ef2d8
commit 37f5351
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/...est/java/org/apache/paimon/flink/action/cdc/kafka/KafkaDebeziumSyncTableActionITCase.java b/...est/java/org/apache/paimon/flink/action/cdc/kafka/KafkaDebeziumSyncTableActionITCase.java
@@ -160,7 +160,8 @@ public void testComputedColumn() throws Exception {
         kafkaConfig.put(TOPIC.key(), topic);
         KafkaSyncTableAction action =
                 syncTableActionBuilder(kafkaConfig)
-                        .withPrimaryKeys("id")
+                        .withPartitionKeys("_year")
+                        .withPrimaryKeys("id", "_year")
                         .withComputedColumnArgs("_year=year(date)")
                         .withTableConfig(getBasicTableConfig())
                         .build();
@@ -169,13 +170,15 @@ public void testComputedColumn() throws Exception {
         RowType rowType =
                 RowType.of(
                         new DataType[] {
-                            DataTypes.STRING().notNull(), DataTypes.STRING(), DataTypes.INT()
+                            DataTypes.STRING().notNull(),
+                            DataTypes.STRING(),
+                            DataTypes.INT().notNull()
                         },
                         new String[] {"id", "date", "_year"});
         waitForResult(
                 Collections.singletonList("+I[101, 2023-03-23, 2023]"),
                 getFileStoreTable(tableName),
                 rowType,
-                Collections.singletonList("id"));
+                Arrays.asList("id", "_year"));
     }
 }