Examples with RaftState org.neo4j.causalclustering.core.consensus.state.RaftState used on opensource projects

Example 1 with RaftState

use of org.neo4j.causalclustering.core.consensus.state.RaftState in project neo4j by neo4j.

the class HeartbeatTest method shouldNotResultInCommitIfHistoryMismatches.

@Test
public void shouldNotResultInCommitIfHistoryMismatches() throws Exception {
    InMemoryRaftLog raftLog = new InMemoryRaftLog();
    RaftState state = raftState().myself(myself).entryLog(raftLog).build();
    long leaderTerm = state.term() + leaderTermDifference;
    raftLog.append(new RaftLogEntry(leaderTerm, content()));
    RaftMessages.Heartbeat heartbeat = heartbeat().from(leader).commitIndex(// The leader is talking about committing stuff we don't know about
    raftLog.appendIndex()).commitIndexTerm(// And is in the same term
    leaderTerm).leaderTerm(leaderTerm).build();
    Outcome outcome = role.handler.handle(heartbeat, state, log());
    assertThat(outcome.getCommitIndex(), Matchers.equalTo(0L));
}

Example 2 with RaftState

use of org.neo4j.causalclustering.core.consensus.state.RaftState in project neo4j by neo4j.

the class LeaderTest method leaderShouldNotStepDownWhenReceivedQuorumOfHeartbeatResponses.

@Test
public void leaderShouldNotStepDownWhenReceivedQuorumOfHeartbeatResponses() throws Exception {
    // given
    RaftState state = raftState().votingMembers(asSet(myself, member1, member2)).build();
    Leader leader = new Leader();
    // when
    Outcome outcome = leader.handle(new RaftMessages.HeartbeatResponse(member1), state, log());
    state.update(outcome);
    // we now have quorum and should not step down
    outcome = leader.handle(new RaftMessages.Timeout.Election(myself), state, log());
    // then
    assertThat(outcome.getRole(), is(LEADER));
}

Example 3 with RaftState

use of org.neo4j.causalclustering.core.consensus.state.RaftState in project neo4j by neo4j.

the class LeaderTest method leaderShouldDecideToAppendToItsLogAndSendAppendEntriesMessageOnReceiptOfClientProposal.

@Test
public void leaderShouldDecideToAppendToItsLogAndSendAppendEntriesMessageOnReceiptOfClientProposal() throws Exception {
    // given
    RaftState state = raftState().votingMembers(asSet(myself, member1, member2)).build();
    Leader leader = new Leader();
    RaftMessages.NewEntry.Request newEntryRequest = new RaftMessages.NewEntry.Request(member(9), CONTENT);
    // when
    Outcome outcome = leader.handle(newEntryRequest, state, log());
    //state.update( outcome );
    // then
    AppendLogEntry logCommand = (AppendLogEntry) single(outcome.getLogCommands());
    assertEquals(0, logCommand.index);
    assertEquals(0, logCommand.entry.term());
    ShipCommand.NewEntries shipCommand = (ShipCommand.NewEntries) single(outcome.getShipCommands());
    assertEquals(shipCommand, new ShipCommand.NewEntries(-1, -1, new RaftLogEntry[] { new RaftLogEntry(0, CONTENT) }));
}

Example 4 with RaftState

use of org.neo4j.causalclustering.core.consensus.state.RaftState in project neo4j by neo4j.

the class LeaderTest method leaderShouldHandleBatch.

@Test
public void leaderShouldHandleBatch() throws Exception {
    // given
    RaftState state = raftState().votingMembers(asSet(myself, member1, member2)).build();
    Leader leader = new Leader();
    int BATCH_SIZE = 3;
    RaftMessages.NewEntry.BatchRequest batchRequest = new RaftMessages.NewEntry.BatchRequest(BATCH_SIZE);
    batchRequest.add(valueOf(0));
    batchRequest.add(valueOf(1));
    batchRequest.add(valueOf(2));
    // when
    Outcome outcome = leader.handle(batchRequest, state, log());
    // then
    BatchAppendLogEntries logCommand = (BatchAppendLogEntries) single(outcome.getLogCommands());
    assertEquals(0, logCommand.baseIndex);
    for (int i = 0; i < BATCH_SIZE; i++) {
        assertEquals(0, logCommand.entries[i].term());
        assertEquals(i, ((ReplicatedInteger) logCommand.entries[i].content()).get());
    }
    ShipCommand.NewEntries shipCommand = (ShipCommand.NewEntries) single(outcome.getShipCommands());
    assertEquals(shipCommand, new ShipCommand.NewEntries(-1, -1, new RaftLogEntry[] { new RaftLogEntry(0, valueOf(0)), new RaftLogEntry(0, valueOf(1)), new RaftLogEntry(0, valueOf(2)) }));
}

Example 5 with RaftState

use of org.neo4j.causalclustering.core.consensus.state.RaftState in project neo4j by neo4j.

the class LeaderTest method leaderShouldStepDownWhenLackingHeartbeatResponses.

@Test
public void leaderShouldStepDownWhenLackingHeartbeatResponses() throws Exception {
    // given
    RaftState state = raftState().votingMembers(asSet(myself, member1, member2)).leader(myself).build();
    Leader leader = new Leader();
    leader.handle(new RaftMessages.Timeout.Election(myself), state, log());
    // when
    Outcome outcome = leader.handle(new RaftMessages.Timeout.Election(myself), state, log());
    // then
    assertThat(outcome.getRole(), not(LEADER));
    assertNull(outcome.getLeader());
}