2 * Copyright (c) 2015 Brocade Communications Systems, Inc. and others. All rights reserved.
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
8 package org.opendaylight.controller.cluster.raft.messages;
10 import static org.junit.Assert.assertEquals;
11 import static org.junit.Assert.assertFalse;
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;
23 * Unit tests for AppendEntries.
25 * @author Thomas Pantelis
27 public class AppendEntriesTest {
30 public void testSerialization() {
31 ReplicatedLogEntry entry1 = new SimpleReplicatedLogEntry(1, 2, new MockPayload("payload1"));
33 ReplicatedLogEntry entry2 = new SimpleReplicatedLogEntry(3, 4, new MockPayload("payload2"));
35 short payloadVersion = 5;
37 // Without leader address
39 AppendEntries expected = new AppendEntries(5L, "node1", 7L, 8L, Arrays.asList(entry1, entry2), 10L,
40 -1, payloadVersion, RaftVersions.CURRENT_VERSION, null);
42 AppendEntries cloned = (AppendEntries) SerializationUtils.clone(expected);
44 verifyAppendEntries(expected, cloned, RaftVersions.CURRENT_VERSION);
46 // With leader address
48 expected = new AppendEntries(5L, "node1", 7L, 8L, Arrays.asList(entry1, entry2), 10L,
49 -1, payloadVersion, RaftVersions.CURRENT_VERSION, "leader address");
51 cloned = (AppendEntries) SerializationUtils.clone(expected);
53 verifyAppendEntries(expected, cloned, RaftVersions.CURRENT_VERSION);
58 public void testPreFluorineSerialization() {
59 ReplicatedLogEntry entry1 = new SimpleReplicatedLogEntry(1, 2, new MockPayload("payload1"));
61 ReplicatedLogEntry entry2 = new SimpleReplicatedLogEntry(3, 4, new MockPayload("payload2"));
63 short payloadVersion = 5;
65 AppendEntries expected = new AppendEntries(5L, "node1", 7L, 8L, Arrays.asList(entry1, entry2), 10L,
66 -1, payloadVersion, RaftVersions.BORON_VERSION, "leader address");
68 AppendEntries cloned = (AppendEntries) SerializationUtils.clone(expected);
70 verifyAppendEntries(expected, cloned, RaftVersions.BORON_VERSION);
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());
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);
88 if (recipientRaftVersion >= RaftVersions.FLUORINE_VERSION) {
89 assertEquals("getLeaderAddress", expected.getLeaderAddress(), actual.getLeaderAddress());
90 assertEquals("getLeaderRaftVersion", RaftVersions.CURRENT_VERSION, actual.getLeaderRaftVersion());
92 assertFalse(actual.getLeaderAddress().isPresent());
93 assertEquals("getLeaderRaftVersion", RaftVersions.BORON_VERSION, actual.getLeaderRaftVersion());
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());