opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/messages/MergeDataTest.java

   1 package org.opendaylight.controller.cluster.datastore.messages;
   2
   3 import static org.junit.Assert.assertEquals;
   4 import java.io.Serializable;
   5 import org.apache.commons.lang.SerializationUtils;
   6 import org.junit.Test;
   7 import org.opendaylight.controller.cluster.datastore.DataStoreVersions;
   8 import org.opendaylight.controller.md.cluster.datastore.model.TestModel;
   9 import org.opendaylight.controller.protobuff.messages.common.NormalizedNodeMessages.InstanceIdentifier;
  10 import org.opendaylight.controller.protobuff.messages.common.NormalizedNodeMessages.Node;
  11 import org.opendaylight.controller.protobuff.messages.transaction.ShardTransactionMessages;
  12 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  13 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  14 import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNodes;
  15 import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableContainerNodeBuilder;
  16
  17 public class MergeDataTest {
  18
  19     @Test
  20     public void testSerialization() {
  21         YangInstanceIdentifier path = TestModel.TEST_PATH;
  22         NormalizedNode<?, ?> data = ImmutableContainerNodeBuilder.create().withNodeIdentifier(
  23                 new YangInstanceIdentifier.NodeIdentifier(TestModel.TEST_QNAME)).
  24                 withChild(ImmutableNodes.leafNode(TestModel.DESC_QNAME, "foo")).build();
  25
  26         MergeData expected = new MergeData(path, data);
  27
  28         Object serialized = expected.toSerializable(DataStoreVersions.CURRENT_VERSION);
  29         assertEquals("Serialized type", MergeData.class, serialized.getClass());
  30         assertEquals("Version", DataStoreVersions.CURRENT_VERSION, ((MergeData)serialized).getVersion());
  31
  32         Object clone = SerializationUtils.clone((Serializable) serialized);
  33         assertEquals("Version", DataStoreVersions.CURRENT_VERSION, ((MergeData)clone).getVersion());
  34         MergeData actual = MergeData.fromSerializable(clone);
  35         assertEquals("getPath", expected.getPath(), actual.getPath());
  36         assertEquals("getData", expected.getData(), actual.getData());
  37     }
  38
  39     @Test
  40     public void testIsSerializedType() {
  41         assertEquals("isSerializedType", true, MergeData.isSerializedType(
  42                 ShardTransactionMessages.MergeData.newBuilder()
  43                     .setInstanceIdentifierPathArguments(InstanceIdentifier.getDefaultInstance())
  44                     .setNormalizedNode(Node.getDefaultInstance()).build()));
  45         assertEquals("isSerializedType", true,
  46                 MergeData.isSerializedType(new MergeData()));
  47         assertEquals("isSerializedType", false, MergeData.isSerializedType(new Object()));
  48     }
  49
  50     /**
  51      * Tests backwards compatible serialization/deserialization of a MergeData message with the
  52      * base and R1 Helium versions, which used the protobuff MergeData message.
  53      */
  54     @Test
  55     public void testSerializationWithHeliumR1Version() throws Exception {
  56         YangInstanceIdentifier path = TestModel.TEST_PATH;
  57         NormalizedNode<?, ?> data = ImmutableContainerNodeBuilder.create().withNodeIdentifier(
  58                 new YangInstanceIdentifier.NodeIdentifier(TestModel.TEST_QNAME)).
  59                 withChild(ImmutableNodes.leafNode(TestModel.DESC_QNAME, "foo")).build();
  60
  61         MergeData expected = new MergeData(path, data);
  62
  63         Object serialized = expected.toSerializable(DataStoreVersions.HELIUM_1_VERSION);
  64         assertEquals("Serialized type", ShardTransactionMessages.MergeData.class, serialized.getClass());
  65
  66         MergeData actual = MergeData.fromSerializable(SerializationUtils.clone((Serializable) serialized));
  67         assertEquals("getPath", expected.getPath(), actual.getPath());
  68         assertEquals("getData", expected.getData(), actual.getData());
  69     }
  70 }