diff --git a/spring-pulsar/src/main/java/org/springframework/pulsar/config/GenericListenerEndpointRegistry.java b/spring-pulsar/src/main/java/org/springframework/pulsar/config/GenericListenerEndpointRegistry.java index 7a027b4a1..8704dc471 100644 --- a/spring-pulsar/src/main/java/org/springframework/pulsar/config/GenericListenerEndpointRegistry.java +++ b/spring-pulsar/src/main/java/org/springframework/pulsar/config/GenericListenerEndpointRegistry.java @@ -60,6 +60,7 @@ * @author Soby Chacko * @author Christophe Bornet * @author Chris Bono + * @author Hyun Lee */ public class GenericListenerEndpointRegistry> implements PulsarListenerContainerRegistry, DisposableBean, SmartLifecycle, ApplicationContextAware, @@ -135,6 +136,9 @@ public void registerListenerContainer(E endpoint, ListenerContainerFactory registerContainer(boolean autoStartup, + boolean startImmediately) { + var endpoint = mock(PulsarListenerEndpoint.class); + when(endpoint.getId()).thenReturn("foo"); + ConcurrentPulsarMessageListenerContainer container = mock(); + when(container.isAutoStartup()).thenReturn(autoStartup); + var factory = mock(PulsarListenerContainerFactory.class); + when(factory.createRegisteredContainer(endpoint)).thenReturn(container); + this.registry.registerListenerContainer(endpoint, factory, startImmediately); + return container; + } + +}