Fix test failures: unwrap CachingJWKSetSource chain and handle Duration timeout fields

Author: rujche

sdk/spring/spring-cloud-azure-autoconfigure/src/test/java/com/azure/spring/cloud/autoconfigure/implementation/aad/configuration/AadResourceServerConfigurationTests.java

@@ -427,21 +427,28 @@ private static void verifyJwtDecoderRestTemplateTimeouts(JwtDecoder jwtDecoder,
             ((com.nimbusds.jose.proc.JWSVerificationKeySelector<?>) keySelector).getJWKSource();
         assertThat(jwkSource).isInstanceOf(com.nimbusds.jose.jwk.source.JWKSetBasedJWKSource.class);
 
-        // JWKSetBasedJWKSource -> JWKSetSource (SpringJWKSource)
+        // JWKSetBasedJWKSource -> JWKSetSource (CachingJWKSetSource -> JWKSetSourceWrapper -> actual source)
         Object jwkSetSource =
             ((com.nimbusds.jose.jwk.source.JWKSetBasedJWKSource<?>) jwkSource).getJWKSetSource();
 
-        // SpringJWKSource -> restOperations (RestTemplate)
+        // Unwrap JWKSetSourceWrapper chain to find the source with restOperations
+        while (jwkSetSource instanceof com.nimbusds.jose.jwk.source.JWKSetSourceWrapper<?> wrapper) {
+            jwkSetSource = wrapper.getSource();
+        }
+
+        // actual source -> restOperations (RestTemplate)
         Object restOperations = ReflectionTestUtils.getField(jwkSetSource, "restOperations");
         assertThat(restOperations).isInstanceOf(org.springframework.web.client.RestTemplate.class);
 
         // RestTemplate -> ClientHttpRequestFactory
         org.springframework.http.client.ClientHttpRequestFactory requestFactory =
             ((org.springframework.web.client.RestTemplate) restOperations).getRequestFactory();
 
-        // Verify timeouts on the request factory
-        int connectTimeout = (int) ReflectionTestUtils.getField(requestFactory, "connectTimeout");
-        int readTimeout = (int) ReflectionTestUtils.getField(requestFactory, "readTimeout");
+        // Verify timeouts on the request factory (may be stored as Duration or int)
+        Object connectTimeoutValue = ReflectionTestUtils.getField(requestFactory, "connectTimeout");
+        Object readTimeoutValue = ReflectionTestUtils.getField(requestFactory, "readTimeout");
+        int connectTimeout = connectTimeoutValue instanceof java.time.Duration d ? (int) d.toMillis() : (int) connectTimeoutValue;
+        int readTimeout = readTimeoutValue instanceof java.time.Duration d ? (int) d.toMillis() : (int) readTimeoutValue;
         assertThat(connectTimeout).isEqualTo(expectedConnectTimeoutMs);
         assertThat(readTimeout).isEqualTo(expectedReadTimeoutMs);
     }

sdk/spring/spring-cloud-azure-autoconfigure/src/test/java/com/azure/spring/cloud/autoconfigure/implementation/aadb2c/configuration/AadB2cResourceServerAutoConfigurationTests.java

@@ -354,11 +354,13 @@ private static void verifyResourceRetrieverRestTemplateTimeouts(ApplicationConte
         org.springframework.http.client.ClientHttpRequestFactory requestFactory =
             ((org.springframework.web.client.RestTemplate) restOperations).getRequestFactory();
 
-        // Verify timeouts on the request factory
-        int connectTimeout = (int) org.springframework.test.util.ReflectionTestUtils
+        // Verify timeouts on the request factory (may be stored as Duration or int)
+        Object connectTimeoutValue = org.springframework.test.util.ReflectionTestUtils
             .getField(requestFactory, "connectTimeout");
-        int readTimeout = (int) org.springframework.test.util.ReflectionTestUtils
+        Object readTimeoutValue = org.springframework.test.util.ReflectionTestUtils
             .getField(requestFactory, "readTimeout");
+        int connectTimeout = connectTimeoutValue instanceof java.time.Duration d ? (int) d.toMillis() : (int) connectTimeoutValue;
+        int readTimeout = readTimeoutValue instanceof java.time.Duration d ? (int) d.toMillis() : (int) readTimeoutValue;
         assertThat(connectTimeout).isEqualTo(expectedConnectTimeoutMs);
         assertThat(readTimeout).isEqualTo(expectedReadTimeoutMs);
     }