|
17 | 17 | import software.amazon.awssdk.core.ResponseInputStream; |
18 | 18 | import software.amazon.awssdk.core.async.AsyncRequestBody; |
19 | 19 | import software.amazon.awssdk.core.async.AsyncResponseTransformer; |
| 20 | +import software.amazon.awssdk.core.async.BlockingInputStreamAsyncRequestBody; |
20 | 21 | import software.amazon.awssdk.core.sync.RequestBody; |
21 | 22 | import software.amazon.awssdk.services.s3.S3AsyncClient; |
22 | 23 | import software.amazon.awssdk.services.s3.S3Client; |
23 | 24 | import software.amazon.awssdk.services.s3.model.GetObjectResponse; |
24 | 25 | import software.amazon.awssdk.services.s3.model.PutObjectRequest; |
25 | 26 | import software.amazon.awssdk.services.s3.model.PutObjectResponse; |
26 | 27 | import software.amazon.awssdk.utils.IoUtils; |
27 | | -import software.amazon.encryption.s3.utils.BoundedStreamBufferer; |
28 | 28 | import software.amazon.encryption.s3.utils.BoundedInputStream; |
| 29 | +import software.amazon.encryption.s3.utils.BoundedStreamBufferer; |
29 | 30 | import software.amazon.encryption.s3.utils.MarkResetBoundedZerosInputStream; |
30 | 31 | import software.amazon.encryption.s3.utils.S3EncryptionClientTestResources; |
31 | 32 |
|
@@ -135,6 +136,63 @@ public void ordinaryInputStreamV3Encrypt() throws IOException { |
135 | 136 | v3Client.close(); |
136 | 137 | } |
137 | 138 |
|
| 139 | + @Test |
| 140 | + public void ordinaryInputStreamV3UnboundedAsync() { |
| 141 | + try (S3AsyncClient s3AsyncEncryptionClient = S3AsyncEncryptionClient.builder().aesKey(AES_KEY).build()) { |
| 142 | + final String objectKey = appendTestSuffix("ordinaryInputStreamV3UnboundedAsync"); |
| 143 | + BlockingInputStreamAsyncRequestBody body = |
| 144 | + AsyncRequestBody.forBlockingInputStream(null); |
| 145 | + try { |
| 146 | + s3AsyncEncryptionClient.putObject(r -> r.bucket(BUCKET).key(objectKey), body); |
| 147 | + fail("Expected exception!"); |
| 148 | + } catch (S3EncryptionClientException exception) { |
| 149 | + // expected |
| 150 | + assertTrue(exception.getMessage().contains("Unbounded streams are currently not supported")); |
| 151 | + } |
| 152 | + } |
| 153 | + } |
| 154 | + |
| 155 | + @Test |
| 156 | + public void ordinaryInputStreamV3UnboundedMultipartAsync() { |
| 157 | + try (S3AsyncClient s3AsyncEncryptionClient = S3AsyncEncryptionClient.builder() |
| 158 | + .aesKey(AES_KEY) |
| 159 | + .enableMultipartPutObject(true) |
| 160 | + .build()) { |
| 161 | + final String objectKey = appendTestSuffix("ordinaryInputStreamV3UnboundedAsync"); |
| 162 | + BlockingInputStreamAsyncRequestBody body = |
| 163 | + AsyncRequestBody.forBlockingInputStream(null); |
| 164 | + try { |
| 165 | + s3AsyncEncryptionClient.putObject(r -> r.bucket(BUCKET).key(objectKey), body); |
| 166 | + fail("Expected exception!"); |
| 167 | + } catch (S3EncryptionClientException exception) { |
| 168 | + // expected |
| 169 | + assertTrue(exception.getMessage().contains("Unbounded streams are currently not supported")); |
| 170 | + } |
| 171 | + } |
| 172 | + } |
| 173 | + |
| 174 | + @Test |
| 175 | + public void ordinaryInputStreamV3UnboundedCrt() { |
| 176 | + try (S3AsyncClient s3CrtAsyncClient = S3AsyncClient.crtCreate()) { |
| 177 | + try (S3AsyncClient s3AsyncEncryptionClient = S3AsyncEncryptionClient.builder() |
| 178 | + .aesKey(AES_KEY) |
| 179 | + .enableMultipartPutObject(true) |
| 180 | + .wrappedClient(s3CrtAsyncClient) |
| 181 | + .build()) { |
| 182 | + final String objectKey = appendTestSuffix("ordinaryInputStreamV3UnboundedCrt"); |
| 183 | + BlockingInputStreamAsyncRequestBody body = |
| 184 | + AsyncRequestBody.forBlockingInputStream(null); |
| 185 | + try { |
| 186 | + s3AsyncEncryptionClient.putObject(r -> r.bucket(BUCKET).key(objectKey), body); |
| 187 | + fail("Expected exception!"); |
| 188 | + } catch (S3EncryptionClientException exception) { |
| 189 | + // expected |
| 190 | + assertTrue(exception.getMessage().contains("Unbounded streams are currently not supported")); |
| 191 | + } |
| 192 | + } |
| 193 | + } |
| 194 | + } |
| 195 | + |
138 | 196 | @Test |
139 | 197 | public void ordinaryInputStreamV3Decrypt() throws IOException { |
140 | 198 | final String objectKey = appendTestSuffix("ordinaryInputStreamV3Decrypt"); |
@@ -274,9 +332,9 @@ public void customSetBufferSizeWithLargeObject() throws IOException { |
274 | 332 | final long fileSizeExceedingDefaultLimit = 1024 * 1024 * 32 + 1; |
275 | 333 | final InputStream largeObjectStream = new BoundedInputStream(fileSizeExceedingDefaultLimit); |
276 | 334 | v3ClientWithBuffer32MiB.putObject(PutObjectRequest.builder() |
277 | | - .bucket(BUCKET) |
278 | | - .key(objectKey) |
279 | | - .build(), RequestBody.fromInputStream(largeObjectStream, fileSizeExceedingDefaultLimit)); |
| 335 | + .bucket(BUCKET) |
| 336 | + .key(objectKey) |
| 337 | + .build(), RequestBody.fromInputStream(largeObjectStream, fileSizeExceedingDefaultLimit)); |
280 | 338 |
|
281 | 339 | largeObjectStream.close(); |
282 | 340 |
|
@@ -327,9 +385,9 @@ public void customSetBufferSizeWithLargeObjectAsyncClient() throws IOException { |
327 | 385 | final InputStream largeObjectStream = new BoundedInputStream(fileSizeExceedingDefaultLimit); |
328 | 386 | ExecutorService singleThreadExecutor = Executors.newSingleThreadExecutor(); |
329 | 387 | CompletableFuture<PutObjectResponse> futurePut = v3ClientWithBuffer32MiB.putObject(PutObjectRequest.builder() |
330 | | - .bucket(BUCKET) |
331 | | - .key(objectKey) |
332 | | - .build(), AsyncRequestBody.fromInputStream(largeObjectStream, fileSizeExceedingDefaultLimit, singleThreadExecutor)); |
| 388 | + .bucket(BUCKET) |
| 389 | + .key(objectKey) |
| 390 | + .build(), AsyncRequestBody.fromInputStream(largeObjectStream, fileSizeExceedingDefaultLimit, singleThreadExecutor)); |
333 | 391 |
|
334 | 392 | futurePut.join(); |
335 | 393 | largeObjectStream.close(); |
@@ -387,7 +445,7 @@ public void delayedAuthModeWithLargeObject() throws IOException { |
387 | 445 | assertThrows(S3EncryptionClientException.class, () -> v3Client.getObjectAsBytes(builder -> builder |
388 | 446 | .bucket(BUCKET) |
389 | 447 | .key(objectKey))); |
390 | | - |
| 448 | + |
391 | 449 | S3Client v3ClientWithDelayedAuth = S3EncryptionClient.builder() |
392 | 450 | .aesKey(AES_KEY) |
393 | 451 | .enableDelayedAuthenticationMode(true) |
|
0 commit comments