Updated AsyncS3ObjectIterator to improve code coverage

Author: aneesh-neelam

build.gradle

@@ -14,7 +14,7 @@ ext {
 }
 
 group = 'me.aneeshneelam'
-version = '1.0'
+version = '1.1'
 
 java {
     sourceCompatibility JavaVersion.VERSION_1_8

src/main/java/me/aneeshneelam/lib/aws/s3/v2/AsyncS3ObjectIterator.java

@@ -62,21 +62,21 @@ private void setListObjectsV2ResponsePaginationState(ListObjectsV2Response listO
 
     @Override
     public boolean hasNext() {
-        try {
-            return this.checkListObjectsV2ResponseState()
-                    .thenApply(aVoid -> this.s3ObjectIterator.hasNext())
-                    .get(requestTimeoutDuration.getSeconds(), TimeUnit.SECONDS);
-        } catch (InterruptedException | ExecutionException | TimeoutException e) {
-            throw new RuntimeException(e);
-        }
+        CompletableFuture<Boolean> booleanCompletableFuture = this.checkListObjectsV2ResponseState()
+                .thenApply(aVoid -> this.s3ObjectIterator.hasNext());
+        return this.completableFutureGet(booleanCompletableFuture);
     }
 
     @Override
     public S3Object next() {
+        CompletableFuture<S3Object> s3ObjectCompletableFuture = this.checkListObjectsV2ResponseState()
+                .thenApply(aVoid -> this.s3ObjectIterator.next());
+        return this.completableFutureGet(s3ObjectCompletableFuture);
+    }
+
+    private <T> T completableFutureGet(CompletableFuture<T> completableFuture) {
         try {
-            return this.checkListObjectsV2ResponseState()
-                    .thenApply(aVoid -> this.s3ObjectIterator.next())
-                    .get(requestTimeoutDuration.getSeconds(), TimeUnit.SECONDS);
+            return completableFuture.get(this.requestTimeoutDuration.getSeconds(), TimeUnit.SECONDS);
         } catch (InterruptedException | ExecutionException | TimeoutException e) {
             throw new RuntimeException(e);
         }