1313import static org .junit .Assert .assertEquals ;
1414import static org .junit .Assert .assertThrows ;
1515import static org .junit .jupiter .api .Assertions .assertNotNull ;
16- import static org .mockito .ArgumentMatchers .any ;
1716import static org .mockito .Mockito .mock ;
1817import static org .mockito .Mockito .verify ;
1918import static org .mockito .Mockito .when ;
@@ -39,9 +38,9 @@ void setUp() {
3938
4039 @ Test
4140 void expectExceptionWhenS3ClientCantDeliverResponse () {
42- s3StorageConnector = new S3StorageConnector (S3Client .builder ().build (), config );
41+ S3StorageConnector storageConnector = new S3StorageConnector (S3Client .builder ().build (), config );
4342 Exception exception = assertThrows (StorageConnectorException .class ,
44- () -> s3StorageConnector .downloadFromStorage ("s3File" ));
43+ () -> storageConnector .downloadFromStorage ("s3File" ));
4544
4645 assertEquals ("Error occurred downloading from S3 Bucket" , exception .getMessage ());
4746 }
@@ -51,12 +50,13 @@ void downloadFromStorage() {
5150 var mockResponse = mock (GetObjectResponse .class );
5251 var mockInputStream = new ByteArrayInputStream ("dummy-content" .getBytes ());
5352 var mockResponseInputStream = new ResponseInputStream <>(mockResponse , mockInputStream );
53+ final var request = GetObjectRequest .builder ().bucket (config .getContainerName ()).key (FILE_NAME ).build ();
5454
55- when (mockS3Client .getObject (any ( GetObjectRequest . class ) )).thenReturn (mockResponseInputStream );
55+ when (mockS3Client .getObject (request )).thenReturn (mockResponseInputStream );
5656
5757 var result = s3StorageConnector .downloadFromStorage (FILE_NAME );
5858
5959 assertNotNull (result );
60- verify (mockS3Client ).getObject (any ( GetObjectRequest . class ) );
60+ verify (mockS3Client ).getObject (request );
6161 }
6262}
0 commit comments