Add unit tests

Author: samikshya-db

databricks-sdk-java/src/main/java/com/databricks/sdk/core/DatabricksConfig.java

@@ -410,13 +410,17 @@ public DatabricksConfig setAzureUseMsi(boolean azureUseMsi) {
     return this;
   }
 
-  /** @deprecated Use {@link #getAzureUseMsi()} instead. */
+  /**
+   * @deprecated Use {@link #getAzureUseMsi()} instead.
+   */
   @Deprecated()
   public boolean getAzureUseMSI() {
     return azureUseMsi;
   }
 
-  /** @deprecated Use {@link #setAzureUseMsi(boolean)} instead. */
+  /**
+   * @deprecated Use {@link #setAzureUseMsi(boolean)} instead.
+   */
   @Deprecated
   public DatabricksConfig setAzureUseMSI(boolean azureUseMsi) {
     this.azureUseMsi = azureUseMsi;
@@ -745,12 +749,11 @@ public String getEffectiveOAuthRedirectUrl() {
   }
 
   /**
-   * Determines if Azure-specific OIDC endpoints should be used.
-   * This is true in two cases:
-   * 1. When auth type is not specified (this is only in case of external browser auth)
-   * 2. When Azure client ID is present (service principal auth)
+   * Determines if Azure-specific OIDC endpoints should be used. This is true in two cases: 1. When
+   * auth type is not specified (this is only in case of external browser auth) 2. When Azure client
+   * ID is present (service principal auth)
    */
-  private boolean shouldUseAzureOidcEndpoints() {
+  boolean shouldUseAzureOidcEndpoints() {
     return Objects.equals(getAuthType(), null) || getAzureClientId() != null;
   }
 }

databricks-sdk-java/src/test/java/com/databricks/sdk/core/DatabricksConfigTest.java

@@ -251,4 +251,35 @@ public void testGetTokenSourceWithOAuth() {
     assertFalse(tokenSource instanceof ErrorTokenSource);
     assertEquals(tokenSource.getToken().getAccessToken(), "test-token");
   }
+
+  @Test
+  public void testShouldUseAzureOidcEndpointsForExternalBrowserAuth() {
+    DatabricksConfig config =
+        new DatabricksConfig()
+            .setHost("https://adb-1234567890.0.azuredatabricks.net/")
+            .setAuthType(null);
+    assertTrue(config.shouldUseAzureOidcEndpoints());
+  }
+
+  @Test
+  public void testShouldUseAzureOidcEndpointsForServicePrincipal() {
+    DatabricksConfig config =
+        new DatabricksConfig()
+            .setHost("https://adb-1234567890.0.azuredatabricks.net/")
+            .setAzureClientId("test-client-id")
+            .setAzureClientSecret("test-client-secret")
+            .setAzureTenantId("test-tenant-id");
+    assertTrue(config.shouldUseAzureOidcEndpoints());
+  }
+
+  @Test
+  public void testShouldNotUseAzureOidcEndpointsForAzureM2M() {
+    DatabricksConfig config =
+        new DatabricksConfig()
+            .setHost("https://adb-1234567890.0.azuredatabricks.net/")
+            .setAuthType("oauth-m2m")
+            .setClientId("test-client-id")
+            .setClientSecret("test-client-secret");
+    assertFalse(config.shouldUseAzureOidcEndpoints());
+  }
 }