From b636edd007ed5a37e93d403ca063a252b3c25dbe Mon Sep 17 00:00:00 2001
From: Ashwin Giridharan <girida@amazon.com>
Date: Wed, 24 Jun 2020 13:07:42 -0700
Subject: [PATCH] ShutdownTask bug fix

---
 .../software/amazon/kinesis/lifecycle/ShutdownTaskTest.java     | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/amazon-kinesis-client/src/test/java/software/amazon/kinesis/lifecycle/ShutdownTaskTest.java b/amazon-kinesis-client/src/test/java/software/amazon/kinesis/lifecycle/ShutdownTaskTest.java
index 5920646c..d5af6627 100644
--- a/amazon-kinesis-client/src/test/java/software/amazon/kinesis/lifecycle/ShutdownTaskTest.java
+++ b/amazon-kinesis-client/src/test/java/software/amazon/kinesis/lifecycle/ShutdownTaskTest.java
@@ -153,7 +153,7 @@ public class ShutdownTaskTest {
 
         final TaskResult result = task.call();
         assertNotNull(result.getException());
-        assertTrue(result.getException() instanceof KinesisClientLibIOException);
+        assertTrue(result.getException() instanceof IllegalStateException);
         verify(recordsPublisher, never()).shutdown();
         verify(shardRecordProcessor, never()).shardEnded(ShardEndedInput.builder().checkpointer(recordProcessorCheckpointer).build());
         verify(shardRecordProcessor, never()).leaseLost(LeaseLostInput.builder().build());