Use the actual number of messages

apache · Sep 28, 2021 · 5fdcbc3 · 5fdcbc3
1 parent 2bed74b
commit 5fdcbc3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pulsar-broker/src/test/java/org/apache/pulsar/client/converter/CustomBatchConverter.java b/pulsar-broker/src/test/java/org/apache/pulsar/client/converter/CustomBatchConverter.java
@@ -48,7 +48,7 @@ public <T> Iterable<Message<T>> convert(EntryContext context, MessagePayload pay
         bufList.add(buf);
         final CustomBatchFormat.StringIterable strings = CustomBatchFormat.deserialize(buf);
         final Iterator<String> stringIterator = strings.iterator();
-        final int numMessages = context.getNumMessages();
+        final int numMessages = strings.size();
 
         return () -> new Iterator<Message<T>>() {
             int index = 0;