use of org.neo4j.cluster.protocol.cluster.ClusterContext in project neo4j by neo4j.
the class ElectionContextTest method electorLeavingAndRejoiningWithNoElectionsInBetweenMustStillHaveElectionsGoThrough.
/*
* This assumes an instance leaves the cluster normally and then rejoins, without any elections in between. The
* expected result is that it will succeed in sending election results.
*/
@Test
public void electorLeavingAndRejoiningWithNoElectionsInBetweenMustStillHaveElectionsGoThrough() throws Exception {
// Given
final String role1 = "coordinator1";
InstanceId me = new InstanceId(1);
InstanceId leavingInstance = new InstanceId(2);
InstanceId forQuorum = new InstanceId(3);
Config config = mock(Config.class);
when(config.get(ClusterSettings.max_acceptors)).thenReturn(10);
ClusterConfiguration clusterConfiguration = mock(ClusterConfiguration.class);
List<InstanceId> clusterMemberIds = new LinkedList<InstanceId>();
clusterMemberIds.add(leavingInstance);
clusterMemberIds.add(me);
clusterMemberIds.add(forQuorum);
when(clusterConfiguration.getMemberIds()).thenReturn(clusterMemberIds);
MultiPaxosContext context = new MultiPaxosContext(me, Iterables.<ElectionRole, ElectionRole>iterable(new ElectionRole(role1)), clusterConfiguration, new Executor() {
@Override
public void execute(Runnable command) {
command.run();
}
}, NullLogProvider.getInstance(), mock(ObjectInputStreamFactory.class), mock(ObjectOutputStreamFactory.class), mock(AcceptorInstanceStore.class), mock(Timeouts.class), mock(ElectionCredentialsProvider.class), config);
HeartbeatContext heartbeatContext = context.getHeartbeatContext();
ClusterContext clusterContext = context.getClusterContext();
clusterContext.setLastElector(leavingInstance);
clusterContext.setLastElectorVersion(8);
// When the elector leaves the cluster
clusterContext.left(leavingInstance);
// Then the elector is reset to defaults
assertEquals(clusterContext.getLastElector(), InstanceId.NONE);
assertEquals(clusterContext.getLastElectorVersion(), ClusterContext.NO_ELECTOR_VERSION);
// When the elector comes back with an election result
// We don't need to join, election results do not check for elector membership
clusterContext.elected(role1, forQuorum, leavingInstance, 9);
// Then the result is actually respected
assertEquals(clusterContext.getLastElector(), leavingInstance);
assertEquals(clusterContext.getLastElectorVersion(), 9);
}
use of org.neo4j.cluster.protocol.cluster.ClusterContext in project neo4j by neo4j.
the class ElectionContextTest method failedElectorRejoiningMustHaveItsVersionFromVoteRequestsSetTheElectorVersion.
@Test
public void failedElectorRejoiningMustHaveItsVersionFromVoteRequestsSetTheElectorVersion() throws Throwable {
// Given
final String role1 = "coordinator1";
InstanceId me = new InstanceId(1);
InstanceId failingInstance = new InstanceId(2);
InstanceId forQuorum = new InstanceId(3);
Config config = mock(Config.class);
when(config.get(ClusterSettings.max_acceptors)).thenReturn(10);
ClusterConfiguration clusterConfiguration = mock(ClusterConfiguration.class);
List<InstanceId> clusterMemberIds = new LinkedList<InstanceId>();
clusterMemberIds.add(failingInstance);
clusterMemberIds.add(me);
clusterMemberIds.add(forQuorum);
when(clusterConfiguration.getMemberIds()).thenReturn(clusterMemberIds);
MultiPaxosContext context = new MultiPaxosContext(me, Iterables.iterable(new ElectionRole(role1)), clusterConfiguration, new Executor() {
@Override
public void execute(Runnable command) {
command.run();
}
}, NullLogProvider.getInstance(), mock(ObjectInputStreamFactory.class), mock(ObjectOutputStreamFactory.class), mock(AcceptorInstanceStore.class), mock(Timeouts.class), mock(ElectionCredentialsProvider.class), config);
HeartbeatContext heartbeatContext = context.getHeartbeatContext();
ClusterContext clusterContext = context.getClusterContext();
clusterContext.setLastElector(failingInstance);
clusterContext.setLastElectorVersion(8);
// When the elector fails
heartbeatContext.suspicions(forQuorum, Collections.singleton(failingInstance));
heartbeatContext.suspect(failingInstance);
// Then the elector is reset to defaults
assertEquals(clusterContext.getLastElector(), InstanceId.NONE);
assertEquals(clusterContext.getLastElectorVersion(), ClusterContext.NO_ELECTOR_VERSION);
// When the elector comes back with an election result
clusterContext.elected(role1, forQuorum, failingInstance, 9);
// Then the result is actually respected
assertEquals(clusterContext.getLastElector(), failingInstance);
assertEquals(clusterContext.getLastElectorVersion(), 9);
}
use of org.neo4j.cluster.protocol.cluster.ClusterContext in project neo4j by neo4j.
the class ElectionContextTest method twoVotesFromSameInstanceForSameRoleShouldBeConsolidated.
@Test
public void twoVotesFromSameInstanceForSameRoleShouldBeConsolidated() throws Exception {
// Given
final String coordinatorRole = "coordinator";
HeartbeatContext heartbeatContext = mock(HeartbeatContext.class);
when(heartbeatContext.getFailed()).thenReturn(Collections.<InstanceId>emptySet());
Map<InstanceId, URI> members = new HashMap<InstanceId, URI>();
members.put(new InstanceId(1), URI.create("server1"));
members.put(new InstanceId(2), URI.create("server2"));
members.put(new InstanceId(3), URI.create("server3"));
Config config = mock(Config.class);
when(config.get(ClusterSettings.max_acceptors)).thenReturn(10);
ClusterConfiguration clusterConfiguration = mock(ClusterConfiguration.class);
when(clusterConfiguration.getMembers()).thenReturn(members);
ClusterContext clusterContext = mock(ClusterContext.class);
when(clusterContext.getConfiguration()).thenReturn(clusterConfiguration);
MultiPaxosContext context = new MultiPaxosContext(new InstanceId(1), Iterables.iterable(new ElectionRole(coordinatorRole)), clusterConfiguration, mock(Executor.class), NullLogProvider.getInstance(), mock(ObjectInputStreamFactory.class), mock(ObjectOutputStreamFactory.class), mock(AcceptorInstanceStore.class), mock(Timeouts.class), mock(ElectionCredentialsProvider.class), config);
ElectionContext toTest = context.getElectionContext();
// When
toTest.startElectionProcess(coordinatorRole);
toTest.voted(coordinatorRole, new InstanceId(1), new IntegerElectionCredentials(100), ClusterContext.NO_ELECTOR_VERSION);
toTest.voted(coordinatorRole, new InstanceId(2), new IntegerElectionCredentials(100), ClusterContext.NO_ELECTOR_VERSION);
toTest.voted(coordinatorRole, new InstanceId(2), new IntegerElectionCredentials(101), ClusterContext.NO_ELECTOR_VERSION);
// Then
assertNull(toTest.getElectionWinner(coordinatorRole));
assertEquals(2, toTest.getVoteCount(coordinatorRole));
}
use of org.neo4j.cluster.protocol.cluster.ClusterContext in project neo4j by neo4j.
the class ElectionContextTest method testElectionNotOkQuorumFailedTwoInstances.
@Test
public void testElectionNotOkQuorumFailedTwoInstances() {
Set<InstanceId> failed = new HashSet<InstanceId>();
failed.add(new InstanceId(2));
Map<InstanceId, URI> members = new HashMap<InstanceId, URI>();
members.put(new InstanceId(1), URI.create("server1"));
members.put(new InstanceId(2), URI.create("server2"));
Config config = mock(Config.class);
when(config.get(ClusterSettings.max_acceptors)).thenReturn(10);
ClusterConfiguration clusterConfiguration = mock(ClusterConfiguration.class);
when(clusterConfiguration.getMembers()).thenReturn(members);
ClusterContext clusterContext = mock(ClusterContext.class);
when(clusterContext.getConfiguration()).thenReturn(clusterConfiguration);
MultiPaxosContext context = new MultiPaxosContext(new InstanceId(1), Iterables.<ElectionRole, ElectionRole>iterable(new ElectionRole("coordinator")), clusterConfiguration, mock(Executor.class), NullLogProvider.getInstance(), mock(ObjectInputStreamFactory.class), mock(ObjectOutputStreamFactory.class), mock(AcceptorInstanceStore.class), mock(Timeouts.class), mock(ElectionCredentialsProvider.class), config);
context.getHeartbeatContext().getFailed().addAll(failed);
ElectionContext toTest = context.getElectionContext();
assertFalse(toTest.electionOk());
}
use of org.neo4j.cluster.protocol.cluster.ClusterContext in project neo4j by neo4j.
the class ElectionContextTest method baseTestForElectionOk.
private void baseTestForElectionOk(Set<InstanceId> failed, boolean moreThanQuorum) {
Map<InstanceId, URI> members = new HashMap<InstanceId, URI>();
members.put(new InstanceId(1), URI.create("server1"));
members.put(new InstanceId(2), URI.create("server2"));
members.put(new InstanceId(3), URI.create("server3"));
Config config = mock(Config.class);
when(config.get(ClusterSettings.max_acceptors)).thenReturn(10);
ClusterConfiguration clusterConfiguration = mock(ClusterConfiguration.class);
when(clusterConfiguration.getMembers()).thenReturn(members);
ClusterContext clusterContext = mock(ClusterContext.class);
when(clusterContext.getConfiguration()).thenReturn(clusterConfiguration);
MultiPaxosContext context = new MultiPaxosContext(new InstanceId(1), Iterables.iterable(new ElectionRole("coordinator")), clusterConfiguration, mock(Executor.class), NullLogProvider.getInstance(), mock(ObjectInputStreamFactory.class), mock(ObjectOutputStreamFactory.class), mock(AcceptorInstanceStore.class), mock(Timeouts.class), mock(ElectionCredentialsProvider.class), config);
context.getHeartbeatContext().getFailed().addAll(failed);
ElectionContext toTest = context.getElectionContext();
assertEquals(moreThanQuorum, !toTest.electionOk());
}
Aggregations