Skip to content

Commit 4159327

Browse files
committed
fixed unit tests
1 parent 2cbfbd5 commit 4159327

File tree

2 files changed

+9
-7
lines changed

2 files changed

+9
-7
lines changed

polaris-core/src/test/java/org/apache/polaris/core/storage/azure/AzureCredentialsStorageIntegrationTest.java

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -48,16 +48,17 @@ public void testAzureCredentialFormatting() {
4848
"some_account." + AzureLocation.ADLS_ENDPOINT,
4949
expiresAt,
5050
Optional.of("endpoint/credentials"));
51-
Assertions.assertThat(adlsSuffixResult.credentials()).hasSize(5);
51+
Assertions.assertThat(adlsSuffixResult.credentials()).hasSize(3);
5252
Assertions.assertThat(adlsSuffixResult.credentials())
5353
.containsKey("adls.sas-token.some_account");
5454
Assertions.assertThat(adlsSuffixResult.credentials())
5555
.containsKey("adls.sas-token.some_account." + AzureLocation.ADLS_ENDPOINT);
56-
Assertions.assertThat(adlsSuffixResult.credentials())
56+
57+
Assertions.assertThat(adlsSuffixResult.extraProperties())
5758
.containsEntry(
5859
StorageAccessProperty.AZURE_REFRESH_CREDENTIALS_ENDPOINT.getPropertyName(),
5960
"endpoint/credentials");
60-
Assertions.assertThat(adlsSuffixResult.credentials())
61+
Assertions.assertThat(adlsSuffixResult.extraProperties())
6162
.containsEntry(
6263
StorageAccessProperty.AZURE_REFRESH_CREDENTIALS_ENABLED.getPropertyName(), "true");
6364

polaris-core/src/test/java/org/apache/polaris/service/storage/aws/AwsCredentialsStorageIntegrationTest.java

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -103,14 +103,15 @@ public void testGetSubscopedCreds(String scheme) {
103103
.containsEntry(StorageAccessProperty.AWS_TOKEN.getPropertyName(), "sess")
104104
.containsEntry(StorageAccessProperty.AWS_KEY_ID.getPropertyName(), "accessKey")
105105
.containsEntry(StorageAccessProperty.AWS_SECRET_KEY.getPropertyName(), "secretKey")
106+
.containsEntry(
107+
StorageAccessProperty.AWS_SESSION_TOKEN_EXPIRES_AT_MS.getPropertyName(),
108+
String.valueOf(EXPIRE_TIME.toEpochMilli()));
109+
assertThat(accessConfig.extraProperties())
106110
.containsEntry(
107111
StorageAccessProperty.AWS_REFRESH_CREDENTIALS_ENABLED.getPropertyName(), "true")
108112
.containsEntry(
109113
StorageAccessProperty.AWS_REFRESH_CREDENTIALS_ENDPOINT.getPropertyName(),
110-
"/namespace/table/credentials")
111-
.containsEntry(
112-
StorageAccessProperty.AWS_SESSION_TOKEN_EXPIRES_AT_MS.getPropertyName(),
113-
String.valueOf(EXPIRE_TIME.toEpochMilli()));
114+
"/namespace/table/credentials");
114115
}
115116

116117
@ParameterizedTest

0 commit comments

Comments
 (0)