sr: fix Raft replay leader election

1 files changed, 41 insertions, 0 deletions

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
@@ -390,6 +390,33 @@ class VSRaftProtocolTest {
     }
 
     @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 {
         protocol.currentContextIsServer(false);
@@ -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);