3232import software .amazon .awssdk .core .ResponseBytes ;
3333import software .amazon .awssdk .core .async .AsyncRequestBody ;
3434import software .amazon .awssdk .core .async .AsyncResponseTransformer ;
35+ import software .amazon .awssdk .core .async .ResponsePublisher ;
3536import software .amazon .awssdk .services .s3 .internal .crt .S3CrtAsyncClient ;
3637import software .amazon .awssdk .services .s3 .model .CopyObjectRequest ;
3738import software .amazon .awssdk .services .s3 .model .CopyObjectResponse ;
4849import software .amazon .awssdk .transfer .s3 .model .DownloadDirectoryRequest ;
4950import software .amazon .awssdk .transfer .s3 .model .DownloadFileRequest ;
5051import software .amazon .awssdk .transfer .s3 .S3TransferManager ;
52+ import software .amazon .awssdk .transfer .s3 .model .DownloadRequest ;
5153import software .amazon .awssdk .transfer .s3 .model .UploadDirectoryRequest ;
5254import software .amazon .awssdk .transfer .s3 .model .UploadFileRequest ;
5355
@@ -206,13 +208,13 @@ void download_cancel_shouldForwardCancellation() {
206208 CompletableFuture <GetObjectResponse > s3CrtFuture = new CompletableFuture <>();
207209 when (mockS3Crt .getObject (any (GetObjectRequest .class ), any (AsyncResponseTransformer .class )))
208210 .thenReturn (s3CrtFuture );
211+ DownloadRequest <ResponseBytes <GetObjectResponse >> downloadRequest =
212+ DownloadRequest .builder ()
213+ .getObjectRequest (g -> g .bucket ("bucket" ).key ("key" ))
214+ .responseTransformer (AsyncResponseTransformer .toBytes ()).build ();
209215
210216 CompletableFuture <CompletedDownload <ResponseBytes <GetObjectResponse >>> future =
211- tm .download (d -> d
212- .getObjectRequest (g -> g .bucket ("bucket" )
213- .key ("key" ))
214- .responseTransformer (AsyncResponseTransformer .toBytes ()))
215- .completionFuture ();
217+ tm .download (downloadRequest ).completionFuture ();
216218 future .cancel (true );
217219 assertThat (s3CrtFuture ).isCancelled ();
218220 }
@@ -236,8 +238,14 @@ void objectLambdaArnBucketProvided_shouldThrowException() {
236238 .completionFuture ().join ())
237239 .hasMessageContaining ("support S3 Object Lambda resources" ).hasCauseInstanceOf (IllegalArgumentException .class );
238240
239- assertThatThrownBy (() -> tm .download (b -> b .getObjectRequest (p -> p .bucket (objectLambdaArn ).key ("key" ))
240- .responseTransformer (AsyncResponseTransformer .toBytes ()))
241+
242+ DownloadRequest <ResponseBytes <GetObjectResponse >> downloadRequest =
243+ DownloadRequest .builder ()
244+ .getObjectRequest (g -> g .bucket (objectLambdaArn ).key ("key" ))
245+ .responseTransformer (AsyncResponseTransformer .toBytes ())
246+ .build ();
247+
248+ assertThatThrownBy (() -> tm .download (downloadRequest )
241249 .completionFuture ().join ())
242250 .hasMessageContaining ("support S3 Object Lambda resources" ).hasCauseInstanceOf (IllegalArgumentException .class );
243251
@@ -280,9 +288,12 @@ void mrapArnProvided_shouldThrowException() {
280288 .completionFuture ().join ())
281289 .hasMessageContaining ("multi-region access point ARN" ).hasCauseInstanceOf (IllegalArgumentException .class );
282290
283- assertThatThrownBy (() -> tm .download (b -> b .getObjectRequest (p -> p .bucket (mrapArn ).key ("key" ))
284- .responseTransformer (AsyncResponseTransformer .toBytes ()))
285- .completionFuture ().join ())
291+ DownloadRequest <ResponseBytes <GetObjectResponse >> downloadRequest =
292+ DownloadRequest .builder ()
293+ .getObjectRequest (g -> g .bucket (mrapArn ).key ("key" ))
294+ .responseTransformer (AsyncResponseTransformer .toBytes ()).build ();
295+
296+ assertThatThrownBy (() -> tm .download (downloadRequest ).completionFuture ().join ())
286297 .hasMessageContaining ("multi-region access point ARN" ).hasCauseInstanceOf (IllegalArgumentException .class );
287298
288299 assertThatThrownBy (() -> tm .uploadDirectory (b -> b .bucket (mrapArn )
0 commit comments