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 import static org.junit.Assert.assertFalse;
  12
  13 import java.util.Arrays;
  14 import java.util.Iterator;
  15 import org.apache.commons.lang.SerializationUtils;
  16 import org.junit.Test;
  17 import org.opendaylight.controller.cluster.raft.MockRaftActorContext.MockPayload;
  18 import org.opendaylight.controller.cluster.raft.RaftVersions;
  19 import org.opendaylight.controller.cluster.raft.ReplicatedLogEntry;
  20 import org.opendaylight.controller.cluster.raft.persisted.SimpleReplicatedLogEntry;
  21
  22 /**
  23  * Unit tests for AppendEntries.
  24  *
  25  * @author Thomas Pantelis
  26  */
  27 public class AppendEntriesTest {
  28
  29     @Test
  30     public void testSerialization() {
  31         ReplicatedLogEntry entry1 = new SimpleReplicatedLogEntry(1, 2, new MockPayload("payload1"));
  32
  33         ReplicatedLogEntry entry2 = new SimpleReplicatedLogEntry(3, 4, new MockPayload("payload2"));
  34
  35         short payloadVersion = 5;
  36
  37         // Without leader address
  38
  39         AppendEntries expected = new AppendEntries(5L, "node1", 7L, 8L, Arrays.asList(entry1, entry2), 10L,
  40                 -1, payloadVersion, RaftVersions.CURRENT_VERSION, null);
  41
  42         AppendEntries cloned = (AppendEntries) SerializationUtils.clone(expected);
  43
  44         verifyAppendEntries(expected, cloned, RaftVersions.CURRENT_VERSION);
  45
  46         // With leader address
  47
  48         expected = new AppendEntries(5L, "node1", 7L, 8L, Arrays.asList(entry1, entry2), 10L,
  49                 -1, payloadVersion, RaftVersions.CURRENT_VERSION, "leader address");
  50
  51         cloned = (AppendEntries) SerializationUtils.clone(expected);
  52
  53         verifyAppendEntries(expected, cloned, RaftVersions.CURRENT_VERSION);
  54     }
  55
  56     @Test
  57     @Deprecated
  58     public void testPreFluorineSerialization() {
  59         ReplicatedLogEntry entry1 = new SimpleReplicatedLogEntry(1, 2, new MockPayload("payload1"));
  60
  61         ReplicatedLogEntry entry2 = new SimpleReplicatedLogEntry(3, 4, new MockPayload("payload2"));
  62
  63         short payloadVersion = 5;
  64
  65         AppendEntries expected = new AppendEntries(5L, "node1", 7L, 8L, Arrays.asList(entry1, entry2), 10L,
  66                 -1, payloadVersion, RaftVersions.BORON_VERSION, "leader address");
  67
  68         AppendEntries cloned = (AppendEntries) SerializationUtils.clone(expected);
  69
  70         verifyAppendEntries(expected, cloned, RaftVersions.BORON_VERSION);
  71     }
  72
  73     private static void verifyAppendEntries(AppendEntries expected, AppendEntries actual, short recipientRaftVersion) {
  74         assertEquals("getLeaderId", expected.getLeaderId(), actual.getLeaderId());
  75         assertEquals("getTerm", expected.getTerm(), actual.getTerm());
  76         assertEquals("getLeaderCommit", expected.getLeaderCommit(), actual.getLeaderCommit());
  77         assertEquals("getPrevLogIndex", expected.getPrevLogIndex(), actual.getPrevLogIndex());
  78         assertEquals("getPrevLogTerm", expected.getPrevLogTerm(), actual.getPrevLogTerm());
  79         assertEquals("getReplicatedToAllIndex", expected.getReplicatedToAllIndex(), actual.getReplicatedToAllIndex());
  80         assertEquals("getPayloadVersion", expected.getPayloadVersion(), actual.getPayloadVersion());
  81
  82         assertEquals("getEntries size", expected.getEntries().size(), actual.getEntries().size());
  83         Iterator<ReplicatedLogEntry> iter = expected.getEntries().iterator();
  84         for (ReplicatedLogEntry e: actual.getEntries()) {
  85             verifyReplicatedLogEntry(iter.next(), e);
  86         }
  87
  88         if (recipientRaftVersion >= RaftVersions.FLUORINE_VERSION) {
  89             assertEquals("getLeaderAddress", expected.getLeaderAddress(), actual.getLeaderAddress());
  90             assertEquals("getLeaderRaftVersion", RaftVersions.CURRENT_VERSION, actual.getLeaderRaftVersion());
  91         } else {
  92             assertFalse(actual.getLeaderAddress().isPresent());
  93             assertEquals("getLeaderRaftVersion", RaftVersions.BORON_VERSION, actual.getLeaderRaftVersion());
  94         }
  95     }
  96
  97     private static void verifyReplicatedLogEntry(ReplicatedLogEntry expected, ReplicatedLogEntry actual) {
  98         assertEquals("getIndex", expected.getIndex(), actual.getIndex());
  99         assertEquals("getTerm", expected.getTerm(), actual.getTerm());
 100         assertEquals("getData", expected.getData().toString(), actual.getData().toString());
 101     }
 102 }