From eb07782f4f2ec100c10103c12b684adc3363c0ab Mon Sep 17 00:00:00 2001
From: Meher Mankikar <mmankika@amazon.com>
Date: Thu, 15 Jun 2023 08:27:35 -0700
Subject: [PATCH] Fixed producer cancelling before final task was finished

---
 .../test/java/software/amazon/kinesis/utils/TestConsumer.java   | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/amazon-kinesis-client/src/test/java/software/amazon/kinesis/utils/TestConsumer.java b/amazon-kinesis-client/src/test/java/software/amazon/kinesis/utils/TestConsumer.java
index 888c3ba7..f2581b39 100644
--- a/amazon-kinesis-client/src/test/java/software/amazon/kinesis/utils/TestConsumer.java
+++ b/amazon-kinesis-client/src/test/java/software/amazon/kinesis/utils/TestConsumer.java
@@ -125,7 +125,7 @@ public class TestConsumer {
          */
         log.info("Cancelling producer and shutting down executor.");
         producerFuture.cancel(false);
-        producerExecutor.shutdownNow();
+        producerExecutor.shutdown();
 
         /**
          * Wait a few seconds for the last few records to be processed