Search in sources :

Example 1 with TopologyChangeEvent

use of com.datastax.oss.protocol.internal.response.event.TopologyChangeEvent in project java-driver by datastax.

the class ControlConnectionEventsTest method should_process_topology_change_events.

@Test
public void should_process_topology_change_events() {
    // Given
    DriverChannel channel1 = newMockDriverChannel(1);
    ArgumentCaptor<DriverChannelOptions> optionsCaptor = ArgumentCaptor.forClass(DriverChannelOptions.class);
    when(channelFactory.connect(eq(node1), optionsCaptor.capture())).thenReturn(CompletableFuture.completedFuture(channel1));
    controlConnection.init(true, false, false);
    await().until(() -> optionsCaptor.getValue() != null);
    EventCallback callback = optionsCaptor.getValue().eventCallback;
    TopologyChangeEvent event = new TopologyChangeEvent(ProtocolConstants.TopologyChangeType.NEW_NODE, ADDRESS1);
    // When
    callback.onEvent(event);
    // Then
    verify(eventBus).fire(TopologyEvent.suggestAdded(ADDRESS1));
}
Also used : DriverChannel(com.datastax.oss.driver.internal.core.channel.DriverChannel) DriverChannelOptions(com.datastax.oss.driver.internal.core.channel.DriverChannelOptions) TopologyChangeEvent(com.datastax.oss.protocol.internal.response.event.TopologyChangeEvent) EventCallback(com.datastax.oss.driver.internal.core.channel.EventCallback) Test(org.junit.Test)

Aggregations

DriverChannel (com.datastax.oss.driver.internal.core.channel.DriverChannel)1 DriverChannelOptions (com.datastax.oss.driver.internal.core.channel.DriverChannelOptions)1 EventCallback (com.datastax.oss.driver.internal.core.channel.EventCallback)1 TopologyChangeEvent (com.datastax.oss.protocol.internal.response.event.TopologyChangeEvent)1 Test (org.junit.Test)1