fix

Author: vikrantpuppala

databricks-sdk-java/src/main/java/com/databricks/sdk/core/oauth/ExternalBrowserCredentialsProvider.java

@@ -53,7 +53,7 @@ public HeaderFactory configure(DatabricksConfig config) {
 
           LOGGER.debug("Using cached token, will immediately refresh");
           cachedCreds.token = cachedCreds.refresh();
-          tokenCache.save(cachedToken);
+          tokenCache.save(cachedCreds.getToken());
           return cachedCreds.configure(config);
         } catch (Exception e) {
           // If token refresh fails, log and continue to browser auth

databricks-sdk-java/src/test/java/com/databricks/sdk/core/oauth/ExternalBrowserCredentialsProviderTest.java

@@ -18,6 +18,7 @@
 import java.util.HashMap;
 import java.util.Map;
 import org.junit.jupiter.api.Test;
+import org.mockito.ArgumentCaptor;
 import org.mockito.Mockito;
 
 public class ExternalBrowserCredentialsProviderTest {
@@ -264,6 +265,21 @@ void cacheWithValidTokenTest() throws IOException {
 
     // Verify token was saved back to cache
     Mockito.verify(mockTokenCache, Mockito.times(1)).save(any(Token.class));
+
+    // Capture the token that was saved to cache to verify it's the refreshed token
+    ArgumentCaptor<Token> tokenCaptor = ArgumentCaptor.forClass(Token.class);
+    Mockito.verify(mockTokenCache).save(tokenCaptor.capture());
+    Token savedToken = tokenCaptor.getValue();
+
+    // Verify the saved token contains the refreshed values from the HTTP response
+    assertEquals(
+        "refreshed_access_token",
+        savedToken.getAccessToken(),
+        "Should save refreshed access token to cache");
+    assertEquals(
+        "new_refresh_token",
+        savedToken.getRefreshToken(),
+        "Should save new refresh token to cache");
   }
 
   @Test
@@ -326,6 +342,21 @@ void cacheWithInvalidAccessTokenValidRefreshTest() throws IOException {
 
     // Verify token was saved back to cache
     Mockito.verify(mockTokenCache, Mockito.times(1)).save(any(Token.class));
+
+    // Capture the token that was saved to cache to verify it's the refreshed token
+    ArgumentCaptor<Token> tokenCaptor = ArgumentCaptor.forClass(Token.class);
+    Mockito.verify(mockTokenCache).save(tokenCaptor.capture());
+    Token savedToken = tokenCaptor.getValue();
+
+    // Verify the saved token contains the refreshed values from the HTTP response
+    assertEquals(
+        "refreshed_access_token",
+        savedToken.getAccessToken(),
+        "Should save refreshed access token to cache");
+    assertEquals(
+        "new_refresh_token",
+        savedToken.getRefreshToken(),
+        "Should save new refresh token to cache");
   }
 
   @Test