improve the test testGetTopicsOfGetTopicsResult

apache · May 29, 2024 · d942001 · d942001
1 parent a802298
commit d942001
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/pulsar-broker/src/test/java/org/apache/pulsar/client/impl/LookupServiceTest.java b/pulsar-broker/src/test/java/org/apache/pulsar/client/impl/LookupServiceTest.java
@@ -105,6 +105,20 @@ public void testGetTopicsOfGetTopicsResult(boolean isUsingHttpLookup) throws Exc
         assertTrue(topics.contains(partitionedTopic));
         assertFalse(topics.contains(nonPersistentTopic));
         assertFalse(topics.contains(TopicName.get(partitionedTopic).getPartition(0).toString()));
+        // Verify the new method "GetTopicsResult.nonPartitionedOrPartitionTopics" works as expected.
+        Collection<String> nonPartitionedOrPartitionTopics =
+                lookupService.getTopicsUnderNamespace(NamespaceName.get("public/default"),
+                Mode.PERSISTENT, "public/default/.*", null).join()
+                .getNonPartitionedOrPartitionTopics();
+        assertTrue(nonPartitionedOrPartitionTopics.contains(nonPartitionedTopic));
+        assertFalse(nonPartitionedOrPartitionTopics.contains(partitionedTopic));
+        assertFalse(nonPartitionedOrPartitionTopics.contains(nonPersistentTopic));
+        assertTrue(nonPartitionedOrPartitionTopics.contains(TopicName.get(partitionedTopic).getPartition(0)
+                .toString()));
+        assertTrue(nonPartitionedOrPartitionTopics.contains(TopicName.get(partitionedTopic).getPartition(1)
+                .toString()));
+        assertTrue(nonPartitionedOrPartitionTopics.contains(TopicName.get(partitionedTopic).getPartition(2)
+                .toString()));
 
         // Cleanup.
         admin.topics().deletePartitionedTopic(partitionedTopic, false);