Fix Raft follower timeout re-arming tests (0bac83d3-1322-4940-a9ee-58eb1e0d6245)

1 files changed, 16 insertions, 0 deletions

diff --git a/src/main/java/protocols/implementations/VSRaftProtocol.java b/src/main/java/protocols/implementations/VSRaftProtocol.java
index 4a310ee..eaf63a7 100644
--- a/src/main/java/protocols/implementations/VSRaftProtocol.java
+++ b/src/main/java/protocols/implementations/VSRaftProtocol.java
@@ -166,6 +166,7 @@ public class VSRaftProtocol extends VSAbstractProtocol {
      * @param newLeaderId the known leader in that term, or -1 if unknown
      */
     private void becomeFollower(int term, int newLeaderId) {
+        clearServerSchedules();
         isLeader = false;
         isCandidate = false;
         currentTerm = term;
@@ -181,9 +182,24 @@ public class VSRaftProtocol extends VSAbstractProtocol {
     private void resetElectionTimeout() {
         long jitterPercentage = Math.abs(process.getRandomPercentage());
         long jitter = (getLong("electionJitter") * jitterPercentage) / 100L;
+        boolean previousContextIsServer = currentContextIsServer();
 
+        currentContextIsServer(false);
         removeSchedules();
         scheduleAt(process.getTime() + getLong("electionTimeout") + jitter);
+        currentContextIsServer(previousContextIsServer);
+    }
+
+    /**
+     * Clears any active server-side schedules while preserving the caller
+     * context.
+     */
+    private void clearServerSchedules() {
+        boolean previousContextIsServer = currentContextIsServer();
+
+        currentContextIsServer(true);
+        removeSchedules();
+        currentContextIsServer(previousContextIsServer);
     }
 
     /**