Add unit tests for TokenCredential resolution logic

Copilot · rujche · Copilot · commit da32873cf3b8 · 2025-12-04T02:38:16.000Z
Co-authored-by: rujche &lt;171773178+rujche@users.noreply.github.com&gt;
diff --git a/sdk/spring/spring-cloud-azure-stream-binder-eventhubs/src/main/java/com/azure/spring/cloud/stream/binder/eventhubs/implementation/config/EventHubsBinderConfiguration.java b/sdk/spring/spring-cloud-azure-stream-binder-eventhubs/src/main/java/com/azure/spring/cloud/stream/binder/eventhubs/implementation/config/EventHubsBinderConfiguration.java
@@ -139,7 +139,7 @@ EventHubsProcessorFactoryCustomizer defaultEventHubsProcessorFactoryCustomizer(
         return new DefaultProcessorFactoryCustomizer(credential, azureTokenCredentialResolver, processorClientBuilderCustomizers);
     }
 
-    private static TokenCredential resolveTokenCredential(
+    static TokenCredential resolveTokenCredential(
         AzureTokenCredentialResolver azureTokenCredentialResolver,
         ObjectProvider<AzureEventHubsProperties> eventHubsProperties,
         TokenCredential defaultCredential) {
diff --git a/sdk/spring/spring-cloud-azure-stream-binder-eventhubs/src/test/java/com/azure/spring/cloud/stream/binder/eventhubs/implementation/config/EventHubsBinderConfigurationTests.java b/sdk/spring/spring-cloud-azure-stream-binder-eventhubs/src/test/java/com/azure/spring/cloud/stream/binder/eventhubs/implementation/config/EventHubsBinderConfigurationTests.java
@@ -228,6 +228,70 @@ void processorBuilderCustomizerShouldBeConfiguredToProcessorFactoryCustomizer()
             });
     }
 
+    @Test
+    @SuppressWarnings("unchecked")
+    void resolveTokenCredentialReturnsCustomCredentialWhenBeanNameProvided() {
+        com.azure.core.credential.TokenCredential customCredential = mock(com.azure.core.credential.TokenCredential.class);
+        com.azure.core.credential.TokenCredential defaultCredential = mock(com.azure.core.credential.TokenCredential.class);
+        com.azure.spring.cloud.core.implementation.credential.resolver.AzureTokenCredentialResolver resolver = 
+            mock(com.azure.spring.cloud.core.implementation.credential.resolver.AzureTokenCredentialResolver.class);
+        
+        AzureEventHubsProperties properties = new AzureEventHubsProperties();
+        properties.setNamespace("fake-namespace");
+        properties.getCredential().setTokenCredentialBeanName("customCredential");
+        
+        org.springframework.beans.factory.ObjectProvider<AzureEventHubsProperties> propertiesProvider = 
+            mock(org.springframework.beans.factory.ObjectProvider.class);
+        org.mockito.Mockito.when(propertiesProvider.getIfAvailable()).thenReturn(properties);
+        org.mockito.Mockito.when(resolver.resolve(properties)).thenReturn(customCredential);
+        
+        com.azure.core.credential.TokenCredential result = 
+            com.azure.spring.cloud.stream.binder.eventhubs.implementation.config.EventHubsBinderConfiguration
+                .resolveTokenCredential(resolver, propertiesProvider, defaultCredential);
+        
+        assertThat(result).isEqualTo(customCredential);
+    }
+
+    @Test
+    @SuppressWarnings("unchecked")
+    void resolveTokenCredentialReturnsDefaultWhenNoCustomCredentialResolved() {
+        com.azure.core.credential.TokenCredential defaultCredential = mock(com.azure.core.credential.TokenCredential.class);
+        com.azure.spring.cloud.core.implementation.credential.resolver.AzureTokenCredentialResolver resolver = 
+            mock(com.azure.spring.cloud.core.implementation.credential.resolver.AzureTokenCredentialResolver.class);
+        
+        AzureEventHubsProperties properties = new AzureEventHubsProperties();
+        properties.setNamespace("fake-namespace");
+        
+        org.springframework.beans.factory.ObjectProvider<AzureEventHubsProperties> propertiesProvider = 
+            mock(org.springframework.beans.factory.ObjectProvider.class);
+        org.mockito.Mockito.when(propertiesProvider.getIfAvailable()).thenReturn(properties);
+        org.mockito.Mockito.when(resolver.resolve(properties)).thenReturn(null);
+        
+        com.azure.core.credential.TokenCredential result = 
+            com.azure.spring.cloud.stream.binder.eventhubs.implementation.config.EventHubsBinderConfiguration
+                .resolveTokenCredential(resolver, propertiesProvider, defaultCredential);
+        
+        assertThat(result).isEqualTo(defaultCredential);
+    }
+
+    @Test
+    @SuppressWarnings("unchecked")
+    void resolveTokenCredentialReturnsDefaultWhenPropertiesNotAvailable() {
+        com.azure.core.credential.TokenCredential defaultCredential = mock(com.azure.core.credential.TokenCredential.class);
+        com.azure.spring.cloud.core.implementation.credential.resolver.AzureTokenCredentialResolver resolver = 
+            mock(com.azure.spring.cloud.core.implementation.credential.resolver.AzureTokenCredentialResolver.class);
+        
+        org.springframework.beans.factory.ObjectProvider<AzureEventHubsProperties> propertiesProvider = 
+            mock(org.springframework.beans.factory.ObjectProvider.class);
+        org.mockito.Mockito.when(propertiesProvider.getIfAvailable()).thenReturn(null);
+        
+        com.azure.core.credential.TokenCredential result = 
+            com.azure.spring.cloud.stream.binder.eventhubs.implementation.config.EventHubsBinderConfiguration
+                .resolveTokenCredential(resolver, propertiesProvider, defaultCredential);
+        
+        assertThat(result).isEqualTo(defaultCredential);
+    }
+
     @Configuration
     @EnableConfigurationProperties(EventHubsExtendedBindingProperties.class)
     static class TestProcessorContainerConfiguration {
diff --git a/sdk/spring/spring-cloud-azure-stream-binder-servicebus/src/main/java/com/azure/spring/cloud/stream/binder/servicebus/implementation/config/ServiceBusBinderConfiguration.java b/sdk/spring/spring-cloud-azure-stream-binder-servicebus/src/main/java/com/azure/spring/cloud/stream/binder/servicebus/implementation/config/ServiceBusBinderConfiguration.java
@@ -148,7 +148,7 @@ ServiceBusProcessorFactoryCustomizer defaultServiceBusProcessorFactoryCustomizer
             sessionProcessorClientBuilderCustomizers);
     }
 
