--- /dev/null
+/*
+ * Copyright (c) 2015 Brocade Communications Systems, Inc. and others. All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
+package org.opendaylight.controller.cluster.raft.messages;
+
+import static org.junit.Assert.assertArrayEquals;
+import static org.junit.Assert.assertEquals;
+import com.google.common.base.Optional;
+import java.io.Serializable;
+import org.apache.commons.lang.SerializationUtils;
+import org.junit.Test;
+import org.opendaylight.controller.cluster.raft.RaftVersions;
+import org.opendaylight.controller.protobuff.messages.cluster.raft.InstallSnapshotMessages;
+
+/**
+ * Unit tests for InstallSnapshot.
+ *
+ * @author Thomas Pantelis
+ */
+public class InstallSnapshotTest {
+
+ @Test
+ public void testSerialization() {
+ byte[] data = new byte[1000];
+ int j = 0;
+ for(int i = 0; i < data.length; i++) {
+ data[i] = (byte)j;
+ if(++j >= 255) {
+ j = 0;
+ }
+ }
+
+ InstallSnapshot expected = new InstallSnapshot(3L, "leaderId", 11L, 2L, data, 5, 6, Optional.<Integer>of(54321));
+
+ Object serialized = expected.toSerializable(RaftVersions.CURRENT_VERSION);
+ assertEquals("Serialized type", InstallSnapshot.class, serialized.getClass());
+
+ InstallSnapshot actual = InstallSnapshot.fromSerializable(SerializationUtils.clone((Serializable) serialized));
+ verifyInstallSnapshot(expected, actual);
+
+ expected = new InstallSnapshot(3L, "leaderId", 11L, 2L, data, 5, 6);
+ actual = InstallSnapshot.fromSerializable(SerializationUtils.clone(
+ (Serializable) expected.toSerializable(RaftVersions.CURRENT_VERSION)));
+ verifyInstallSnapshot(expected, actual);
+ }
+
+ @Test
+ public void testSerializationWithPreBoronVersion() {
+ byte[] data = {0,1,2,3,4,5,7,8,9};
+ InstallSnapshot expected = new InstallSnapshot(3L, "leaderId", 11L, 2L, data, 5, 6, Optional.<Integer>of(54321));
+
+ Object serialized = expected.toSerializable(RaftVersions.LITHIUM_VERSION);
+ assertEquals("Serialized type", InstallSnapshot.SERIALIZABLE_CLASS, serialized.getClass());
+
+ InstallSnapshot actual = InstallSnapshot.fromSerializable(SerializationUtils.clone((Serializable) serialized));
+ verifyInstallSnapshot(expected, actual);
+ }
+
+ @Test
+ public void testIsSerializedType() {
+ assertEquals("isSerializedType", true, InstallSnapshot.isSerializedType(
+ InstallSnapshotMessages.InstallSnapshot.newBuilder().build()));
+ assertEquals("isSerializedType", true, InstallSnapshot.isSerializedType(new InstallSnapshot()));
+ assertEquals("isSerializedType", false, InstallSnapshot.isSerializedType(new Object()));
+ }
+
+ private void verifyInstallSnapshot(InstallSnapshot expected, InstallSnapshot actual) {
+ assertEquals("getTerm", expected.getTerm(), actual.getTerm());
+ assertEquals("getChunkIndex", expected.getChunkIndex(), actual.getChunkIndex());
+ assertEquals("getTotalChunks", expected.getTotalChunks(), actual.getTotalChunks());
+ assertEquals("getLastIncludedTerm", expected.getLastIncludedTerm(), actual.getLastIncludedTerm());
+ assertEquals("getLastIncludedIndex", expected.getLastIncludedIndex(), actual.getLastIncludedIndex());
+ assertEquals("getLeaderId", expected.getLeaderId(), actual.getLeaderId());
+ assertEquals("getChunkIndex", expected.getChunkIndex(), actual.getChunkIndex());
+ assertArrayEquals("getData", expected.getData(), actual.getData());
+ assertEquals("getLastChunkHashCode present", expected.getLastChunkHashCode().isPresent(),
+ actual.getLastChunkHashCode().isPresent());
+ if(expected.getLastChunkHashCode().isPresent()) {
+ assertEquals("getLastChunkHashCode", expected.getLastChunkHashCode().get(),
+ actual.getLastChunkHashCode().get());
+ }
+ }
+}