Use 1 initial credit in test

rabbitmq · Dec 11, 2024 · 3d6b1dc · 3d6b1dc
1 parent fbdd7f2
commit 3d6b1dc
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/src/test/java/com/rabbitmq/stream/impl/StreamConsumerTest.java b/src/test/java/com/rabbitmq/stream/impl/StreamConsumerTest.java
@@ -203,7 +203,7 @@ void consumeWithAsyncConsumerFlowControl() throws Exception {
         environment.consumerBuilder().stream(stream)
             .offset(OffsetSpecification.first())
             .flow()
-            .strategy(creditWhenHalfMessagesProcessed())
+            .strategy(creditWhenHalfMessagesProcessed(1))
             .builder();
 
     List<MessageHandler.Context> messageContexts = synchronizedList(new ArrayList<>());
@@ -244,14 +244,13 @@ void asynchronousProcessingWithFlowControl() {
     int messageCount = 100_000;
     publishAndWaitForConfirms(cf, messageCount, stream);
 
-    ExecutorService executorService =
-        Executors.newFixedThreadPool(getRuntime().availableProcessors());
-    try {
+    try (ExecutorService executorService =
+        Executors.newFixedThreadPool(getRuntime().availableProcessors())) {
       CountDownLatch latch = new CountDownLatch(messageCount);
       environment.consumerBuilder().stream(stream)
           .offset(OffsetSpecification.first())
           .flow()
-          .strategy(creditWhenHalfMessagesProcessed())
+          .strategy(creditWhenHalfMessagesProcessed(1))
           .builder()
           .messageHandler(
               (ctx, message) ->
@@ -262,8 +261,6 @@ void asynchronousProcessingWithFlowControl() {
                       }))
           .build();
       assertThat(latch).is(completed());
-    } finally {
-      executorService.shutdownNow();
     }
   }