Skip to content

Commit 45348e5

Browse files
committed
Misc S3BlobStoreRepositoryTests improvements
1 parent e9de7fc commit 45348e5

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

modules/repository-s3/src/internalClusterTest/java/org/elasticsearch/repositories/s3/S3BlobStoreRepositoryTests.java

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010

1111
import fixture.s3.S3HttpHandler;
1212
import software.amazon.awssdk.awscore.AwsRequestOverrideConfiguration;
13+
import software.amazon.awssdk.core.internal.http.pipeline.stages.ApplyTransactionIdStage;
1314
import software.amazon.awssdk.services.s3.model.CreateMultipartUploadRequest;
1415
import software.amazon.awssdk.services.s3.model.ListMultipartUploadsRequest;
1516
import software.amazon.awssdk.services.s3.model.MultipartUpload;
@@ -121,9 +122,9 @@ public void setUp() throws Exception {
121122
region = "test-region";
122123
}
123124
if (region != null && randomBoolean()) {
124-
signerOverride = randomFrom("AWS3SignerType", "AWS4SignerType");
125+
signerOverride = randomFrom("SDKV2_AWS_V4_SIGNER_TYPE", "SDKV2_AWS_S3_V4_SIGNER_TYPE");
125126
} else if (randomBoolean()) {
126-
signerOverride = "AWS3SignerType";
127+
signerOverride = "SDKV2_AWS_S3_V4_SIGNER_TYPE";
127128
}
128129
shouldFailCompleteMultipartUploadRequest.set(false);
129130
super.setUp();
@@ -683,8 +684,7 @@ protected static class S3ErroneousHttpHandler extends ErroneousHttpHandler {
683684
@Override
684685
protected String requestUniqueId(final HttpExchange exchange) {
685686
// Amazon SDK client provides a unique ID per request
686-
// TODO NOMERGE: make "x-amz-request-id" into a constant someplace.
687-
return exchange.getRequestHeaders().getFirst("x-amz-request-id");
687+
return exchange.getRequestHeaders().getFirst(ApplyTransactionIdStage.HEADER_SDK_TRANSACTION_ID);
688688
}
689689
}
690690

0 commit comments

Comments
 (0)