3 * Copyright (c) 2014 Cisco Systems, Inc. and others. All rights reserved.
5 * This program and the accompanying materials are made available under the
6 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
7 * and is available at http://www.eclipse.org/legal/epl-v10.html
10 package org.opendaylight.controller.cluster.datastore.node.utils.stream;
12 import java.io.ByteArrayInputStream;
13 import java.io.ByteArrayOutputStream;
14 import java.io.IOException;
15 import org.apache.commons.lang.SerializationUtils;
16 import org.junit.Assert;
17 import org.junit.Test;
18 import org.opendaylight.controller.cluster.datastore.node.NormalizedNodeToNodeCodec;
19 import org.opendaylight.controller.cluster.datastore.util.InstanceIdentifierUtils;
20 import org.opendaylight.controller.cluster.datastore.util.TestModel;
21 import org.opendaylight.controller.protobuff.messages.transaction.ShardTransactionMessages;
22 import org.opendaylight.yangtools.yang.common.QName;
23 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
24 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
25 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
26 import org.opendaylight.yangtools.yang.data.api.schema.LeafSetEntryNode;
27 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
28 import org.opendaylight.yangtools.yang.data.impl.schema.Builders;
29 import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNodes;
30 import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableLeafSetEntryNodeBuilder;
31 import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableLeafSetNodeBuilder;
32 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
34 public class NormalizedNodeStreamReaderWriterTest {
37 public void testNormalizedNodeStreaming() throws IOException {
39 ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
40 NormalizedNodeOutputStreamWriter writer = new NormalizedNodeOutputStreamWriter(byteArrayOutputStream);
42 NormalizedNode<?, ?> testContainer = createTestContainer();
43 writer.writeNormalizedNode(testContainer);
45 QName toaster = QName.create("http://netconfcentral.org/ns/toaster","2009-11-20","toaster");
46 QName darknessFactor = QName.create("http://netconfcentral.org/ns/toaster","2009-11-20","darknessFactor");
47 ContainerNode toasterNode = Builders.containerBuilder().
48 withNodeIdentifier(new NodeIdentifier(toaster)).
49 withChild(ImmutableNodes.leafNode(darknessFactor, "1000")).build();
51 ContainerNode toasterContainer = Builders.containerBuilder().
52 withNodeIdentifier(new NodeIdentifier(SchemaContext.NAME)).
53 withChild(toasterNode).build();
54 writer.writeNormalizedNode(toasterContainer);
56 NormalizedNodeInputStreamReader reader = new NormalizedNodeInputStreamReader(
57 new ByteArrayInputStream(byteArrayOutputStream.toByteArray()));
59 NormalizedNode<?,?> node = reader.readNormalizedNode();
60 Assert.assertEquals(testContainer, node);
62 node = reader.readNormalizedNode();
63 Assert.assertEquals(toasterContainer, node);
68 private NormalizedNode<?, ?> createTestContainer() {
69 byte[] bytes1 = {1,2,3};
70 LeafSetEntryNode<Object> entry1 = ImmutableLeafSetEntryNodeBuilder.create().withNodeIdentifier(
71 new YangInstanceIdentifier.NodeWithValue(TestModel.BINARY_LEAF_LIST_QNAME, bytes1)).
72 withValue(bytes1).build();
75 LeafSetEntryNode<Object> entry2 = ImmutableLeafSetEntryNodeBuilder.create().withNodeIdentifier(
76 new YangInstanceIdentifier.NodeWithValue(TestModel.BINARY_LEAF_LIST_QNAME, bytes2)).
77 withValue(bytes2).build();
79 LeafSetEntryNode<Object> entry3 = ImmutableLeafSetEntryNodeBuilder.create().withNodeIdentifier(
80 new YangInstanceIdentifier.NodeWithValue(TestModel.BINARY_LEAF_LIST_QNAME, null)).
81 withValue(null).build();
84 return TestModel.createBaseTestContainerBuilder().
85 withChild(ImmutableLeafSetNodeBuilder.create().withNodeIdentifier(
86 new YangInstanceIdentifier.NodeIdentifier(TestModel.BINARY_LEAF_LIST_QNAME)).
87 withChild(entry1).withChild(entry2).withChild(entry3).build()).
88 withChild(ImmutableNodes.leafNode(TestModel.SOME_BINARY_DATA_QNAME, new byte[]{1,2,3,4})).
89 withChild(Builders.orderedMapBuilder().
90 withNodeIdentifier(new YangInstanceIdentifier.NodeIdentifier(TestModel.ORDERED_LIST_QNAME)).
91 withChild(ImmutableNodes.mapEntry(TestModel.ORDERED_LIST_ENTRY_QNAME,
92 TestModel.ID_QNAME, 11)).build()).
97 public void testYangInstanceIdentifierStreaming() throws IOException {
98 YangInstanceIdentifier path = YangInstanceIdentifier.builder(TestModel.TEST_PATH).
99 node(TestModel.OUTER_LIST_QNAME).nodeWithKey(
100 TestModel.INNER_LIST_QNAME, TestModel.ID_QNAME, 10).build();
102 ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
103 NormalizedNodeOutputStreamWriter writer =
104 new NormalizedNodeOutputStreamWriter(byteArrayOutputStream);
105 writer.writeYangInstanceIdentifier(path);
107 NormalizedNodeInputStreamReader reader = new NormalizedNodeInputStreamReader(
108 new ByteArrayInputStream(byteArrayOutputStream.toByteArray()));
110 YangInstanceIdentifier newPath = reader.readYangInstanceIdentifier();
111 Assert.assertEquals(path, newPath);
117 public void testNormalizedNodeAndYangInstanceIdentifierStreaming() throws IOException {
119 ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
120 NormalizedNodeOutputStreamWriter writer = new NormalizedNodeOutputStreamWriter(byteArrayOutputStream);
122 NormalizedNode<?, ?> testContainer = TestModel.createBaseTestContainerBuilder().build();
123 writer.writeNormalizedNode(testContainer);
125 YangInstanceIdentifier path = YangInstanceIdentifier.builder(TestModel.TEST_PATH).
126 node(TestModel.OUTER_LIST_QNAME).nodeWithKey(
127 TestModel.INNER_LIST_QNAME, TestModel.ID_QNAME, 10).build();
129 writer.writeYangInstanceIdentifier(path);
131 NormalizedNodeInputStreamReader reader = new NormalizedNodeInputStreamReader(
132 new ByteArrayInputStream(byteArrayOutputStream.toByteArray()));
134 NormalizedNode<?,?> node = reader.readNormalizedNode();
135 Assert.assertEquals(testContainer, node);
137 YangInstanceIdentifier newPath = reader.readYangInstanceIdentifier();
138 Assert.assertEquals(path, newPath);
143 @Test(expected=InvalidNormalizedNodeStreamException.class, timeout=10000)
144 public void testInvalidNormalizedNodeStream() throws IOException {
145 byte[] protobufBytes = new NormalizedNodeToNodeCodec(null).encode(
146 TestModel.createBaseTestContainerBuilder().build()).getNormalizedNode().toByteArray();
148 NormalizedNodeInputStreamReader reader = new NormalizedNodeInputStreamReader(
149 new ByteArrayInputStream(protobufBytes));
151 reader.readNormalizedNode();
154 @Test(expected=InvalidNormalizedNodeStreamException.class, timeout=10000)
155 public void testInvalidYangInstanceIdentifierStream() throws IOException {
156 YangInstanceIdentifier path = YangInstanceIdentifier.builder(TestModel.TEST_PATH).build();
158 byte[] protobufBytes = ShardTransactionMessages.DeleteData.newBuilder().setInstanceIdentifierPathArguments(
159 InstanceIdentifierUtils.toSerializable(path)).build().toByteArray();
161 NormalizedNodeInputStreamReader reader = new NormalizedNodeInputStreamReader(
162 new ByteArrayInputStream(protobufBytes));
164 reader.readYangInstanceIdentifier();
168 public void testWithSerializable() {
169 NormalizedNode<?, ?> input = TestModel.createTestContainer();
170 SampleNormalizedNodeSerializable serializable = new SampleNormalizedNodeSerializable(input );
171 SampleNormalizedNodeSerializable clone = (SampleNormalizedNodeSerializable)SerializationUtils.clone(serializable);
173 Assert.assertEquals(input, clone.getInput());