1 package org.opendaylight.controller.cluster.datastore;
3 import java.io.ByteArrayInputStream;
4 import java.io.ByteArrayOutputStream;
5 import java.io.IOException;
6 import java.util.ArrayList;
8 import org.junit.Assert;
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;
22 public class CompositeModificationPayloadTest {
25 public void testBasic() throws IOException {
27 List<ReplicatedLogEntry> entries = new ArrayList<>();
29 entries.add(0, new ReplicatedLogEntry() {
30 @Override public Payload getData() {
31 WriteModification writeModification =
32 new WriteModification(TestModel.TEST_PATH, ImmutableNodes
33 .containerNode(TestModel.TEST_QNAME));
35 MutableCompositeModification compositeModification =
36 new MutableCompositeModification();
38 compositeModification.addModification(writeModification);
40 return new CompositeModificationPayload(compositeModification.toSerializable());
43 @Override public long getTerm() {
47 @Override public long getIndex() {
53 return getData().size();
57 AppendEntries appendEntries =
58 new AppendEntries(1, "member-1", 0, 100, entries, 1, -1, (short)0);
60 AppendEntriesMessages.AppendEntries o = (AppendEntriesMessages.AppendEntries)
61 appendEntries.toSerializable(RaftVersions.HELIUM_VERSION);
63 ByteArrayOutputStream bos = new ByteArrayOutputStream();
64 o.writeDelimitedTo(bos);
66 AppendEntriesMessages.AppendEntries appendEntries2 =
67 AppendEntriesMessages.AppendEntries
68 .parseDelimitedFrom(new ByteArrayInputStream(bos.toByteArray()));
70 AppendEntries appendEntries1 = AppendEntries.fromSerializable(appendEntries2);
72 Payload data = appendEntries1.getEntries().get(0).getData();
75 Assert.assertTrue(((CompositeModificationPayload) data).getModification().toString().contains(TestModel.TEST_QNAME.getNamespace().toString()));