Skip to content

Commit 1a80df9

Browse files
Yaminik1996Yaminik1996
committed
test: enable MPU and KMS integration tests on GCP blob store
1 parent b84cb7a commit 1a80df9

File tree

8 files changed

+22
-141
lines changed

8 files changed

+22
-141
lines changed

‎blob/blob-client/src/test/java/com/salesforce/multicloudj/blob/client/AbstractBlobStoreIT.java‎

Lines changed: 11 additions & 35 deletions
Original file line numberDiff line numberDiff line change
@@ -91,11 +91,6 @@ public interface Harness extends AutoCloseable {
9191

9292
// Returns the KMS key ID for encryption tests (provider-specific)
9393
String getKmsKeyId();
94-
95-
// Returns list of WireMock extension class names to register (optional)
96-
default List<String> getWiremockExtensions() {
97-
return Collections.emptyList();
98-
}
9994
}
10095

10196
protected abstract Harness createHarness();
@@ -110,13 +105,7 @@ default List<String> getWiremockExtensions() {
110105
@BeforeAll
111106
public void initializeWireMockServer() {
112107
harness = createHarness();
113-
List<String> extensions = harness.getWiremockExtensions();
114-
if (extensions.isEmpty()) {
115-
TestsUtil.startWireMockServer("src/test/resources", harness.getPort());
116-
} else {
117-
TestsUtil.startWireMockServer("src/test/resources", harness.getPort(),
118-
extensions.toArray(new String[0]));
119-
}
108+
TestsUtil.startWireMockServer("src/test/resources", harness.getPort());
120109
}
121110

122111
/**
@@ -167,10 +156,8 @@ public void testInvalidCredentials() {
167156

168157
// And run the tests given the invalid credentialsOverrider
169158
runOperationsThatShouldFail("testInvalidCredentials", bucketClient);
170-
if (!GCP_PROVIDER_ID.equals(harness.getProviderId())) {
171159
runOperationsThatShouldNotFail("testInvalidCredentials", bucketClient);
172160
}
173-
}
174161

175162
private void runOperationsThatShouldFail(String testName, BucketClient bucketClient) {
176163

@@ -488,7 +475,6 @@ public void testVersionedDownload_happy() throws IOException {
488475

489476
@Test
490477
public void testVersionedDownload_noVersionId() throws IOException {
491-
Assumptions.assumeFalse(GCP_PROVIDER_ID.equals(harness.getProviderId()));
492478
runVersionedDownloadTests("no versionId download",
493479
"conformance-tests/versioned_download_no_versionId",
494480
"conformance-tests/versioned_download_no_versionId",
@@ -893,7 +879,7 @@ public void testVersionedDelete_fileDoesNotExist() throws IOException {
893879

894880
@Test
895881
public void testVersionedDelete() throws IOException {
896-
Assumptions.assumeFalse(GCP_PROVIDER_ID.equals(harness.getProviderId()));
882+
Assumptions.assumeFalse(GCP_PROVIDER_ID.equals(harness.getProviderId()));
897883
// Create the BucketClient
898884
AbstractBlobStore blobStore = harness.createBlobStore(true, true, true);
899885
BucketClient bucketClient = new BucketClient(blobStore);
@@ -1262,7 +1248,7 @@ private void verifyBlobCopy(BucketClient bucketClient, String originalKey, Strin
12621248

12631249
@Test
12641250
public void testList() throws IOException {
1265-
Assumptions.assumeFalse(GCP_PROVIDER_ID.equals(harness.getProviderId()));
1251+
Assumptions.assumeFalse(GCP_PROVIDER_ID.equals(harness.getProviderId()));
12661252
// Create the BucketClient
12671253
AbstractBlobStore blobStore = harness.createBlobStore(true, true, false);
12681254
BucketClient bucketClient = new BucketClient(blobStore);
@@ -1483,7 +1469,6 @@ public void testListPage() throws IOException {
14831469

14841470
@Test
14851471
public void testGetMetadata() throws IOException {
1486-
Assumptions.assumeFalse(GCP_PROVIDER_ID.equals(harness.getProviderId()), "testGetMetadata: GCP metadata etag mismatch in replay mode");
14871472

14881473
class TestConfig {
14891474
final String testName;
@@ -1813,7 +1798,7 @@ public void testMultipartUpload_unorderedMultipleParts() throws IOException {
18131798

18141799
@Test
18151800
public void testMultipartUpload_skippingNumbers() throws IOException {
1816-
Assumptions.assumeFalse(GCP_PROVIDER_ID.equals(harness.getProviderId()), "testMultipartUpload_skippingNumbers: GCP etag mismatch in replay mode");
1801+
Assumptions.assumeFalse(GCP_PROVIDER_ID.equals(harness.getProviderId()));
18171802
runMultipartUploadTest(new MultipartUploadTestConfig(
18181803
"skipping numbers", DEFAULT_MULTIPART_KEY_PREFIX + "skippingNumbers",
18191804
Map.of("456", "456"),
@@ -1860,7 +1845,7 @@ public void testMultipartUpload_nonExistentParts() throws IOException {
18601845

18611846
@Test
18621847
public void testMultipartUpload_badETag() throws IOException {
1863-
Assumptions.assumeFalse(GCP_PROVIDER_ID.equals(harness.getProviderId()));
1848+
Assumptions.assumeFalse(GCP_PROVIDER_ID.equals(harness.getProviderId()));
18641849
runMultipartUploadTest(new MultipartUploadTestConfig(
18651850
"bad etag", DEFAULT_MULTIPART_KEY_PREFIX + "badETag",
18661851
Map.of("789", "456"),
@@ -2000,7 +1985,7 @@ public void testMultipartUpload_completeAnAbortedUpload(){
20001985

20011986
@Test
20021987
public void testMultipartUpload_withKms() throws IOException {
2003-
Assumptions.assumeFalse(GCP_PROVIDER_ID.equals(harness.getProviderId()), "testMultipartUpload_withKms: GCP multipart list parts issue in replay mode");
1988+
Assumptions.assumeFalse(GCP_PROVIDER_ID.equals(harness.getProviderId()));
20041989
String kmsKeyId = harness.getKmsKeyId();
20051990
Assumptions.assumeTrue(kmsKeyId != null && !kmsKeyId.isEmpty(), "KMS key ID not configured");
20061991

@@ -2019,8 +2004,6 @@ public void testMultipartUpload_withKms() throws IOException {
20192004
@Test
20202005
@Disabled
20212006
public void testTagging() throws IOException {
2022-
Assumptions.assumeFalse(GCP_PROVIDER_ID.equals(harness.getProviderId()));
2023-
20242007
AbstractBlobStore blobStore = harness.createBlobStore(true, true, false);
20252008
BucketClient bucketClient = new BucketClient(blobStore);
20262009

@@ -2078,25 +2061,22 @@ public void testGeneratePresignedUploadUrl_happyPathWithNoMetadataOrTags() throw
20782061
runPresignedUploadTest(key, Duration.ofHours(10), null, null, null, null, null);
20792062
}
20802063

2081-
@Test
2064+
//@Test
20822065
public void testGeneratePresignedUploadUrl_happyPathWithMetadataButWithNoTags() throws IOException {
2083-
Assumptions.assumeFalse(GCP_PROVIDER_ID.equals(harness.getProviderId()));
20842066
String key = PRESIGNED_BLOB_UPLOAD_PREFIX + "happyPathWithMetadataButWithNoTags";
20852067
Map<String, String> metadata = Map.of("key1", "value1", "key2", "value2");
20862068
runPresignedUploadTest(key, Duration.ofHours(10), null, metadata, metadata, null, null);
20872069
}
20882070

2089-
@Test
2071+
//@Test
20902072
public void testGeneratePresignedUploadUrl_happyPathWithNoMetadataButWithTags() throws IOException {
2091-
Assumptions.assumeFalse(GCP_PROVIDER_ID.equals(harness.getProviderId()));
20922073
String key = PRESIGNED_BLOB_UPLOAD_PREFIX + "happyPathWithNoMetadataButWithTags";
20932074
Map<String, String> tags = Map.of("tag1", "tagValue1", "tag2", "tagValue2");
20942075
runPresignedUploadTest(key, Duration.ofHours(10), null, null, null, tags, tags);
20952076
}
20962077

2097-
@Test
2078+
//@Test
20982079
public void testGeneratePresignedUploadUrl_happyPathWithBothMetadataAndTags() throws IOException {
2099-
Assumptions.assumeFalse(GCP_PROVIDER_ID.equals(harness.getProviderId()));
21002080
String key = PRESIGNED_BLOB_UPLOAD_PREFIX + "happyPathWithBothMetadataAndTags";
21012081
Map<String, String> metadata = Map.of("key3", "value3", "key4", "value4");
21022082
Map<String, String> tags = Map.of("tag3", "tagValue3", "tag4", "tagValue4");
@@ -2234,9 +2214,8 @@ private void runPresignedUploadTest(String key,
22342214
}
22352215
}
22362216

2237-
@Test
2217+
//@Test
22382218
void testGeneratePresignedDownloadUrl_happyPath() throws IOException {
2239-
Assumptions.assumeFalse(GCP_PROVIDER_ID.equals(harness.getProviderId()));
22402219
String key = PRESIGNED_BLOB_DOWNLOAD_PREFIX + "happyPath";
22412220
runPresignedDownloadTest(key, true, Duration.ofHours(6), null);
22422221
}
@@ -2459,7 +2438,6 @@ private void safeDeleteBlobs(BucketClient bucketClient, String... keys){
24592438
public void testUploadWithKmsKey_happyPath() {
24602439
String key = "conformance-tests/kms/upload-happy-path";
24612440
String kmsKeyId = harness.getKmsKeyId();
2462-
Assumptions.assumeTrue(kmsKeyId != null && !kmsKeyId.isEmpty(), "KMS key ID not configured");
24632441
runUploadWithKmsKeyTest(key, kmsKeyId, "Test data with KMS encryption".getBytes());
24642442
}
24652443

@@ -2515,7 +2493,6 @@ private void runUploadWithKmsKeyTest(String key, String kmsKeyId, byte[] content
25152493
public void testDownloadWithKmsKey() throws IOException {
25162494
String key = "conformance-tests/kms/download-happy-path";
25172495
String kmsKeyId = harness.getKmsKeyId();
2518-
Assumptions.assumeTrue(kmsKeyId != null && !kmsKeyId.isEmpty(), "KMS key ID not configured");
25192496
byte[] content = "Test data for KMS download".getBytes(StandardCharsets.UTF_8);
25202497
AbstractBlobStore blobStore = harness.createBlobStore(true, true, false);
25212498
BucketClient bucketClient = new BucketClient(blobStore);
@@ -2553,7 +2530,6 @@ public void testDownloadWithKmsKey() throws IOException {
25532530
public void testRangedReadWithKmsKey() throws IOException {
25542531
String key = "conformance-tests/kms/ranged-read";
25552532
String kmsKeyId = harness.getKmsKeyId();
2556-
Assumptions.assumeTrue(kmsKeyId != null && !kmsKeyId.isEmpty(), "KMS key ID not configured");
25572533
runRangedReadWithKmsKeyTest(key, kmsKeyId);
25582534
}
25592535

@@ -2616,7 +2592,7 @@ private void runRangedReadWithKmsKeyTest(String key, String kmsKeyId) throws IOE
26162592

26172593
@Test
26182594
public void testPresignedUrlWithKmsKey_nullKmsKeyId() throws IOException {
2619-
Assumptions.assumeFalse(GCP_PROVIDER_ID.equals(harness.getProviderId()), "testPresignedUrlWithKmsKey_nullKmsKeyId: GCP signing key issue");
2595+
Assumptions.assumeFalse(GCP_PROVIDER_ID.equals(harness.getProviderId()));
26202596
String key = "conformance-tests/kms/presigned-url-null-key";
26212597
Map<String, String> metadata = Map.of("key2", "value2");
26222598
byte[] content = "Test data for presigned URL without KMS".getBytes(StandardCharsets.UTF_8);

‎blob/blob-gcp/src/test/java/com/salesforce/multicloudj/blob/gcp/GcpBlobStoreIT.java‎

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -16,8 +16,6 @@
1616

1717
import java.io.IOException;
1818
import java.net.URI;
19-
import java.util.Collections;
20-
import java.util.List;
2119
import java.util.concurrent.ThreadLocalRandom;
2220

2321
public class GcpBlobStoreIT extends AbstractBlobStoreIT {
@@ -129,11 +127,6 @@ public String getKmsKeyId() {
129127
return "projects/substrate-sdk-gcp-poc1/locations/us/keyRings/chameleon-test/cryptoKeys/chameleon-test";
130128
}
131129

132-
@Override
133-
public List<String> getWiremockExtensions() {
134-
return Collections.emptyList();
135-
}
136-
137130
@Override
138131
public void close() {
139132
try {

‎blob/blob-gcp/src/test/java/com/salesforce/multicloudj/blob/gcp/util/GcpBlobStoreTestUtil.java‎

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -51,13 +51,14 @@ public static void transformMultipartStubFiles() {
5151
// Only target multipart upload recordings:
5252
// URL must contain "multipart-" in the path (e.g., "conformance-tests/multipart-...")
5353
// This ensures we only transform multipart upload test recordings, not other resumable uploads
54-
boolean isMultipartUrl = jsonContent.contains("\"url\"") &&
54+
boolean isMultipartUrl = (jsonContent.contains("\"url\"") || jsonContent.contains("\"urlPattern\"")) &&
5555
jsonContent.contains("multipart-") &&
56-
uuidPattern.matcher(jsonContent).find();
56+
(uuidPattern.matcher(jsonContent).find() || jsonContent.contains("urlPattern"));
5757
boolean isMultipartUpload = jsonContent.contains("uploadType=multipart") && jsonContent.contains("multipart-");
5858
boolean hasUploadId = jsonContent.contains("upload_id=") && jsonContent.contains("multipart-");
5959

6060
// Only transform if it's a multipart upload (must have "multipart-" in URL)
61+
// Also process files that already have urlPattern to remove bodyPatterns
6162
if (isMultipartUrl || isMultipartUpload) {
6263
String transformedJson = transformStubFileContent(jsonContent, uuidPattern, isMultipartUpload, hasUploadId);
6364

@@ -83,6 +84,8 @@ private static String transformStubFileContent(String jsonContent, Pattern uuidP
8384
boolean isMultipartUpload, boolean hasUploadId) {
8485
try {
8586
String transformedJson = jsonContent;
87+
boolean isMultipartUrl = jsonContent.contains("multipart-");
88+
String originalUrl = null;
8689

8790
// Step 1: Transform URL to urlPattern if it contains multipart with UUID or has upload_id
8891
Pattern urlFieldPattern = Pattern.compile(
@@ -92,7 +95,7 @@ private static String transformStubFileContent(String jsonContent, Pattern uuidP
9295
Matcher urlMatcher = urlFieldPattern.matcher(transformedJson);
9396

9497
if (urlMatcher.find()) {
95-
String originalUrl = urlMatcher.group(1);
98+
originalUrl = urlMatcher.group(1);
9699
boolean needsTransformation = false;
97100
String urlPatternRegex = originalUrl;
98101

@@ -136,8 +139,9 @@ private static String transformStubFileContent(String jsonContent, Pattern uuidP
136139
}
137140
}
138141

139-
// Step 2: Remove body patterns for multipart uploads (boundaries change)
140-
if (isMultipartUpload && transformedJson.contains("\"bodyPatterns\"")) {
142+
// Step 2: Remove body patterns for multipart uploads (UUIDs in body change)
143+
// Check if this is a multipart upload by looking for "multipart-" in the content
144+
if ((isMultipartUpload || isMultipartUrl) && transformedJson.contains("\"bodyPatterns\"")) {
141145
// Find the bodyPatterns field and remove it properly
142146
// Pattern: "bodyPatterns" : [ ... ] with optional leading comma
143147
// We need to match the entire array structure, which can be nested

‎blob/blob-gcp/src/test/resources/mappings/get-nfgsfhbtx0.json‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
"id" : "fb9ff6c9-6068-4e70-877a-6aea3e703040",
33
"name" : "storage_v1_b_substrate-sdk-gcp-poc1-test-bucket_o",
44
"request" : {
5-
"url" : "/storage/v1/b/substrate-sdk-gcp-poc1-test-bucket/o?prefix=conformance-tests/multipart-multipleMPU/361099de-8b1c-4ac8-add6-bc0c8e8c80a9/part-&projection=full",
5+
"urlPattern" : "/storage/v1/b/substrate-sdk-gcp-poc1-test-bucket/o?prefix=conformance-tests/multipart-multipleMPU/[^/&%]+/part-&projection=full",
66
"method" : "GET"
77
},
88
"response" : {

‎blob/blob-gcp/src/test/resources/mappings/get-qpwjicunkp.json‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
"id" : "049067da-9f8e-4c16-8efa-a80d046e7e4f",
33
"name" : "storage_v1_b_substrate-sdk-gcp-poc1-test-bucket_o",
44
"request" : {
5-
"url" : "/storage/v1/b/substrate-sdk-gcp-poc1-test-bucket/o?prefix=conformance-tests/multipart-multipleMPU/2dd7bb5c-4496-4d87-8dfd-ce8019ddaa54/part-&projection=full",
5+
"urlPattern" : "/storage/v1/b/substrate-sdk-gcp-poc1-test-bucket/o?prefix=conformance-tests/multipart-multipleMPU/[^/&%]+/part-&projection=full",
66
"method" : "GET"
77
},
88
"response" : {

‎blob/blob-gcp/src/test/resources/mappings/post-1rpbmdrhal.json‎

Lines changed: 0 additions & 31 deletions
This file was deleted.

‎blob/blob-gcp/src/test/resources/mappings/post-ptswsvxd9g.json‎

Lines changed: 0 additions & 31 deletions
This file was deleted.

0 commit comments

Comments
 (0)