From 0bebec08cd89039c32bd9b9e73d80d573b6bf0b3 Mon Sep 17 00:00:00 2001
From: Paul Buetow <paul@buetow.org>
Date: Fri, 27 Mar 2026 13:30:14 +0200
Subject: sr: fix Raft replay leader election

---
 .../implementations/VSRaftProtocolTest.java        | 41 ++++++++++++++++++++++
 1 file changed, 41 insertions(+)

(limited to 'src/test/java/protocols/implementations/VSRaftProtocolTest.java')

diff --git a/src/test/java/protocols/implementations/VSRaftProtocolTest.java b/src/test/java/protocols/implementations/VSRaftProtocolTest.java
index 40dfd10..802c80f 100644
--- a/src/test/java/protocols/implementations/VSRaftProtocolTest.java
+++ b/src/test/java/protocols/implementations/VSRaftProtocolTest.java
@@ -389,6 +389,33 @@ class VSRaftProtocolTest {
         assertEquals(4750L, taskCaptor.getValue().getTaskTime());
     }
 
+    @Test
+    void testDualRoleVoteRequestIsHandledOnce() throws Exception {
+        protocol.currentContextIsServer(false);
+        protocol.isClient(true);
+        protocol.isServer(true);
+        protocol.onInit();
+        clearInvocations(mockProcess, mockTaskManager);
+        when(mockProcess.getTime()).thenReturn(200L, 200L);
+
+        VSMessage voteRequest = new VSMessage();
+        setMessageProtocolClassname(voteRequest, VSRaftProtocol.class.getName());
+        setMessageServerFlag(voteRequest, true);
+        voteRequest.setString("type", "voteRequest");
+        voteRequest.setInteger("term", 2);
+        voteRequest.setInteger("candidateId", 11);
+
+        ArgumentCaptor<VSMessage> messageCaptor =
+            ArgumentCaptor.forClass(VSMessage.class);
+
+        protocol.onMessageRecvStart(voteRequest);
+
+        verify(mockProcess).sendMessage(messageCaptor.capture());
+        assertEquals("voteResponse", messageCaptor.getValue().getString("type"));
+        assertEquals(2, messageCaptor.getValue().getInteger("term"));
+        assertTrue(messageCaptor.getValue().getBoolean("voteGranted"));
+    }
+
     @Test
     void testClientReceiveHeartbeatBecomesFollowerResetsTimeoutAndSendsAck()
     throws Exception {
@@ -764,6 +791,20 @@ class VSRaftProtocolTest {
         field.setBoolean(protocol, value);
     }
 
+    private void setMessageProtocolClassname(VSMessage message, String classname)
+    throws Exception {
+        Field field = VSMessage.class.getDeclaredField("protocolClassname");
+        field.setAccessible(true);
+        field.set(message, classname);
+    }
+
+    private void setMessageServerFlag(VSMessage message, boolean isServerMessage)
+    throws Exception {
+        Field field = VSMessage.class.getDeclaredField("isServerMessage");
+        field.setAccessible(true);
+        field.setBoolean(message, isServerMessage);
+    }
+
     private int getIntField(String fieldName) throws Exception {
         Field field = VSRaftProtocol.class.getDeclaredField(fieldName);
         field.setAccessible(true);
-- 
cgit v1.2.3