commit b1ac8e97ed902fffa082caa54eac4fc628d8d137
author Lokesh Jain <ljain@apache.org> Wed Sep 04 10:56:37 2024 +0530
committer GitHub <noreply@github.com> Tue Sep 03 22:26:37 2024 -0700
tree 715dae22d6a123f4b596feb813e7465cbb9c884f
parent f64ff333369bb7de3a78afd28263f162262773c8

[MINOR] Add backward compatibility test for HoodieTableConfig partition field APIs (#11824)

hudi-hadoop-common/src/test/java/org/apache/hudi/common/table/TestHoodieTableConfig.java

diff --git a/hudi-hadoop-common/src/test/java/org/apache/hudi/common/table/TestHoodieTableConfig.java b/hudi-hadoop-common/src/test/java/org/apache/hudi/common/table/TestHoodieTableConfig.java
index 3490bc7..e968855 100644
--- a/hudi-hadoop-common/src/test/java/org/apache/hudi/common/table/TestHoodieTableConfig.java
+++ b/hudi-hadoop-common/src/test/java/org/apache/hudi/common/table/TestHoodieTableConfig.java
@@ -220,16 +220,17 @@
     assertEquals("p1,p2", config.getPartitionFieldProp());
   }
 
-  @Test
-  public void testPartitionFieldAPIs() {
+  @ParameterizedTest
+  @ValueSource(strings = {"p1:simple,p2:timestamp", "p1,p2"})
+  public void testPartitionFieldAPIs(String partitionFields) {
     Properties updatedProps = new Properties();
-    updatedProps.setProperty(HoodieTableConfig.PARTITION_FIELDS.key(), "p1:simple,p2:timestamp");
+    updatedProps.setProperty(HoodieTableConfig.PARTITION_FIELDS.key(), partitionFields);
     HoodieTableConfig.update(storage, metaPath, updatedProps);
 
     HoodieTableConfig config = new HoodieTableConfig(storage, metaPath, null, null);
-    assertEquals("p1:simple,p2:timestamp", HoodieTableConfig.getPartitionFieldPropForKeyGenerator(config).get());
+    assertEquals(partitionFields, HoodieTableConfig.getPartitionFieldPropForKeyGenerator(config).get());
     assertEquals("p1,p2", HoodieTableConfig.getPartitionFieldProp(config).get());
     assertArrayEquals(new String[] {"p1", "p2"}, HoodieTableConfig.getPartitionFields(config).get());
-    assertEquals("p1", HoodieTableConfig.getPartitionFieldWithoutKeyGenPartitionType("p1:simple", config));
+    assertEquals("p1", HoodieTableConfig.getPartitionFieldWithoutKeyGenPartitionType(partitionFields.split(",")[0], config));
   }
 }