-    private static TokenCredential resolveTokenCredential(
+    static TokenCredential resolveTokenCredential(
         AzureTokenCredentialResolver azureTokenCredentialResolver,
         ObjectProvider<AzureServiceBusProperties> serviceBusProperties,
         TokenCredential defaultCredential) {
diff --git a/sdk/spring/spring-cloud-azure-stream-binder-servicebus/src/test/java/com/azure/spring/cloud/stream/binder/servicebus/implementation/config/ServiceBusBinderConfigurationTests.java b/sdk/spring/spring-cloud-azure-stream-binder-servicebus/src/test/java/com/azure/spring/cloud/stream/binder/servicebus/implementation/config/ServiceBusBinderConfigurationTests.java
@@ -200,6 +200,69 @@ void processorBuilderCustomizerShouldBeConfiguredToProcessorFactoryCustomizer()
             });
     }
 
+    @Test
+    @SuppressWarnings("unchecked")
+    void resolveTokenCredentialReturnsCustomCredentialWhenBeanNameProvided() {
+        com.azure.core.credential.TokenCredential customCredential = mock(com.azure.core.credential.TokenCredential.class);
+        com.azure.core.credential.TokenCredential defaultCredential = mock(com.azure.core.credential.TokenCredential.class);
+        com.azure.spring.cloud.core.implementation.credential.resolver.AzureTokenCredentialResolver resolver = 
+            mock(com.azure.spring.cloud.core.implementation.credential.resolver.AzureTokenCredentialResolver.class);
+        
+        com.azure.spring.cloud.autoconfigure.implementation.servicebus.properties.AzureServiceBusProperties properties = 
+            new com.azure.spring.cloud.autoconfigure.implementation.servicebus.properties.AzureServiceBusProperties();
+        properties.setNamespace("fake-namespace");
+        properties.getCredential().setTokenCredentialBeanName("customCredential");
+        
+        org.springframework.beans.factory.ObjectProvider<com.azure.spring.cloud.autoconfigure.implementation.servicebus.properties.AzureServiceBusProperties> propertiesProvider = 
+            mock(org.springframework.beans.factory.ObjectProvider.class);
+        org.mockito.Mockito.when(propertiesProvider.getIfAvailable()).thenReturn(properties);
+        org.mockito.Mockito.when(resolver.resolve(properties)).thenReturn(customCredential);
+        
+        com.azure.core.credential.TokenCredential result = 
+            ServiceBusBinderConfiguration.resolveTokenCredential(resolver, propertiesProvider, defaultCredential);
+        
+        assertThat(result).isEqualTo(customCredential);
+    }
+
+    @Test
+    @SuppressWarnings("unchecked")
+    void resolveTokenCredentialReturnsDefaultWhenNoCustomCredentialResolved() {
+        com.azure.core.credential.TokenCredential defaultCredential = mock(com.azure.core.credential.TokenCredential.class);
+        com.azure.spring.cloud.core.implementation.credential.resolver.AzureTokenCredentialResolver resolver = 
+            mock(com.azure.spring.cloud.core.implementation.credential.resolver.AzureTokenCredentialResolver.class);
+        
+        com.azure.spring.cloud.autoconfigure.implementation.servicebus.properties.AzureServiceBusProperties properties = 
+            new com.azure.spring.cloud.autoconfigure.implementation.servicebus.properties.AzureServiceBusProperties();
+        properties.setNamespace("fake-namespace");
+        
+        org.springframework.beans.factory.ObjectProvider<com.azure.spring.cloud.autoconfigure.implementation.servicebus.properties.AzureServiceBusProperties> propertiesProvider = 
+            mock(org.springframework.beans.factory.ObjectProvider.class);
+        org.mockito.Mockito.when(propertiesProvider.getIfAvailable()).thenReturn(properties);
+        org.mockito.Mockito.when(resolver.resolve(properties)).thenReturn(null);
+        
+        com.azure.core.credential.TokenCredential result = 
+            ServiceBusBinderConfiguration.resolveTokenCredential(resolver, propertiesProvider, defaultCredential);
+        
+        assertThat(result).isEqualTo(defaultCredential);
+    }
+
+    @Test
+    @SuppressWarnings("unchecked")
+    void resolveTokenCredentialReturnsDefaultWhenPropertiesNotAvailable() {
+        com.azure.core.credential.TokenCredential defaultCredential = mock(com.azure.core.credential.TokenCredential.class);
+        com.azure.spring.cloud.core.implementation.credential.resolver.AzureTokenCredentialResolver resolver = 
+            mock(com.azure.spring.cloud.core.implementation.credential.resolver.AzureTokenCredentialResolver.class);
+        
+        org.springframework.beans.factory.ObjectProvider<com.azure.spring.cloud.autoconfigure.implementation.servicebus.properties.AzureServiceBusProperties> propertiesProvider = 
+            mock(org.springframework.beans.factory.ObjectProvider.class);
+        org.mockito.Mockito.when(propertiesProvider.getIfAvailable()).thenReturn(null);
+        
+        com.azure.core.credential.TokenCredential result = 
+            ServiceBusBinderConfiguration.resolveTokenCredential(resolver, propertiesProvider, defaultCredential);
+        
+        assertThat(result).isEqualTo(defaultCredential);
+    }
+
     private ApplicationContextRunner getCustomizedApplicationContextRunner() {
         return this.contextRunner
             .withBean(ServiceBusProvisioner.class, () -> mock(ServiceBusProvisioner.class))

Original file line number	Diff line number	Diff line change
`@@ -139,7 +139,7 @@ EventHubsProcessorFactoryCustomizer defaultEventHubsProcessorFactoryCustomizer(`
`139`	`139`	`return new DefaultProcessorFactoryCustomizer(credential, azureTokenCredentialResolver, processorClientBuilderCustomizers);`
`140`	`140`	`}`
`141`	`141`
`142`		`- private static TokenCredential resolveTokenCredential(`
	`142`	`+ static TokenCredential resolveTokenCredential(`
`143`	`143`	`AzureTokenCredentialResolver azureTokenCredentialResolver,`
`144`	`144`	`ObjectProvider<AzureEventHubsProperties> eventHubsProperties,`
`145`	`145`	`TokenCredential defaultCredential) {`
Original file line number	Diff line number	Diff line change
`@@ -148,7 +148,7 @@ ServiceBusProcessorFactoryCustomizer defaultServiceBusProcessorFactoryCustomizer`
`148`	`148`	`sessionProcessorClientBuilderCustomizers);`
`149`	`149`	`}`
`150`	`150`
`151`		`- private static TokenCredential resolveTokenCredential(`
	`151`	`+ static TokenCredential resolveTokenCredential(`
`152`	`152`	`AzureTokenCredentialResolver azureTokenCredentialResolver,`
`153`	`153`	`ObjectProvider<AzureServiceBusProperties> serviceBusProperties,`
`154`	`154`	`TokenCredential defaultCredential) {`