[rest] Add list partitions tests for non-metastore part table

Author: JingsongLi

paimon-core/src/test/java/org/apache/paimon/rest/RESTCatalogServer.java

@@ -107,6 +107,7 @@
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Collections;
+import java.util.Comparator;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
@@ -982,13 +983,22 @@ private MockResponse generateFinalListDatabasesResponse(
     }
 
     private <T> PagedList<T> buildPagedEntities(List<T> names, int maxResults, String pageToken) {
-        List<T> sortedNames = names.stream().sorted(this::compareTo).collect(Collectors.toList());
+        return buildPagedEntities(names, maxResults, pageToken, false);
+    }
+
+    private <T> PagedList<T> buildPagedEntities(
+            List<T> names, int maxResults, String pageToken, boolean desc) {
+        Comparator<Object> comparator = this::compareTo;
+        if (desc) {
+            comparator = comparator.reversed();
+        }
+        List<T> sortedNames = names.stream().sorted(comparator).collect(Collectors.toList());
         List<T> pagedNames = new ArrayList<>();
         for (T sortedName : sortedNames) {
             if (pagedNames.size() < maxResults) {
                 if (pageToken == null) {
                     pagedNames.add(sortedName);
-                } else if (this.compareTo(sortedName, pageToken) > 0) {
+                } else if (comparator.compare(sortedName, pageToken) > 0) {
                     pagedNames.add(sortedName);
                 }
             } else {
@@ -1455,7 +1465,7 @@ private MockResponse generateFinalListPartitionsResponse(
             String pageToken = parameters.getOrDefault(PAGE_TOKEN, null);
 
             PagedList<Partition> pagedPartitions =
-                    buildPagedEntities(partitions, maxResults, pageToken);
+                    buildPagedEntities(partitions, maxResults, pageToken, true);
             response =
                     new ListPartitionsResponse(
                             pagedPartitions.getElements(), pagedPartitions.getNextPageToken());

paimon-core/src/test/java/org/apache/paimon/rest/RESTCatalogTest.java

@@ -75,7 +75,6 @@
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Collections;
-import java.util.Comparator;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
@@ -1033,6 +1032,15 @@ void testListPartitionsFromFile() throws Exception {
 
     @Test
     void testListPartitions() throws Exception {
+        innerTestListPartitions(true);
+    }
+
+    @Test
+    void testListPartitionsNonMetastore() throws Exception {
+        innerTestListPartitions(false);
+    }
+
+    private void innerTestListPartitions(boolean metastore) throws Exception {
         if (!supportPartitions()) {
             return;
         }
@@ -1046,14 +1054,14 @@ void testListPartitions() throws Exception {
                         singletonMap("dt", "20250103"));
         Map[] sortedSpecs =
                 partitionSpecs.stream()
-                        .sorted(Comparator.comparing(i -> i.get("dt")))
+                        .sorted((o1, o2) -> o2.get("dt").compareTo(o1.get("dt")))
                         .toArray(Map[]::new);
 
-        String databaseName = "partitions_db";
+        String databaseName = "partitions_db" + metastore;
         Identifier identifier = Identifier.create(databaseName, "table");
         Schema schema =
                 Schema.newBuilder()
-                        .option(METASTORE_PARTITIONED_TABLE.key(), "true")
+                        .option(METASTORE_PARTITIONED_TABLE.key(), Boolean.toString(metastore))
                         .option(METASTORE_TAG_TO_PARTITION.key(), "dt")
                         .column("col", DataTypes.INT())
                         .column("dt", DataTypes.STRING())
@@ -1073,7 +1081,12 @@ void testListPartitions() throws Exception {
         }
 
         List<Partition> restPartitions = restCatalog.listPartitions(identifier);
-        assertThat(restPartitions.stream().map(Partition::spec)).containsExactly(sortedSpecs);
+        if (metastore) {
+            assertThat(restPartitions.stream().map(Partition::spec)).containsExactly(sortedSpecs);
+        } else {
+            assertThat(restPartitions.stream().map(Partition::spec))
+                    .containsExactlyInAnyOrder(sortedSpecs);
+        }
     }
 
     @Test
@@ -1123,34 +1136,31 @@ public void testListPartitionsPaged() throws Exception {
                 catalog.listPartitionsPaged(identifier, null, null, null);
         Map[] sortedSpecs =
                 partitionSpecs.stream()
-                        .sorted(Comparator.comparing(i -> i.get("dt")))
+                        .sorted((o1, o2) -> o2.get("dt").compareTo(o1.get("dt")))
                         .toArray(Map[]::new);
         assertPagedPartitions(pagedPartitions, partitionSpecs.size(), sortedSpecs);
 
         int maxResults = 2;
         pagedPartitions = catalog.listPartitionsPaged(identifier, maxResults, null, null);
-        assertPagedPartitions(
-                pagedPartitions, maxResults, partitionSpecs.get(2), partitionSpecs.get(0));
-        assertEquals("dt=20250101", pagedPartitions.getNextPageToken());
+        assertPagedPartitions(pagedPartitions, maxResults, sortedSpecs[0], sortedSpecs[1]);
+        assertEquals(sortedSpecs[1].toString(), "{" + pagedPartitions.getNextPageToken() + "}");
 
         pagedPartitions =
                 catalog.listPartitionsPaged(
                         identifier, maxResults, pagedPartitions.getNextPageToken(), null);
-        assertPagedPartitions(
-                pagedPartitions, maxResults, partitionSpecs.get(1), partitionSpecs.get(5));
-        assertEquals("dt=20250103", pagedPartitions.getNextPageToken());
+        assertPagedPartitions(pagedPartitions, maxResults, sortedSpecs[2], sortedSpecs[3]);
+        assertEquals(sortedSpecs[3].toString(), "{" + pagedPartitions.getNextPageToken() + "}");
 
         pagedPartitions =
                 catalog.listPartitionsPaged(
                         identifier, maxResults, pagedPartitions.getNextPageToken(), null);
-        assertPagedPartitions(
-                pagedPartitions, maxResults, partitionSpecs.get(4), partitionSpecs.get(6));
-        assertEquals("dt=2025010_test", pagedPartitions.getNextPageToken());
+        assertPagedPartitions(pagedPartitions, maxResults, sortedSpecs[4], sortedSpecs[5]);
+        assertEquals(sortedSpecs[5].toString(), "{" + pagedPartitions.getNextPageToken() + "}");
 
         pagedPartitions =
                 catalog.listPartitionsPaged(
                         identifier, maxResults, pagedPartitions.getNextPageToken(), null);
-        assertPagedPartitions(pagedPartitions, 1, partitionSpecs.get(3));
+        assertPagedPartitions(pagedPartitions, 1, sortedSpecs[6]);
         assertNull(pagedPartitions.getNextPageToken());
 
         maxResults = 8;
@@ -1168,11 +1178,11 @@ public void testListPartitionsPaged() throws Exception {
         assertPagedPartitions(
                 pagedPartitions,
                 5,
-                partitionSpecs.get(0),
-                partitionSpecs.get(1),
-                partitionSpecs.get(5),
+                partitionSpecs.get(6),
                 partitionSpecs.get(4),
-                partitionSpecs.get(6));
+                partitionSpecs.get(5),
+                partitionSpecs.get(1),
+                partitionSpecs.get(0));
         assertNull(pagedPartitions.getNextPageToken());
 
         pagedPartitions = catalog.listPartitionsPaged(identifier, maxResults, null, "dt=2025010_%");
@@ -1707,7 +1717,6 @@ protected boolean supportsFormatTable() {
 
     @Override
     protected boolean supportPartitions() {
-        // TODO support this
         return true;
     }