diff --git a/src/main/java/com/databricks/jdbc/dbclient/impl/common/ClientConfigurator.java b/src/main/java/com/databricks/jdbc/dbclient/impl/common/ClientConfigurator.java index a041bfa76e..b07be2f22f 100644 --- a/src/main/java/com/databricks/jdbc/dbclient/impl/common/ClientConfigurator.java +++ b/src/main/java/com/databricks/jdbc/dbclient/impl/common/ClientConfigurator.java @@ -26,6 +26,7 @@ import java.net.ServerSocket; import java.nio.file.Path; import java.nio.file.Paths; +import java.time.Duration; import java.util.ArrayList; import java.util.Arrays; import java.util.List; @@ -196,6 +197,8 @@ public void setupU2MConfig() throws DatabricksParsingException { .setAuthType(DatabricksJdbcConstants.U2M_AUTH_TYPE) .setHost(host) .setClientId(clientId) + .setOAuthBrowserAuthTimeout( + Duration.ofHours(1)) // TODO : add a browser timeout connection config .setClientSecret(connectionContext.getClientSecret()) .setOAuthRedirectUrl(redirectUrl); diff --git a/src/test/java/com/databricks/jdbc/dbclient/impl/common/ClientConfiguratorTest.java b/src/test/java/com/databricks/jdbc/dbclient/impl/common/ClientConfiguratorTest.java index 0ad94669b5..e93efce43d 100644 --- a/src/test/java/com/databricks/jdbc/dbclient/impl/common/ClientConfiguratorTest.java +++ b/src/test/java/com/databricks/jdbc/dbclient/impl/common/ClientConfiguratorTest.java @@ -25,6 +25,7 @@ import com.databricks.sdk.core.utils.Cloud; import java.io.IOException; import java.net.ServerSocket; +import java.time.Duration; import java.util.List; import java.util.Properties; import org.junit.jupiter.api.Test; @@ -453,6 +454,7 @@ void getWorkspaceClient_OAuthWithBrowserBasedAuthentication_SetsCustomRedirectUr assertEquals("browser-client-id", config.getClientId()); assertEquals("browser-client-secret", config.getClientSecret()); assertEquals(List.of("scope1", "scope2"), config.getScopes()); + assertEquals(Duration.ofHours(1), config.getOAuthBrowserAuthTimeout()); assertEquals("http://localhost:" + testPort, config.getOAuthRedirectUrl()); assertEquals(DatabricksJdbcConstants.U2M_AUTH_TYPE, config.getAuthType()); }