Use captors correctly in tests

Author: ivanyu

storage/s3/src/test/java/io/aiven/kafka/tieredstorage/storage/s3/S3MultiPartOutputStreamTest.java

@@ -41,7 +41,6 @@
 import static org.assertj.core.api.Assertions.assertThat;
 import static org.assertj.core.api.Assertions.assertThatThrownBy;
 import static org.mockito.ArgumentMatchers.any;
-import static org.mockito.Mockito.doNothing;
 import static org.mockito.Mockito.never;
 import static org.mockito.Mockito.times;
 import static org.mockito.Mockito.verify;
@@ -72,7 +71,6 @@ class S3MultiPartOutputStreamTest {
     void sendAbortForAnyExceptionWhileWriting() {
         when(mockedS3.initiateMultipartUpload(any(InitiateMultipartUploadRequest.class)))
             .thenReturn(newInitiateMultipartUploadResult());
-        doNothing().when(mockedS3).abortMultipartUpload(abortMultipartUploadRequestCaptor.capture());
 
         final RuntimeException testException = new RuntimeException("test");
         when(mockedS3.uploadPart(any(UploadPartRequest.class)))
@@ -87,7 +85,7 @@ void sendAbortForAnyExceptionWhileWriting() {
 
         verify(mockedS3).initiateMultipartUpload(any(InitiateMultipartUploadRequest.class));
         verify(mockedS3).uploadPart(any(UploadPartRequest.class));
-        verify(mockedS3).abortMultipartUpload(abortMultipartUploadRequestCaptor.getValue());
+        verify(mockedS3).abortMultipartUpload(abortMultipartUploadRequestCaptor.capture());
 
         assertAbortMultipartUploadRequest(abortMultipartUploadRequestCaptor.getValue());
     }
@@ -96,7 +94,6 @@ void sendAbortForAnyExceptionWhileWriting() {
     void sendAbortForAnyExceptionWhenClose() throws Exception {
         when(mockedS3.initiateMultipartUpload(any(InitiateMultipartUploadRequest.class)))
             .thenReturn(newInitiateMultipartUploadResult());
-        doNothing().when(mockedS3).abortMultipartUpload(abortMultipartUploadRequestCaptor.capture());
 
         when(mockedS3.uploadPart(any(UploadPartRequest.class)))
             .thenThrow(RuntimeException.class);
@@ -113,7 +110,7 @@ void sendAbortForAnyExceptionWhenClose() throws Exception {
             .isInstanceOf(RuntimeException.class);
 
         verify(mockedS3, never()).completeMultipartUpload(any(CompleteMultipartUploadRequest.class));
-        verify(mockedS3).abortMultipartUpload(abortMultipartUploadRequestCaptor.getValue());
+        verify(mockedS3).abortMultipartUpload(abortMultipartUploadRequestCaptor.capture());
 
         assertAbortMultipartUploadRequest(abortMultipartUploadRequestCaptor.getValue());
     }
@@ -122,18 +119,18 @@ void sendAbortForAnyExceptionWhenClose() throws Exception {
     void writesOneByte() throws Exception {
         when(mockedS3.initiateMultipartUpload(any(InitiateMultipartUploadRequest.class)))
             .thenReturn(newInitiateMultipartUploadResult());
-        when(mockedS3.uploadPart(uploadPartRequestCaptor.capture()))
+        when(mockedS3.uploadPart(any(UploadPartRequest.class)))
             .thenReturn(newUploadPartResult(1, "SOME_ETAG"));
-        when(mockedS3.completeMultipartUpload(completeMultipartUploadRequestCaptor.capture()))
+        when(mockedS3.completeMultipartUpload(any(CompleteMultipartUploadRequest.class)))
             .thenReturn(new CompleteMultipartUploadResult());
 
         try (final S3MultiPartOutputStream out = new S3MultiPartOutputStream(BUCKET_NAME, FILE_KEY, 100, mockedS3)) {
             out.write(1);
         }
 
         verify(mockedS3).initiateMultipartUpload(any(InitiateMultipartUploadRequest.class));
-        verify(mockedS3).uploadPart(any(UploadPartRequest.class));
-        verify(mockedS3).completeMultipartUpload(any(CompleteMultipartUploadRequest.class));
+        verify(mockedS3).uploadPart(uploadPartRequestCaptor.capture());
+        verify(mockedS3).completeMultipartUpload(completeMultipartUploadRequestCaptor.capture());
 
         assertUploadPartRequest(
             uploadPartRequestCaptor.getValue(),
@@ -153,12 +150,12 @@ void writesMultipleMessages() throws Exception {
 
         when(mockedS3.initiateMultipartUpload(any(InitiateMultipartUploadRequest.class)))
             .thenReturn(newInitiateMultipartUploadResult());
-        when(mockedS3.uploadPart(uploadPartRequestCaptor.capture()))
+        when(mockedS3.uploadPart(any(UploadPartRequest.class)))
             .thenAnswer(a -> {
                 final UploadPartRequest up = a.getArgument(0);
                 return newUploadPartResult(up.getPartNumber(), "SOME_TAG#" + up.getPartNumber());
             });
-        when(mockedS3.completeMultipartUpload(completeMultipartUploadRequestCaptor.capture()))
+        when(mockedS3.completeMultipartUpload(any(CompleteMultipartUploadRequest.class)))
             .thenReturn(new CompleteMultipartUploadResult());
 
         final List<byte[]> expectedMessagesList = new ArrayList<>();
@@ -172,8 +169,8 @@ void writesMultipleMessages() throws Exception {
         }
 
         verify(mockedS3).initiateMultipartUpload(any(InitiateMultipartUploadRequest.class));
-        verify(mockedS3, times(3)).uploadPart(any(UploadPartRequest.class));
-        verify(mockedS3).completeMultipartUpload(any(CompleteMultipartUploadRequest.class));
+        verify(mockedS3, times(3)).uploadPart(uploadPartRequestCaptor.capture());
+        verify(mockedS3).completeMultipartUpload(completeMultipartUploadRequestCaptor.capture());
 
         final List<UploadPartRequest> uploadRequests = uploadPartRequestCaptor.getAllValues();
         int counter = 0;
@@ -208,9 +205,9 @@ void writesTailMessages() throws Exception {
                 up.setInputStream(new ByteArrayInputStream(up.getInputStream().readAllBytes()));
                 uploadPartRequests.add(up);
 
-                return newUploadPartResult(up.getPartNumber(), "SOME_TAG#" + up.getPartNumber());
+                return newUploadPartResult(up.getPartNumber(), "SOME_ETAG#" + up.getPartNumber());
             });
-        when(mockedS3.completeMultipartUpload(completeMultipartUploadRequestCaptor.capture()))
+        when(mockedS3.completeMultipartUpload(any(CompleteMultipartUploadRequest.class)))
             .thenReturn(new CompleteMultipartUploadResult());
 
         final byte[] message = new byte[messageSize];
@@ -235,6 +232,8 @@ void writesTailMessages() throws Exception {
         verify(mockedS3).initiateMultipartUpload(any(InitiateMultipartUploadRequest.class));
         verify(mockedS3, times(2)).uploadPart(any(UploadPartRequest.class));
         verify(mockedS3).completeMultipartUpload(completeMultipartUploadRequestCaptor.capture());
+        assertCompleteMultipartUploadRequest(completeMultipartUploadRequestCaptor.getValue(),
+            List.of(new PartETag(1, "SOME_ETAG#1"), new PartETag(2, "SOME_ETAG#2")));
     }
 
     private static InitiateMultipartUploadResult newInitiateMultipartUploadResult() {