Search in sources :

Example 11 with RaftMachine

use of org.neo4j.causalclustering.core.consensus.RaftMachine in project neo4j by neo4j.

the class ElectionTest method candidateShouldWinElectionAndBecomeLeader.

@Test
public void candidateShouldWinElectionAndBecomeLeader() throws Exception {
    // given
    FakeClock fakeClock = Clocks.fakeClock();
    ControlledRenewableTimeoutService timeouts = new ControlledRenewableTimeoutService(fakeClock);
    RaftMachine raft = new RaftMachineBuilder(myself, 3, RaftTestMemberSetBuilder.INSTANCE).outbound(outbound).timeoutService(timeouts).clock(fakeClock).build();
    raft.installCoreState(new RaftCoreState(new MembershipEntry(0, asSet(myself, member1, member2))));
    raft.startTimers();
    timeouts.invokeTimeout(RaftMachine.Timeouts.ELECTION);
    // when
    raft.handle(voteResponse().from(member1).term(1).grant().build());
    raft.handle(voteResponse().from(member2).term(1).grant().build());
    // then
    assertEquals(1, raft.term());
    assertEquals(LEADER, raft.currentRole());
    verify(outbound).send(eq(member1), isA(RaftMessages.AppendEntries.Request.class));
    verify(outbound).send(eq(member2), isA(RaftMessages.AppendEntries.Request.class));
}
Also used : RaftMachine(org.neo4j.causalclustering.core.consensus.RaftMachine) RaftMachineBuilder(org.neo4j.causalclustering.core.consensus.RaftMachineBuilder) MembershipEntry(org.neo4j.causalclustering.core.consensus.membership.MembershipEntry) RaftCoreState(org.neo4j.causalclustering.core.state.snapshot.RaftCoreState) FakeClock(org.neo4j.time.FakeClock) TestMessageBuilders.voteRequest(org.neo4j.causalclustering.core.consensus.TestMessageBuilders.voteRequest) RaftMessages(org.neo4j.causalclustering.core.consensus.RaftMessages) ControlledRenewableTimeoutService(org.neo4j.causalclustering.core.consensus.schedule.ControlledRenewableTimeoutService) Test(org.junit.Test)

Example 12 with RaftMachine

use of org.neo4j.causalclustering.core.consensus.RaftMachine in project neo4j by neo4j.

the class ElectionTest method candidateShouldVoteForTheSameCandidateInTheSameTerm.

@Test
public void candidateShouldVoteForTheSameCandidateInTheSameTerm() throws Exception {
    // given
    FakeClock fakeClock = Clocks.fakeClock();
    ControlledRenewableTimeoutService timeouts = new ControlledRenewableTimeoutService(fakeClock);
    RaftMachine raft = new RaftMachineBuilder(myself, 3, RaftTestMemberSetBuilder.INSTANCE).outbound(outbound).timeoutService(timeouts).clock(fakeClock).build();
    raft.installCoreState(new RaftCoreState(new MembershipEntry(0, asSet(myself, member1, member2))));
    // when
    raft.handle(voteRequest().from(member1).candidate(member1).term(1).build());
    raft.handle(voteRequest().from(member1).candidate(member1).term(1).build());
    // then
    verify(outbound, times(2)).send(member1, voteResponse().term(1).grant().build());
}
Also used : RaftMachine(org.neo4j.causalclustering.core.consensus.RaftMachine) RaftMachineBuilder(org.neo4j.causalclustering.core.consensus.RaftMachineBuilder) MembershipEntry(org.neo4j.causalclustering.core.consensus.membership.MembershipEntry) RaftCoreState(org.neo4j.causalclustering.core.state.snapshot.RaftCoreState) FakeClock(org.neo4j.time.FakeClock) ControlledRenewableTimeoutService(org.neo4j.causalclustering.core.consensus.schedule.ControlledRenewableTimeoutService) Test(org.junit.Test)

Aggregations

RaftMachine (org.neo4j.causalclustering.core.consensus.RaftMachine)12 Test (org.junit.Test)10 ExposedRaftState (org.neo4j.causalclustering.core.consensus.state.ExposedRaftState)5 OnDemandJobScheduler (org.neo4j.test.OnDemandJobScheduler)5 TimeoutException (java.util.concurrent.TimeoutException)4 RaftMachineBuilder (org.neo4j.causalclustering.core.consensus.RaftMachineBuilder)3 MembershipEntry (org.neo4j.causalclustering.core.consensus.membership.MembershipEntry)3 ControlledRenewableTimeoutService (org.neo4j.causalclustering.core.consensus.schedule.ControlledRenewableTimeoutService)3 RaftCoreState (org.neo4j.causalclustering.core.state.snapshot.RaftCoreState)3 FakeClock (org.neo4j.time.FakeClock)3 ArrayList (java.util.ArrayList)2 RaftMessages (org.neo4j.causalclustering.core.consensus.RaftMessages)2 TestMessageBuilders.voteRequest (org.neo4j.causalclustering.core.consensus.TestMessageBuilders.voteRequest)2 InMemoryRaftLog (org.neo4j.causalclustering.core.consensus.log.InMemoryRaftLog)2 RaftLogEntry (org.neo4j.causalclustering.core.consensus.log.RaftLogEntry)2 MemberId (org.neo4j.causalclustering.identity.MemberId)2 ProcedureException (org.neo4j.kernel.api.exceptions.ProcedureException)2 HashMap (java.util.HashMap)1 List (java.util.List)1 CompletableFuture (java.util.concurrent.CompletableFuture)1