Fix unit tests that expected IllegalStateException thrown

awslabs · Feb 19, 2024 · c05e67e · c05e67e
1 parent 62061a2
commit c05e67e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...src/test/java/software/amazon/kinesis/retrieval/polling/PrefetchRecordsPublisherTest.java b/...src/test/java/software/amazon/kinesis/retrieval/polling/PrefetchRecordsPublisherTest.java
@@ -374,13 +374,13 @@ record = Record.builder().data(createByteBufferWithSize(1024)).build();
         assertTrue(spyQueue.size() <= MAX_SIZE);
     }
 
-    @Test(expected = IllegalStateException.class)
+    @Test
     public void testGetNextRecordsWithoutStarting() {
         verify(executorService, never()).execute(any());
         getRecordsCache.drainQueueForRequests();
     }
 
-    @Test(expected = IllegalStateException.class)
+    @Test
     public void testCallAfterShutdown() {
         when(executorService.isShutdown()).thenReturn(true);
         getRecordsCache.drainQueueForRequests();