add escapePathName units

apache · Oct 9, 2021 · 4c92835 · 4c92835
1 parent 0807e52
commit 4c92835
Show file tree

Hide file tree

Showing 2 changed files with 46 additions and 1 deletion.
diff --git a/...ble/flink-table-common/src/main/java/org/apache/flink/table/utils/PartitionPathUtils.java b/...ble/flink-table-common/src/main/java/org/apache/flink/table/utils/PartitionPathUtils.java
@@ -108,7 +108,7 @@ public static String generatePartitionPath(LinkedHashMap<String, String> partiti
      * @param path The path to escape.
      * @return An escaped path name.
      */
-    private static String escapePathName(String path) {
+    static String escapePathName(String path) {
         if (path == null || path.length() == 0) {
             throw new TableException("Path should not be null or empty: " + path);
         }

diff --git a/...flink-table-common/src/test/java/org/apache/flink/table/utils/PartitionPathUtilsTest.java b/...flink-table-common/src/test/java/org/apache/flink/table/utils/PartitionPathUtilsTest.java
@@ -33,4 +33,49 @@ public void testEscapeChar() {
             assertEquals(expected, actual);
         }
     }
+
+    @Test
+    public void testEscapePathNameWithHeadControl() {
+        String origin = "[00";
+        String expected = "%5B00";
+        String actual = PartitionPathUtils.escapePathName(origin);
+        assertEquals(expected, actual);
+        assertEquals(origin, PartitionPathUtils.unescapePathName(actual));
+    }
+
+    @Test
+    public void testEscapePathNameWithTailControl() {
+        String origin = "00]";
+        String expected = "00%5D";
+        String actual = PartitionPathUtils.escapePathName(origin);
+        assertEquals(expected, actual);
+        assertEquals(origin, PartitionPathUtils.unescapePathName(actual));
+    }
+
+    @Test
+    public void testEscapePathNameWithMidControl() {
+        String origin = "00:00";
+        String expected = "00%3A00";
+        String actual = PartitionPathUtils.escapePathName(origin);
+        assertEquals(expected, actual);
+        assertEquals(origin, PartitionPathUtils.unescapePathName(actual));
+    }
+
+    @Test
+    public void testEscapePathName() {
+        String origin = "[00:00]";
+        String expected = "%5B00%3A00%5D";
+        String actual = PartitionPathUtils.escapePathName(origin);
+        assertEquals(expected, actual);
+        assertEquals(origin, PartitionPathUtils.unescapePathName(actual));
+    }
+
+    @Test
+    public void testEscapePathNameWithoutControl() {
+        String origin = "0000";
+        String expected = "0000";
+        String actual = PartitionPathUtils.escapePathName(origin);
+        assertEquals(expected, actual);
+        assertEquals(origin, PartitionPathUtils.unescapePathName(actual));
+    }
 }