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.File;
   4 import java.io.FileInputStream;
   5 import java.io.FileOutputStream;
   6 import java.io.IOException;
   7 import java.util.ArrayList;
   8 import java.util.List;
   9 import org.junit.After;
  10 import org.junit.Assert;
  11 import org.junit.Test;
  12 import org.opendaylight.controller.cluster.datastore.modification.MutableCompositeModification;
  13 import org.opendaylight.controller.cluster.datastore.modification.WriteModification;
  14 import org.opendaylight.controller.cluster.raft.ReplicatedLogEntry;
  15 import org.opendaylight.controller.cluster.raft.messages.AppendEntries;
  16 import org.opendaylight.controller.cluster.raft.protobuff.client.messages.CompositeModificationPayload;
  17 import org.opendaylight.controller.cluster.raft.protobuff.client.messages.Payload;
  18 import org.opendaylight.controller.md.cluster.datastore.model.TestModel;
  19 import org.opendaylight.controller.protobuff.messages.cluster.raft.AppendEntriesMessages;
  20 import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNodes;
  21
  22 public class CompositeModificationPayloadTest {
  23
  24
  25     private static final String SERIALIZE_OUT = "serialize.out";
  26
  27     @After
  28     public void shutDown(){
  29         File f = new File(SERIALIZE_OUT);
  30         if(f.exists()){
  31             f.delete();
  32         }
  33     }
  34
  35     @Test
  36     public void testBasic() throws IOException {
  37
  38         List<ReplicatedLogEntry> entries = new ArrayList<>();
  39
  40         entries.add(0, new ReplicatedLogEntry() {
  41             @Override public Payload getData() {
  42                 WriteModification writeModification =
  43                     new WriteModification(TestModel.TEST_PATH, ImmutableNodes
  44                         .containerNode(TestModel.TEST_QNAME),
  45                         TestModel.createTestContext());
  46
  47                 MutableCompositeModification compositeModification =
  48                     new MutableCompositeModification();
  49
  50                 compositeModification.addModification(writeModification);
  51
  52                 return new CompositeModificationPayload(compositeModification.toSerializable());
  53             }
  54
  55             @Override public long getTerm() {
  56                 return 1;
  57             }
  58
  59             @Override public long getIndex() {
  60                 return 1;
  61             }
  62
  63             @Override
  64             public int size() {
  65                 return getData().size();
  66             }
  67         });
  68
  69         AppendEntries appendEntries =
  70             new AppendEntries(1, "member-1", 0, 100, entries, 1);
  71
  72         AppendEntriesMessages.AppendEntries o = (AppendEntriesMessages.AppendEntries) appendEntries.toSerializable();
  73
  74         o.writeDelimitedTo(new FileOutputStream(SERIALIZE_OUT));
  75
  76         AppendEntriesMessages.AppendEntries appendEntries2 =
  77             AppendEntriesMessages.AppendEntries
  78                 .parseDelimitedFrom(new FileInputStream(SERIALIZE_OUT));
  79
  80         AppendEntries appendEntries1 = AppendEntries.fromSerializable(appendEntries2);
  81
  82         Payload data = appendEntries1.getEntries().get(0).getData();
  83
  84
  85         Assert.assertTrue(((CompositeModificationPayload) data).getModification().toString().contains(TestModel.TEST_QNAME.getNamespace().toString()));
  86
  87     }
  88
  89 }