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

   1 package org.opendaylight.controller.cluster.datastore;
   2
   3 import java.io.ByteArrayInputStream;
   4 import java.io.ByteArrayOutputStream;
   5 import java.io.IOException;
   6 import java.util.ArrayList;
   7 import java.util.List;
   8 import org.junit.Assert;
   9 import org.junit.Test;
  10 import org.opendaylight.controller.cluster.datastore.modification.MutableCompositeModification;
  11 import org.opendaylight.controller.cluster.datastore.modification.WriteModification;
  12 import org.opendaylight.controller.cluster.raft.RaftVersions;
  13 import org.opendaylight.controller.cluster.raft.ReplicatedLogEntry;
  14 import org.opendaylight.controller.cluster.raft.messages.AppendEntries;
  15 import org.opendaylight.controller.cluster.raft.protobuff.client.messages.CompositeModificationPayload;
  16 import org.opendaylight.controller.cluster.raft.protobuff.client.messages.Payload;
  17 import org.opendaylight.controller.md.cluster.datastore.model.TestModel;
  18 import org.opendaylight.controller.protobuff.messages.cluster.raft.AppendEntriesMessages;
  19 import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNodes;
  20
  21 public class CompositeModificationPayloadTest {
  22
  23     @Test
  24     public void testBasic() throws IOException {
  25
  26         List<ReplicatedLogEntry> entries = new ArrayList<>();
  27
  28         entries.add(0, new ReplicatedLogEntry() {
  29             @Override public Payload getData() {
  30                 WriteModification writeModification =
  31                     new WriteModification(TestModel.TEST_PATH, ImmutableNodes
  32                         .containerNode(TestModel.TEST_QNAME));
  33
  34                 MutableCompositeModification compositeModification =
  35                     new MutableCompositeModification();
  36
  37                 compositeModification.addModification(writeModification);
  38
  39                 return new CompositeModificationPayload(compositeModification.toSerializable());
  40             }
  41
  42             @Override public long getTerm() {
  43                 return 1;
  44             }
  45
  46             @Override public long getIndex() {
  47                 return 1;
  48             }
  49
  50             @Override
  51             public int size() {
  52                 return getData().size();
  53             }
  54         });
  55
  56         AppendEntries appendEntries =
  57             new AppendEntries(1, "member-1", 0, 100, entries, 1);
  58
  59         AppendEntriesMessages.AppendEntries o = (AppendEntriesMessages.AppendEntries)
  60                 appendEntries.toSerializable(RaftVersions.HELIUM_VERSION);
  61
  62         ByteArrayOutputStream bos = new ByteArrayOutputStream();
  63         o.writeDelimitedTo(bos);
  64
  65         AppendEntriesMessages.AppendEntries appendEntries2 =
  66             AppendEntriesMessages.AppendEntries
  67                 .parseDelimitedFrom(new ByteArrayInputStream(bos.toByteArray()));
  68
  69         AppendEntries appendEntries1 = AppendEntries.fromSerializable(appendEntries2);
  70
  71         Payload data = appendEntries1.getEntries().get(0).getData();
  72
  73
  74         Assert.assertTrue(((CompositeModificationPayload) data).getModification().toString().contains(TestModel.TEST_QNAME.getNamespace().toString()));
  75     }
  76 }