diff --git a/bundles/aliyun-bundle/src/main/java/org/apache/gravitino/oss/fs/OSSFileSystemProvider.java b/bundles/aliyun-bundle/src/main/java/org/apache/gravitino/oss/fs/OSSFileSystemProvider.java index b7a6ccec94..b47d25335c 100644 --- a/bundles/aliyun-bundle/src/main/java/org/apache/gravitino/oss/fs/OSSFileSystemProvider.java +++ b/bundles/aliyun-bundle/src/main/java/org/apache/gravitino/oss/fs/OSSFileSystemProvider.java @@ -18,6 +18,7 @@ */ package org.apache.gravitino.oss.fs; +import com.google.common.annotations.VisibleForTesting; import com.google.common.collect.ImmutableMap; import java.io.IOException; import java.util.Map; @@ -40,6 +41,8 @@ public class OSSFileSystemProvider implements FileSystemProvider { // For example, User can use oss-endpoint to set the endpoint of OSS 'fs.oss.endpoint' in // Gravitino. // GCS and S3 also have similar mapping relationship. + + @VisibleForTesting public static final Map GRAVITINO_KEY_TO_OSS_HADOOP_KEY = ImmutableMap.of( OSSProperties.GRAVITINO_OSS_ENDPOINT, Constants.ENDPOINT_KEY, diff --git a/bundles/aws-bundle/src/main/java/org/apache/gravitino/s3/fs/S3FileSystemProvider.java b/bundles/aws-bundle/src/main/java/org/apache/gravitino/s3/fs/S3FileSystemProvider.java index 8d902e1258..b61e9d14f6 100644 --- a/bundles/aws-bundle/src/main/java/org/apache/gravitino/s3/fs/S3FileSystemProvider.java +++ b/bundles/aws-bundle/src/main/java/org/apache/gravitino/s3/fs/S3FileSystemProvider.java @@ -19,6 +19,7 @@ package org.apache.gravitino.s3.fs; +import com.google.common.annotations.VisibleForTesting; import com.google.common.collect.ImmutableMap; import java.io.IOException; import java.util.Map; @@ -33,6 +34,7 @@ public class S3FileSystemProvider implements FileSystemProvider { + @VisibleForTesting public static final Map GRAVITINO_KEY_TO_S3_HADOOP_KEY = ImmutableMap.of( S3Properties.GRAVITINO_S3_ENDPOINT, Constants.ENDPOINT, diff --git a/bundles/gcp-bundle/src/main/java/org/apache/gravitino/gcs/fs/GCSFileSystemProvider.java b/bundles/gcp-bundle/src/main/java/org/apache/gravitino/gcs/fs/GCSFileSystemProvider.java index be602c958d..a07ff3d6ec 100644 --- a/bundles/gcp-bundle/src/main/java/org/apache/gravitino/gcs/fs/GCSFileSystemProvider.java +++ b/bundles/gcp-bundle/src/main/java/org/apache/gravitino/gcs/fs/GCSFileSystemProvider.java @@ -19,6 +19,7 @@ package org.apache.gravitino.gcs.fs; import com.google.cloud.hadoop.fs.gcs.GoogleHadoopFileSystem; +import com.google.common.annotations.VisibleForTesting; import com.google.common.collect.ImmutableMap; import java.io.IOException; import java.util.Map; @@ -36,6 +37,7 @@ public class GCSFileSystemProvider implements FileSystemProvider { private static final String GCS_SERVICE_ACCOUNT_JSON_FILE = "fs.gs.auth.service.account.json.keyfile"; + @VisibleForTesting public static final Map GRAVITINO_KEY_TO_GCS_HADOOP_KEY = ImmutableMap.of(GCSProperties.GCS_SERVICE_ACCOUNT_JSON_PATH, GCS_SERVICE_ACCOUNT_JSON_FILE);