Fixed failing unit test

awslabs · Oct 5, 2021 · b72b9b0 · b72b9b0
1 parent b5801ed
commit b72b9b0
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 3 deletions.
diff --git a/...ent/src/main/java/software/amazon/kinesis/retrieval/polling/PrefetchRecordsPublisher.java b/...ent/src/main/java/software/amazon/kinesis/retrieval/polling/PrefetchRecordsPublisher.java
@@ -278,7 +278,6 @@ public void shutdown() {
  // (Re-)Cancel if current thread also interrupted
  executorService.shutdownNow();
  }
- executorService.shutdownNow();
  started = false;
  }
 

diff --git a/...va/software/amazon/kinesis/retrieval/polling/PrefetchRecordsPublisherIntegrationTest.java b/...va/software/amazon/kinesis/retrieval/polling/PrefetchRecordsPublisherIntegrationTest.java
@@ -254,7 +254,7 @@ private RecordsRetrieved evictPublishedEvent(PrefetchRecordsPublisher publisher,
  public void shutdown() {
  getRecordsCache.shutdown();
  sleep(100L);
- verify(executorService).shutdownNow();
+ verify(executorService).shutdown();
 // verify(getRecordsRetrievalStrategy).shutdown();
  }
 

diff --git a/...src/test/java/software/amazon/kinesis/retrieval/polling/PrefetchRecordsPublisherTest.java b/...src/test/java/software/amazon/kinesis/retrieval/polling/PrefetchRecordsPublisherTest.java
@@ -770,7 +770,7 @@ public void onNext(RecordsRetrieved recordsRetrieved) {
  @After
  public void shutdown() {
  getRecordsCache.shutdown();
- verify(executorService).shutdownNow();
+ verify(executorService).shutdown();
  }
 
  private void sleep(long millis) {