/* TODO: check if test case is needed */
void
TestRaft_follower_recv_appendentries_updates_currentterm_if_term_gt_currentterm(
CuTest * tc)
{
msg_appendentries_t ae;
msg_appendentries_response_t aer;
void *r = raft_new();
raft_add_node(r, (void*)1, 1);
raft_add_node(r, (void*)2, 0);
/* older currentterm */
raft_set_current_term(r, 1);
CuAssertTrue(tc, -1 == raft_get_current_leader(r));
/* newer term for appendentry */
memset(&ae, 0, sizeof(msg_appendentries_t));
/* no prev log idx */
ae.prev_log_idx = 0;
ae.term = 2;
/* appendentry has newer term, so we change our currentterm */
raft_recv_appendentries(r, 1, &ae, &aer);
CuAssertTrue(tc, 1 == aer.success);
/* term has been updated */
CuAssertTrue(tc, 2 == raft_get_current_term(r));
/* and leader has been updated */
CuAssertTrue(tc, 1 == raft_get_current_leader(r));
}
void TestRaft_scenario_leader_appears(CuTest * tc)
{
unsigned long i, j;
raft_server_t *r[3];
void* sender[3];
senders_new();
for (j = 0; j < 3; j++)
sender[j] = sender_new((void*)j);
for (j = 0; j < 3; j++)
{
r[j] = raft_new();
sender_set_raft(sender[j], r[j]);
raft_set_election_timeout(r[j], 500);
raft_add_node(r[j], sender[0], 1, j==0);
raft_add_node(r[j], sender[1], 2, j==1);
raft_add_node(r[j], sender[2], 3, j==2);
raft_set_callbacks(r[j],
&((raft_cbs_t) {
.send_requestvote = sender_requestvote,
.send_appendentries = sender_appendentries,
.persist_term = __raft_persist_term,
.persist_vote = __raft_persist_vote,
.log = NULL
}), sender[j]);
}
void TestRaft_follower_recv_appendentries_add_new_entries_not_already_in_log(
CuTest * tc)
{
void *r = raft_new();
raft_add_node(r, (void*)1, 1);
raft_add_node(r, (void*)2, 0);
raft_set_current_term(r, 1);
msg_appendentries_t ae;
msg_appendentries_response_t aer;
memset(&ae, 0, sizeof(msg_appendentries_t));
ae.term = 1;
ae.prev_log_idx = 0;
ae.prev_log_term = 1;
/* include entries */
msg_entry_t e[2];
memset(&e, 0, sizeof(msg_entry_t) * 2);
e[0].id = 1;
e[1].id = 2;
ae.entries = e;
ae.n_entries = 2;
raft_recv_appendentries(r, 1, &ae, &aer);
CuAssertTrue(tc, 1 == aer.success);
CuAssertTrue(tc, 2 == raft_get_log_count(r));
}
/* Candidate 5.2 */
void TestRaft_follower_dont_grant_vote_if_candidate_has_a_less_complete_log(
CuTest * tc)
{
msg_requestvote_t rv;
msg_requestvote_response_t rvr;
void *r = raft_new();
raft_add_node(r, (void*)1, 1);
raft_add_node(r, (void*)2, 0);
/* request vote */
/* vote indicates candidate's log is not complete compared to follower */
memset(&rv, 0, sizeof(msg_requestvote_t));
rv.term = 1;
rv.candidate_id = 1;
rv.last_log_idx = 1;
rv.last_log_term = 1;
/* server's term and idx are more up-to-date */
raft_set_current_term(r, 1);
raft_entry_t ety;
ety.term = 1;
ety.id = 100;
ety.data.len = 4;
ety.data.buf = (unsigned char*)"aaa";
raft_append_entry(r, &ety);
ety.id = 101;
raft_append_entry(r, &ety);
/* vote not granted */
raft_recv_requestvote(r, 1, &rv, &rvr);
CuAssertTrue(tc, 0 == rvr.vote_granted);
}
/* 5.1 */
void
TestRaft_follower_recv_appendentries_reply_false_if_term_less_than_currentterm(
CuTest * tc)
{
msg_appendentries_t ae;
msg_appendentries_response_t aer, *aerr;
void *r = raft_new();
raft_add_node(r, (void*)1, 1);
raft_add_node(r, (void*)2, 0);
void *sender = sender_new(NULL);
/* no leader known at this point */
CuAssertTrue(tc, -1 == raft_get_current_leader(r));
/* term is low */
memset(&ae, 0, sizeof(msg_appendentries_t));
ae.term = 1;
/* higher current term */
raft_set_current_term(r, 5);
raft_recv_appendentries(r, 1, &ae, &aer);
aerr = sender_poll_msg_data(sender);
CuAssertTrue(tc, NULL != aerr);
CuAssertTrue(tc, 0 == aerr->success);
/* rejected appendentries doesn't change the current leader. */
CuAssertTrue(tc, -1 == raft_get_current_leader(r));
}
void TestRaft_follower_recv_appendentries_does_not_log_if_no_entries_are_specified(
CuTest * tc)
{
msg_appendentries_t ae;
msg_appendentries_response_t aer;
void *r = raft_new();
raft_add_node(r, (void*)1, 1);
raft_add_node(r, (void*)2, 0);
raft_set_state(r, RAFT_STATE_FOLLOWER);
/* log size s */
CuAssertTrue(tc, 0 == raft_get_log_count(r));
/* receive an appendentry with commit */
memset(&ae, 0, sizeof(msg_appendentries_t));
ae.term = 1;
ae.prev_log_term = 1;
ae.prev_log_idx = 4;
ae.leader_commit = 5;
ae.n_entries = 0;
raft_recv_appendentries(r, 1, &ae, &aer);
CuAssertTrue(tc, 0 == raft_get_log_count(r));
}
void TestRaft_server_cfg_sets_num_nodes(CuTest * tc)
{
void *r = raft_new();
raft_add_node(r, (void*)1, 1);
raft_add_node(r, (void*)2, 0);
CuAssertTrue(tc, 2 == raft_get_num_nodes(r));
}
/* 5.3 */
void
TestRaft_follower_recv_appendentries_delete_entries_if_conflict_with_new_entries(
CuTest * tc)
{
msg_appendentries_t ae;
msg_appendentries_response_t aer;
raft_entry_t *ety_appended;
void *r = raft_new();
raft_add_node(r, (void*)1, 1);
raft_add_node(r, (void*)2, 0);
raft_set_current_term(r, 1);
raft_entry_t ety;
/* increase log size */
char *str1 = "111";
ety.data.buf = str1;
ety.data.len = 3;
ety.id = 1;
ety.term = 1;
raft_append_entry(r, &ety);
CuAssertTrue(tc, 1 == raft_get_log_count(r));
/* this log will be overwritten by the appendentries below */
char *str2 = "222";
ety.data.buf = str2;
ety.data.len = 3;
ety.id = 2;
ety.term = 1;
raft_append_entry(r, &ety);
CuAssertTrue(tc, 2 == raft_get_log_count(r));
CuAssertTrue(tc, NULL != (ety_appended = raft_get_entry_from_idx(r, 2)));
CuAssertTrue(tc, !strncmp(ety_appended->data.buf, str2, 3));
/* pass a appendentry that is newer */
msg_entry_t mety;
memset(&ae, 0, sizeof(msg_appendentries_t));
ae.term = 2;
ae.prev_log_idx = 1;
ae.prev_log_term = 1;
/* include one entry */
memset(&mety, 0, sizeof(msg_entry_t));
char *str3 = "333";
mety.data.buf = str3;
mety.data.len = 3;
mety.id = 3;
ae.entries = &mety;
ae.n_entries = 1;
raft_recv_appendentries(r, 1, &ae, &aer);
CuAssertTrue(tc, 1 == aer.success);
CuAssertTrue(tc, 2 == raft_get_log_count(r));
CuAssertTrue(tc, NULL != (ety_appended = raft_get_entry_from_idx(r, 1)));
CuAssertTrue(tc, !strncmp(ety_appended->data.buf, str1, 3));
}
void TestRaft_server_cant_get_node_we_dont_have(CuTest * tc)
{
void *r = raft_new();
raft_add_node(r, (void*)1, 1);
raft_add_node(r, (void*)2, 0);
CuAssertTrue(tc, NULL != raft_get_node(r, 0));
CuAssertTrue(tc, NULL != raft_get_node(r, 1));
CuAssertTrue(tc, NULL == raft_get_node(r, 2));
}
void TestRaft_server_increase_votes_for_me_when_receive_request_vote_response(
CuTest * tc
)
{
void *r = raft_new();
raft_add_node(r, (void*)1, 1);
raft_add_node(r, (void*)2, 0);
raft_set_current_term(r, 1);
CuAssertTrue(tc, 0 == raft_get_nvotes_for_me(r));
msg_requestvote_response_t rvr;
memset(&rvr, 0, sizeof(msg_requestvote_response_t));
rvr.term = 1;
rvr.vote_granted = 1;
raft_recv_requestvote_response(r, 1, &rvr);
CuAssertTrue(tc, 1 == raft_get_nvotes_for_me(r));
}
void TestRaft_follower_becomes_candidate_when_election_timeout_occurs(
CuTest * tc)
{
void *r = raft_new();
/* 1 second election timeout */
raft_set_election_timeout(r, 1000);
raft_add_node(r, (void*)1, 1);
raft_add_node(r, (void*)2, 0);
/* 1.001 seconds have passed */
raft_periodic(r, 1001);
/* is a candidate now */
CuAssertTrue(tc, 1 == raft_is_candidate(r));
}
bool RaftConsensus::addNode(int node_id)
{
OSS::mutex_lock lock(_raftMutex);
raft_node_t* node = raft_add_node(_raft, this, node_id, node_id == _opt.node_id);
if (node)
{
_nodes[node_id] = Node(node);
}
return !!(*node);
}
/* Reply false if term < currentTerm (§5.1) */
void TestRaft_server_recv_requestvote_reply_false_if_term_less_than_current_term(
CuTest * tc
)
{
msg_requestvote_response_t rvr;
void *r = raft_new();
raft_add_node(r, (void*)1, 1);
raft_add_node(r, (void*)2, 0);
raft_set_current_term(r, 2);
/* term is less than current term */
msg_requestvote_t rv;
memset(&rv, 0, sizeof(msg_requestvote_t));
rv.term = 1;
raft_recv_requestvote(r, 1, &rv, &rvr);
CuAssertTrue(tc, 0 == rvr.vote_granted);
}
void TestRaft_server_recv_requestvote_candidate_step_down_if_term_is_higher_than_current_term(
CuTest * tc
)
{
void *r = raft_new();
raft_add_node(r, (void*)1, 1);
raft_add_node(r, (void*)2, 0);
raft_become_candidate(r);
raft_set_current_term(r, 1);
/* current term is less than term */
msg_requestvote_t rv;
memset(&rv, 0, sizeof(msg_requestvote_t));
rv.term = 2;
rv.last_log_idx = 1;
msg_requestvote_response_t rvr;
raft_recv_requestvote(r, 1, &rv, &rvr);
CuAssertTrue(tc, 1 == raft_is_follower(r));
}
void TestRaft_follower_recv_appendentries_set_commitidx_to_LeaderCommit(
CuTest * tc)
{
void *r = raft_new();
raft_add_node(r, (void*)1, 1);
raft_add_node(r, (void*)2, 0);
msg_appendentries_t ae;
msg_appendentries_response_t aer;
memset(&ae, 0, sizeof(msg_appendentries_t));
ae.term = 1;
ae.prev_log_idx = 0;
ae.prev_log_term = 1;
/* include entries */
msg_entry_t e[4];
memset(&e, 0, sizeof(msg_entry_t) * 4);
e[0].term = 1;
e[0].id = 1;
e[1].term = 1;
e[1].id = 2;
e[2].term = 1;
e[2].id = 3;
e[3].term = 1;
e[3].id = 4;
ae.entries = e;
ae.n_entries = 4;
raft_recv_appendentries(r, 1, &ae, &aer);
/* receive an appendentry with commit */
memset(&ae, 0, sizeof(msg_appendentries_t));
ae.term = 1;
ae.prev_log_term = 1;
ae.prev_log_idx = 3;
ae.leader_commit = 3;
/* receipt of appendentries changes commit idx */
raft_recv_appendentries(r, 1, &ae, &aer);
CuAssertTrue(tc, 1 == aer.success);
/* set to 3 because leaderCommit is lower */
CuAssertTrue(tc, 3 == raft_get_commit_idx(r));
}
/* Reply true if term >= currentTerm (§5.1) */
void TestRaft_server_recv_requestvote_reply_true_if_term_greater_than_or_equal_to_current_term(
CuTest * tc
)
{
msg_requestvote_t rv;
msg_requestvote_response_t rvr;
void *r = raft_new();
raft_add_node(r, (void*)1, 1);
raft_add_node(r, (void*)2, 0);
raft_set_current_term(r, 1);
/* term is less than current term */
memset(&rv, 0, sizeof(msg_requestvote_t));
rv.term = 2;
rv.last_log_idx = 1;
raft_recv_requestvote(r, 1, &rv, &rvr);
CuAssertTrue(tc, 1 == rvr.vote_granted);
}
/* If votedFor is null or candidateId, and candidate's log is at
* least as up-to-date as local log, grant vote (§5.2, §5.4) */
void TestRaft_server_dont_grant_vote_if_we_didnt_vote_for_this_candidate(
CuTest * tc
)
{
void *r = raft_new();
raft_add_node(r, (void*)1, 1);
raft_add_node(r, (void*)2, 0);
raft_set_current_term(r, 1);
raft_vote(r, 0);
msg_requestvote_t rv;
memset(&rv, 0, sizeof(msg_requestvote_t));
rv.term = 1;
rv.candidate_id = 1;
rv.last_log_idx = 1;
rv.last_log_term = 1;
msg_requestvote_response_t rvr;
raft_recv_requestvote(r, 1, &rv, &rvr);
CuAssertTrue(tc, 0 == rvr.vote_granted);
}
/* 5.3 */
void
TestRaft_follower_recv_appendentries_reply_false_if_doesnt_have_log_at_prev_log_idx_which_matches_prev_log_term(
CuTest * tc)
{
msg_entry_t ety;
char *str = "aaa";
msg_appendentries_t ae;
msg_appendentries_response_t aer;
void *r = raft_new();
raft_add_node(r, (void*)1, 1);
raft_add_node(r, (void*)2, 0);
/* term is different from appendentries */
raft_set_current_term(r, 2);
raft_set_commit_idx(r, 1);
raft_set_last_applied_idx(r, 1);
// TODO at log manually?
/* log idx that server doesn't have */
memset(&ae, 0, sizeof(msg_appendentries_t));
ae.term = 2;
ae.prev_log_idx = 1;
/* prev_log_term is less than current term (ie. 2) */
ae.prev_log_term = 1;
/* include one entry */
memset(&ety, 0, sizeof(msg_entry_t));
ety.data.buf = str;
ety.data.len = 3;
ety.id = 1;
ae.entries = &ety;
ae.n_entries = 1;
/* trigger reply */
raft_recv_appendentries(r, 1, &ae, &aer);
/* reply is false */
CuAssertTrue(tc, 0 == aer.success);
}
void TestRaft_follower_recv_appendentries_increases_log(CuTest * tc)
{
msg_appendentries_t ae;
msg_entry_t ety;
msg_appendentries_response_t aer;
char *str = "aaa";
void *r = raft_new();
raft_add_node(r, (void*)1, 1);
raft_add_node(r, (void*)2, 0);
raft_set_state(r, RAFT_STATE_FOLLOWER);
/* log size s */
CuAssertTrue(tc, 0 == raft_get_log_count(r));
/* receive an appendentry with commit */
memset(&ae, 0, sizeof(msg_appendentries_t));
ae.term = 3;
ae.prev_log_term = 1;
/* first appendentries msg */
ae.prev_log_idx = 0;
ae.leader_commit = 5;
/* include one entry */
memset(&ety, 0, sizeof(msg_entry_t));
ety.data.buf = str;
ety.data.len = 3;
ety.id = 1;
/* check that old terms are passed onto the log */
ety.term = 2;
ae.entries = &ety;
ae.n_entries = 1;
raft_recv_appendentries(r, 1, &ae, &aer);
CuAssertTrue(tc, 1 == aer.success);
CuAssertTrue(tc, 1 == raft_get_log_count(r));
raft_entry_t* log = raft_get_entry_from_idx(r, 1);
CuAssertTrue(tc, 2 == log->term);
}
raft_node_t* raft_add_non_voting_node(raft_server_t* me_, void* udata, int id, int is_self)
{
raft_node_t* node = raft_add_node(me_, udata, id, is_self);
raft_node_set_voting(node, 0);
return node;
}
