ShutdownTask bug fix

This commit is contained in:
Ashwin Giridharan 2020-06-24 13:07:42 -07:00
parent bd6a7d8b90
commit b636edd007

View file

@ -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());