From b5228e80d85312ba06faae03481f63ea14e1411b Mon Sep 17 00:00:00 2001
From: Brendan Lynch <brenplyn@amazon.com>
Date: Wed, 11 Oct 2023 12:58:30 -0700
Subject: [PATCH] Updated guava version to 32.1.1 and added backoff logic for
 ShardSync Integration test

---
 pom.xml                                               |  2 +-
 .../lib/worker/ShardSyncTaskIntegrationTest.java      | 11 +++++++++--
 2 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/pom.xml b/pom.xml
index d00dc58c..c0ee0e8f 100644
--- a/pom.xml
+++ b/pom.xml
@@ -55,7 +55,7 @@
     <dependency>
       <groupId>com.google.guava</groupId>
       <artifactId>guava</artifactId>
-      <version>31.0.1-jre</version>
+      <version>32.1.1-jre</version>
     </dependency>
     <dependency>
       <groupId>com.google.protobuf</groupId>
diff --git a/src/test/java/com/amazonaws/services/kinesis/clientlibrary/lib/worker/ShardSyncTaskIntegrationTest.java b/src/test/java/com/amazonaws/services/kinesis/clientlibrary/lib/worker/ShardSyncTaskIntegrationTest.java
index a4afa053..3df67220 100644
--- a/src/test/java/com/amazonaws/services/kinesis/clientlibrary/lib/worker/ShardSyncTaskIntegrationTest.java
+++ b/src/test/java/com/amazonaws/services/kinesis/clientlibrary/lib/worker/ShardSyncTaskIntegrationTest.java
@@ -42,6 +42,7 @@ import com.amazonaws.services.kinesis.leases.impl.KinesisClientLease;
 import com.amazonaws.services.kinesis.leases.impl.KinesisClientLeaseManager;
 import com.amazonaws.services.kinesis.leases.interfaces.IKinesisClientLeaseManager;
 import com.amazonaws.services.kinesis.model.StreamStatus;
+import com.amazonaws.services.kinesis.model.LimitExceededException;
 
 import static junit.framework.TestCase.fail;
 
@@ -58,6 +59,8 @@ public class ShardSyncTaskIntegrationTest {
     private IKinesisProxy kinesisProxy;
     private final KinesisShardSyncer shardSyncer = new KinesisShardSyncer(new KinesisLeaseCleanupValidator());
 
+    private static final int retryBackoffMillis = 1000;
+
     /**
      * @throws java.lang.Exception
      */
@@ -71,9 +74,13 @@ public class ShardSyncTaskIntegrationTest {
         } catch (AmazonServiceException ase) {
 
         }
-        StreamStatus status;
+        StreamStatus status = null;
         do {
-            status = StreamStatus.fromValue(kinesis.describeStream(STREAM_NAME).getStreamDescription().getStreamStatus());
+            try {
+                status = StreamStatus.fromValue(kinesis.describeStream(STREAM_NAME).getStreamDescription().getStreamStatus());
+            } catch (LimitExceededException e) {
+                Thread.sleep(retryBackoffMillis + (long) (Math.random() * 100));
+            }
         } while (status != StreamStatus.ACTIVE);
 
     }