use of org.neo4j.cluster.com.message.Message in project neo4j by neo4j.
the class ElectionStateTest method shouldSendAtomicBroadcastOnJoiningAClusterWithAnEstablishedCoordinator.
@Test
public void shouldSendAtomicBroadcastOnJoiningAClusterWithAnEstablishedCoordinator() throws Throwable {
// Given
String winnerURI = "some://winner";
InstanceId winner = new InstanceId(2);
final List<Message<?>> messages = new ArrayList<>(1);
MessageHolder holder = new MessageHolder() {
@Override
public void offer(Message<? extends MessageType> message) {
messages.add(message);
}
};
ElectionCredentials voteCredentialComparable = mock(ElectionCredentials.class);
ElectionContext electionContext = mock(ElectionContext.class);
when(electionContext.voted(eq(COORDINATOR), eq(new InstanceId(1)), eq(voteCredentialComparable), anyLong())).thenReturn(true);
when(electionContext.getVoteCount(COORDINATOR)).thenReturn(3);
when(electionContext.getNeededVoteCount()).thenReturn(3);
when(electionContext.getElectionWinner(COORDINATOR)).thenReturn(winner);
when(electionContext.getLog(any(Class.class))).thenReturn(NullLog.getInstance());
when(electionContext.newConfigurationStateChange()).thenReturn(mock(ClusterMessage.VersionedConfigurationStateChange.class));
when(electionContext.getUriForId(winner)).thenReturn(URI.create(winnerURI));
// When
Message<ElectionMessage> votedMessage = Message.to(ElectionMessage.voted, URI.create("some://instance"), new ElectionMessage.VotedData(COORDINATOR, new InstanceId(1), voteCredentialComparable));
votedMessage.setHeader(Message.FROM, "some://other");
election.handle(electionContext, votedMessage, holder);
// Then
assertEquals(1, messages.size());
Message<?> message = messages.get(0);
assertEquals(AtomicBroadcastMessage.broadcast, message.getMessageType());
}
use of org.neo4j.cluster.com.message.Message in project neo4j by neo4j.
the class ElectionStateTest method timeoutMakesElectionBeForgotten.
@Test
public void timeoutMakesElectionBeForgotten() throws Throwable {
// Given
String coordinatorRole = "coordinator";
ElectionContext context = mock(ElectionContext.class);
when(context.getLog(Mockito.<Class>any())).thenReturn(NullLog.getInstance());
MessageHolder holder = mock(MessageHolder.class);
Message timeout = Message.timeout(ElectionMessage.electionTimeout, Message.internal(performRoleElections), new ElectionState.ElectionTimeoutData(coordinatorRole, null));
// When
election.handle(context, timeout, holder);
// Then
verify(context, times(1)).forgetElection(coordinatorRole);
}
use of org.neo4j.cluster.com.message.Message in project neo4j by neo4j.
the class HeartbeatIAmAliveProcessorTest method shouldNotGenerateHeartbeatsForSuspicions.
@Test
public void shouldNotGenerateHeartbeatsForSuspicions() throws Exception {
URI to = URI.create("ha://1");
// GIVEN
MessageHolder outgoing = mock(MessageHolder.class);
ClusterContext mockContext = mock(ClusterContext.class);
ClusterConfiguration mockConfiguration = mock(ClusterConfiguration.class);
when(mockConfiguration.getMembers()).thenReturn(new HashMap<InstanceId, URI>() {
{
put(new InstanceId(1), URI.create("ha://1"));
put(new InstanceId(2), URI.create("ha://2"));
}
});
when(mockContext.getConfiguration()).thenReturn(mockConfiguration);
HeartbeatIAmAliveProcessor processor = new HeartbeatIAmAliveProcessor(outgoing, mockContext);
Message incoming = Message.to(HeartbeatMessage.suspicions, to).setHeader(Message.FROM, to.toASCIIString()).setHeader(Message.INSTANCE_ID, "1");
assertEquals(HeartbeatMessage.suspicions, incoming.getMessageType());
// WHEN
processor.process(incoming);
// THEN
verifyZeroInteractions(outgoing);
}
use of org.neo4j.cluster.com.message.Message in project neo4j by neo4j.
the class HeartbeatIAmAliveProcessorTest method shouldNotGenerateHeartbeatsForHeartbeats.
@Test
public void shouldNotGenerateHeartbeatsForHeartbeats() throws Exception {
URI to = URI.create("ha://1");
// GIVEN
MessageHolder outgoing = mock(MessageHolder.class);
ClusterContext mockContext = mock(ClusterContext.class);
ClusterConfiguration mockConfiguration = mock(ClusterConfiguration.class);
when(mockConfiguration.getMembers()).thenReturn(new HashMap<InstanceId, URI>() {
{
put(new InstanceId(1), URI.create("ha://1"));
put(new InstanceId(2), URI.create("ha://2"));
}
});
when(mockContext.getConfiguration()).thenReturn(mockConfiguration);
HeartbeatIAmAliveProcessor processor = new HeartbeatIAmAliveProcessor(outgoing, mockContext);
Message incoming = Message.to(HeartbeatMessage.i_am_alive, to).setHeader(Message.FROM, to.toASCIIString()).setHeader(Message.INSTANCE_ID, "1");
assertEquals(HeartbeatMessage.i_am_alive, incoming.getMessageType());
// WHEN
processor.process(incoming);
// THEN
verifyZeroInteractions(outgoing);
}
use of org.neo4j.cluster.com.message.Message in project neo4j by neo4j.
the class HeartbeatIAmAliveProcessorTest method shouldRevertToInverseUriLookupIfNoInstanceIdHeader.
/*
* This test is required to ensure compatibility with the previous version. If we fail on non existing INSTANCE_ID
* header then heartbeats may pause during rolling upgrades and cause timeouts, which we don't want.
*/
@Test
public void shouldRevertToInverseUriLookupIfNoInstanceIdHeader() throws Exception {
final List<Message> sentOut = new LinkedList<Message>();
String instance2UriString = "ha://2";
// Given
MessageHolder holder = mock(MessageHolder.class);
// The sender, which adds messages outgoing to the list above.
doAnswer(new Answer() {
@Override
public Object answer(InvocationOnMock invocation) throws Throwable {
sentOut.add((Message) invocation.getArguments()[0]);
return null;
}
}).when(holder).offer(Matchers.<Message<MessageType>>any());
ClusterContext mockContext = mock(ClusterContext.class);
ClusterConfiguration mockConfiguration = mock(ClusterConfiguration.class);
when(mockConfiguration.getIdForUri(URI.create(instance2UriString))).thenReturn(new InstanceId(2));
when(mockConfiguration.getMembers()).thenReturn(new HashMap<InstanceId, URI>() {
{
put(new InstanceId(1), URI.create("ha://1"));
put(new InstanceId(2), URI.create("ha://2"));
}
});
when(mockContext.getConfiguration()).thenReturn(mockConfiguration);
HeartbeatIAmAliveProcessor processor = new HeartbeatIAmAliveProcessor(holder, mockContext);
Message incoming = Message.to(mock(MessageType.class), URI.create("ha://someAwesomeInstanceInJapan")).setHeader(Message.FROM, instance2UriString);
// WHEN
processor.process(incoming);
// THEN
assertEquals(1, sentOut.size());
assertEquals(HeartbeatMessage.i_am_alive, sentOut.get(0).getMessageType());
assertEquals(new InstanceId(2), ((HeartbeatMessage.IAmAliveState) sentOut.get(0).getPayload()).getServer());
}
Aggregations