diff --git a/sdk/servicebus/azure-messaging-servicebus/src/test/java/com/azure/messaging/servicebus/implementation/ServiceBusReactorReceiverTest.java b/sdk/servicebus/azure-messaging-servicebus/src/test/java/com/azure/messaging/servicebus/implementation/ServiceBusReactorReceiverTest.java index 45b8845781a21..1166976704fda 100644 --- a/sdk/servicebus/azure-messaging-servicebus/src/test/java/com/azure/messaging/servicebus/implementation/ServiceBusReactorReceiverTest.java +++ b/sdk/servicebus/azure-messaging-servicebus/src/test/java/com/azure/messaging/servicebus/implementation/ServiceBusReactorReceiverTest.java @@ -51,6 +51,8 @@ class ServiceBusReactorReceiverTest { private static final String ENTITY_PATH = "queue-name"; private static final String LINK_NAME = "a-link-name"; + private static final String CONNECTION_ID = "a-connection-id"; + private final ClientLogger logger = new ClientLogger(ServiceBusReactorReceiver.class); private final EmitterProcessor endpointStates = EmitterProcessor.create(); @@ -107,6 +109,7 @@ void setup(TestInfo testInfo) throws IOException { when(receiveLinkHandler.getDeliveredMessages()).thenReturn(deliveryProcessor); when(receiveLinkHandler.getLinkName()).thenReturn(LINK_NAME); when(receiveLinkHandler.getEndpointStates()).thenReturn(endpointStates); + when(receiveLinkHandler.getConnectionId()).thenReturn(CONNECTION_ID); when(tokenManager.getAuthorizationResults()).thenReturn(Flux.create(sink -> sink.next(AmqpResponseCode.OK)));