Skip to content

Commit 8e07851

Browse files
author
Anirav Kareddy
committed
making stream support mark/reset
1 parent 5d24f6e commit 8e07851

File tree

1 file changed

+10
-3
lines changed

1 file changed

+10
-3
lines changed

src/test/java/software/amazon/encryption/s3/S3EncryptionClientStreamTest.java

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@
3333
import javax.crypto.AEADBadTagException;
3434
import javax.crypto.KeyGenerator;
3535
import javax.crypto.SecretKey;
36+
import java.io.BufferedInputStream;
3637
import java.io.IOException;
3738
import java.io.InputStream;
3839
import java.nio.charset.StandardCharsets;
@@ -331,7 +332,9 @@ public void customSetBufferSizeWithLargeObject() throws IOException {
331332

332333
// Tight bound on the custom buffer size limit of 32MiB
333334
final long fileSizeExceedingDefaultLimit = 1024 * 1024 * 32 + 1;
334-
final InputStream largeObjectStream = new BoundedInputStream(fileSizeExceedingDefaultLimit);
335+
final InputStream largeObjectStream = new BufferedInputStream(
336+
new BoundedInputStream(fileSizeExceedingDefaultLimit)
337+
);
335338
v3ClientWithBuffer32MiB.putObject(PutObjectRequest.builder()
336339
.bucket(BUCKET)
337340
.key(objectKey)
@@ -385,7 +388,9 @@ public void customSetBufferSizeWithLargeObjectAsyncClient() throws IOException {
385388

386389
// Tight bound on the custom buffer size limit of 32MiB
387390
final long fileSizeExceedingDefaultLimit = 1024 * 1024 * 32 + 1;
388-
final InputStream largeObjectStream = new BoundedInputStream(fileSizeExceedingDefaultLimit);
391+
final InputStream largeObjectStream = new BufferedInputStream(
392+
new BoundedInputStream(fileSizeExceedingDefaultLimit)
393+
);
389394
ExecutorService singleThreadExecutor = Executors.newSingleThreadExecutor();
390395
CompletableFuture<PutObjectResponse> futurePut = v3ClientWithBuffer32MiB.putObject(PutObjectRequest.builder()
391396
.bucket(BUCKET)
@@ -438,7 +443,9 @@ public void delayedAuthModeWithLargeObject() throws IOException {
438443

439444
// Tight bound on the default limit of 64MiB
440445
final long fileSizeExceedingDefaultLimit = 1024 * 1024 * 64 + 1;
441-
final InputStream largeObjectStream = new BoundedInputStream(fileSizeExceedingDefaultLimit);
446+
final InputStream largeObjectStream = new BufferedInputStream(
447+
new BoundedInputStream(fileSizeExceedingDefaultLimit)
448+
);
442449
v3Client.putObject(PutObjectRequest.builder()
443450
.bucket(BUCKET)
444451
.key(objectKey)

0 commit comments

Comments
 (0)