diff --git a/integration-tests/src/main/java/org/apache/polaris/service/it/env/IcebergHelper.java b/integration-tests/src/main/java/org/apache/polaris/service/it/env/IcebergHelper.java index 020a12b272..f894489b9e 100644 --- a/integration-tests/src/main/java/org/apache/polaris/service/it/env/IcebergHelper.java +++ b/integration-tests/src/main/java/org/apache/polaris/service/it/env/IcebergHelper.java @@ -19,7 +19,6 @@ package org.apache.polaris.service.it.env; import static org.apache.polaris.service.it.env.PolarisApiEndpoints.REALM_HEADER; -import static org.apache.polaris.service.it.test.PolarisApplicationIntegrationTest.PRINCIPAL_ROLE_ALL; import com.google.common.collect.ImmutableMap; import java.util.Map; @@ -33,10 +32,12 @@ public final class IcebergHelper { private IcebergHelper() {} public static RESTCatalog restCatalog( + PolarisClient client, PolarisApiEndpoints endpoints, PrincipalWithCredentials credentials, String catalog, Map extraProperties) { + String authToken = client.obtainToken(credentials); SessionCatalog.SessionContext context = SessionCatalog.SessionContext.createEmpty(); RESTCatalog restCatalog = new RESTCatalog( @@ -50,12 +51,7 @@ public static RESTCatalog restCatalog( ImmutableMap.builder() .put( org.apache.iceberg.CatalogProperties.URI, endpoints.catalogApiEndpoint().toString()) - .put( - OAuth2Properties.CREDENTIAL, - credentials.getCredentials().getClientId() - + ":" - + credentials.getCredentials().getClientSecret()) - .put(OAuth2Properties.SCOPE, PRINCIPAL_ROLE_ALL) + .put(OAuth2Properties.TOKEN, authToken) .put( org.apache.iceberg.CatalogProperties.FILE_IO_IMPL, "org.apache.iceberg.inmemory.InMemoryFileIO") diff --git a/integration-tests/src/main/java/org/apache/polaris/service/it/test/PolarisApplicationIntegrationTest.java b/integration-tests/src/main/java/org/apache/polaris/service/it/test/PolarisApplicationIntegrationTest.java index 26fbdc32b1..4275ba0ca0 100644 --- a/integration-tests/src/main/java/org/apache/polaris/service/it/test/PolarisApplicationIntegrationTest.java +++ b/integration-tests/src/main/java/org/apache/polaris/service/it/test/PolarisApplicationIntegrationTest.java @@ -106,6 +106,7 @@ public class PolarisApplicationIntegrationTest { private static PolarisClient client; private static ClientCredentials clientCredentials; private static ClientPrincipal admin; + private static String authToken; private String principalRoleName; private String internalCatalogName; @@ -118,6 +119,7 @@ public static void setup(PolarisApiEndpoints apiEndpoints, ClientPrincipal admin realm = endpoints.realm(); admin = adminCredentials; clientCredentials = adminCredentials.credentials(); + authToken = client.obtainToken(clientCredentials); testDir = Path.of("build/test_data/iceberg/" + realm); FileUtils.deleteQuietly(testDir.toFile()); @@ -158,7 +160,7 @@ public void before(TestInfo testInfo) { } @AfterEach - public void cleanUp() throws Exception { + public void cleanUp() { client.cleanUp(clientCredentials); } @@ -235,10 +237,8 @@ private static RESTSessionCatalog newSessionCatalog(String catalog) { Map.of( "uri", endpoints.catalogApiEndpoint().toString(), - OAuth2Properties.CREDENTIAL, - clientCredentials.clientId() + ":" + clientCredentials.clientSecret(), - OAuth2Properties.SCOPE, - PRINCIPAL_ROLE_ALL, + OAuth2Properties.TOKEN, + authToken, "warehouse", catalog, "header." + REALM_HEADER, @@ -579,10 +579,8 @@ public void testWarehouseNotSpecified() throws IOException { Map.of( "uri", endpoints.catalogApiEndpoint().toString(), - OAuth2Properties.CREDENTIAL, - clientCredentials.clientId() + ":" + clientCredentials.clientSecret(), - OAuth2Properties.SCOPE, - PRINCIPAL_ROLE_ALL, + OAuth2Properties.TOKEN, + authToken, "warehouse", emptyEnvironmentVariable, "header." + REALM_HEADER, diff --git a/integration-tests/src/main/java/org/apache/polaris/service/it/test/PolarisRestCatalogIntegrationTest.java b/integration-tests/src/main/java/org/apache/polaris/service/it/test/PolarisRestCatalogIntegrationTest.java index 64f3426c63..87b63acca5 100644 --- a/integration-tests/src/main/java/org/apache/polaris/service/it/test/PolarisRestCatalogIntegrationTest.java +++ b/integration-tests/src/main/java/org/apache/polaris/service/it/test/PolarisRestCatalogIntegrationTest.java @@ -222,7 +222,11 @@ public void before(TestInfo testInfo) { restCatalog = IcebergHelper.restCatalog( - endpoints, principalCredentials, currentCatalogName, extraPropertiesBuilder.build()); + client, + endpoints, + principalCredentials, + currentCatalogName, + extraPropertiesBuilder.build()); } @AfterEach diff --git a/integration-tests/src/main/java/org/apache/polaris/service/it/test/PolarisRestCatalogViewIntegrationBase.java b/integration-tests/src/main/java/org/apache/polaris/service/it/test/PolarisRestCatalogViewIntegrationBase.java index 4f78413e05..34074bba31 100644 --- a/integration-tests/src/main/java/org/apache/polaris/service/it/test/PolarisRestCatalogViewIntegrationBase.java +++ b/integration-tests/src/main/java/org/apache/polaris/service/it/test/PolarisRestCatalogViewIntegrationBase.java @@ -108,7 +108,8 @@ public void before(TestInfo testInfo) { .build(); managementApi.createCatalog(principalRoleName, catalog); - restCatalog = IcebergHelper.restCatalog(endpoints, principalCredentials, catalogName, Map.of()); + restCatalog = + IcebergHelper.restCatalog(client, endpoints, principalCredentials, catalogName, Map.of()); } @AfterEach