opendaylight/md-sal/sal-akka-raft/src/test/java/org/opendaylight/controller/cluster/raft/messages/AppendEntriesTest.java

   1 /*
   2  * Copyright (c) 2015 Brocade Communications Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.controller.cluster.raft.messages;
   9
  10 import static org.junit.Assert.assertEquals;
  11
  12 import java.util.Iterator;
  13 import java.util.List;
  14 import org.apache.commons.lang3.SerializationUtils;
  15 import org.junit.Test;
  16 import org.opendaylight.controller.cluster.raft.MockRaftActorContext.MockPayload;
  17 import org.opendaylight.controller.cluster.raft.RaftVersions;
  18 import org.opendaylight.controller.cluster.raft.ReplicatedLogEntry;
  19 import org.opendaylight.controller.cluster.raft.persisted.SimpleReplicatedLogEntry;
  20
  21 /**
  22  * Unit tests for AppendEntries.
  23  *
  24  * @author Thomas Pantelis
  25  */
  26 public class AppendEntriesTest {
  27     @Test
  28     public void testSerialization() {
  29         ReplicatedLogEntry entry1 = new SimpleReplicatedLogEntry(1, 2, new MockPayload("payload1"));
  30
  31         ReplicatedLogEntry entry2 = new SimpleReplicatedLogEntry(3, 4, new MockPayload("payload2"));
  32
  33         short payloadVersion = 5;
  34
  35         // Without leader address
  36
  37         var expected = new AppendEntries(5L, "node1", 7L, 8L, List.of(entry1, entry2), 10L, -1, payloadVersion,
  38             RaftVersions.CURRENT_VERSION, null);
  39
  40         var bytes = SerializationUtils.serialize(expected);
  41         assertEquals(285, bytes.length);
  42         var cloned = (AppendEntries) SerializationUtils.deserialize(bytes);
  43
  44         verifyAppendEntries(expected, cloned, RaftVersions.CURRENT_VERSION);
  45
  46         // With leader address
  47
  48         expected = new AppendEntries(5L, "node1", 7L, 8L, List.of(entry1, entry2), 10L, -1, payloadVersion,
  49             RaftVersions.CURRENT_VERSION, "leader address");
  50
  51         bytes = SerializationUtils.serialize(expected);
  52         assertEquals(301, bytes.length);
  53         cloned = (AppendEntries) SerializationUtils.deserialize(bytes);
  54
  55         verifyAppendEntries(expected, cloned, RaftVersions.CURRENT_VERSION);
  56     }
  57
  58     private static void verifyAppendEntries(final AppendEntries expected, final AppendEntries actual,
  59             final short recipientRaftVersion) {
  60         assertEquals("getLeaderId", expected.getLeaderId(), actual.getLeaderId());
  61         assertEquals("getTerm", expected.getTerm(), actual.getTerm());
  62         assertEquals("getLeaderCommit", expected.getLeaderCommit(), actual.getLeaderCommit());
  63         assertEquals("getPrevLogIndex", expected.getPrevLogIndex(), actual.getPrevLogIndex());
  64         assertEquals("getPrevLogTerm", expected.getPrevLogTerm(), actual.getPrevLogTerm());
  65         assertEquals("getReplicatedToAllIndex", expected.getReplicatedToAllIndex(), actual.getReplicatedToAllIndex());
  66         assertEquals("getPayloadVersion", expected.getPayloadVersion(), actual.getPayloadVersion());
  67
  68         assertEquals("getEntries size", expected.getEntries().size(), actual.getEntries().size());
  69         Iterator<ReplicatedLogEntry> iter = expected.getEntries().iterator();
  70         for (ReplicatedLogEntry e: actual.getEntries()) {
  71             verifyReplicatedLogEntry(iter.next(), e);
  72         }
  73
  74         assertEquals("getLeaderAddress", expected.leaderAddress(), actual.leaderAddress());
  75         assertEquals("getLeaderRaftVersion", RaftVersions.CURRENT_VERSION, actual.getLeaderRaftVersion());
  76     }
  77
  78     private static void verifyReplicatedLogEntry(final ReplicatedLogEntry expected, final ReplicatedLogEntry actual) {
  79         assertEquals("getIndex", expected.getIndex(), actual.getIndex());
  80         assertEquals("getTerm", expected.getTerm(), actual.getTerm());
  81         assertEquals("getData", expected.getData().toString(), actual.getData().toString());
  82     }
  83 }