Refactoring the test and adding comments

andreev-io · Aug 21, 2021 · 6bf7889 · 6bf7889
1 parent 2645fe1
commit 6bf7889
Show file tree

Hide file tree

Showing 4 changed files with 759 additions and 282 deletions.
diff --git a/little_raft/src/replica.rs b/little_raft/src/replica.rs
@@ -259,7 +259,7 @@ where
     }
 
     fn process_message(&mut self, message: Message<T>) {
-        match self.state {  
+        match self.state {
             State::Leader => self.process_message_as_leader(message),
             State::Candidate => self.process_message_as_candidate(message),
             State::Follower => self.process_message_as_follower(message),
@@ -441,6 +441,7 @@ where
                         vote_granted: true,
                     },
                 );
+                self.voted_for = Some(from_id);
             } else {
                 // If the criteria are not met, do not grant the vote.
                 self.cluster.lock().unwrap().send_message(

diff --git a/little_raft/tests/raft.rs b/little_raft/tests/raft.rs