X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?p=controller.git;a=blobdiff_plain;f=opendaylight%2Fmd-sal%2Fsal-clustering-commons%2Fsrc%2Ftest%2Fjava%2Forg%2Fopendaylight%2Fcontroller%2Fcluster%2Fdatastore%2Fnode%2Futils%2Fstream%2FNormalizedNodeStreamReaderWriterTest.java;h=951170c7a1608a06cbc3f569b193264694f8d3f4;hp=e1bf0447e657c88162a284f384efe6c684742245;hb=683170bee30d6938fbaac9ad422d664259fc1bba;hpb=68f179dcd8483dd7f681e134268a1eee29d09d55 diff --git a/opendaylight/md-sal/sal-clustering-commons/src/test/java/org/opendaylight/controller/cluster/datastore/node/utils/stream/NormalizedNodeStreamReaderWriterTest.java b/opendaylight/md-sal/sal-clustering-commons/src/test/java/org/opendaylight/controller/cluster/datastore/node/utils/stream/NormalizedNodeStreamReaderWriterTest.java index e1bf0447e6..951170c7a1 100644 --- a/opendaylight/md-sal/sal-clustering-commons/src/test/java/org/opendaylight/controller/cluster/datastore/node/utils/stream/NormalizedNodeStreamReaderWriterTest.java +++ b/opendaylight/md-sal/sal-clustering-commons/src/test/java/org/opendaylight/controller/cluster/datastore/node/utils/stream/NormalizedNodeStreamReaderWriterTest.java @@ -5,16 +5,18 @@ * terms of the Eclipse Public License v1.0 which accompanies this distribution, * and is available at http://www.eclipse.org/legal/epl-v10.html */ - package org.opendaylight.controller.cluster.datastore.node.utils.stream; import static org.junit.Assert.assertEquals; +import com.google.common.collect.ImmutableSet; import com.google.common.io.ByteStreams; import java.io.ByteArrayOutputStream; import java.io.IOException; import java.io.StringReader; import java.io.StringWriter; +import java.util.ArrayList; +import java.util.List; import java.util.Optional; import javax.xml.parsers.DocumentBuilderFactory; import javax.xml.transform.OutputKeys; @@ -29,16 +31,22 @@ import org.opendaylight.controller.cluster.datastore.util.TestModel; import org.opendaylight.yangtools.yang.common.Empty; import org.opendaylight.yangtools.yang.common.QName; import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier; +import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.AugmentationIdentifier; import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier; +import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifierWithPredicates; import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeWithValue; import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.PathArgument; -import org.opendaylight.yangtools.yang.data.api.schema.AnyXmlNode; import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode; +import org.opendaylight.yangtools.yang.data.api.schema.DOMSourceAnyxmlNode; import org.opendaylight.yangtools.yang.data.api.schema.DataContainerChild; import org.opendaylight.yangtools.yang.data.api.schema.LeafSetEntryNode; +import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode; +import org.opendaylight.yangtools.yang.data.api.schema.MapNode; import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode; import org.opendaylight.yangtools.yang.data.impl.schema.Builders; import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNodes; +import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.CollectionNodeBuilder; +import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.DataContainerNodeBuilder; import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableContainerNodeBuilder; import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableLeafSetEntryNodeBuilder; import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableLeafSetNodeBuilder; @@ -69,10 +77,12 @@ public class NormalizedNodeStreamReaderWriterTest { .withNodeIdentifier(new NodeIdentifier(SchemaContext.NAME)).withChild(toasterNode).build(); nnout.writeNormalizedNode(toasterContainer); - NormalizedNodeDataInput nnin = NormalizedNodeInputOutput.newDataInput(ByteStreams.newDataInput( - bos.toByteArray())); + final byte[] bytes = bos.toByteArray(); + assertEquals(1049619, bytes.length); + + NormalizedNodeDataInput nnin = NormalizedNodeInputOutput.newDataInput(ByteStreams.newDataInput(bytes)); - NormalizedNode node = nnin.readNormalizedNode(); + NormalizedNode node = nnin.readNormalizedNode(); Assert.assertEquals(testContainer, node); node = nnin.readNormalizedNode(); @@ -111,8 +121,10 @@ public class NormalizedNodeStreamReaderWriterTest { nnout.writeYangInstanceIdentifier(path); - NormalizedNodeDataInput nnin = NormalizedNodeInputOutput.newDataInput(ByteStreams.newDataInput( - bos.toByteArray())); + final byte[] bytes = bos.toByteArray(); + assertEquals(139, bytes.length); + + NormalizedNodeDataInput nnin = NormalizedNodeInputOutput.newDataInput(ByteStreams.newDataInput(bytes)); YangInstanceIdentifier newPath = nnin.readYangInstanceIdentifier(); Assert.assertEquals(path, newPath); @@ -121,9 +133,9 @@ public class NormalizedNodeStreamReaderWriterTest { @Test public void testNormalizedNodeAndYangInstanceIdentifierStreaming() throws IOException { - ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream(); + ByteArrayOutputStream bos = new ByteArrayOutputStream(); NormalizedNodeDataOutput writer = NormalizedNodeInputOutput.newDataOutput( - ByteStreams.newDataOutput(byteArrayOutputStream)); + ByteStreams.newDataOutput(bos)); NormalizedNode testContainer = TestModel.createBaseTestContainerBuilder().build(); writer.writeNormalizedNode(testContainer); @@ -134,8 +146,10 @@ public class NormalizedNodeStreamReaderWriterTest { writer.writeYangInstanceIdentifier(path); - NormalizedNodeDataInput reader = NormalizedNodeInputOutput.newDataInput( - ByteStreams.newDataInput(byteArrayOutputStream.toByteArray())); + final byte[] bytes = bos.toByteArray(); + assertEquals(826, bytes.length); + + NormalizedNodeDataInput reader = NormalizedNodeInputOutput.newDataInput(ByteStreams.newDataInput(bytes)); NormalizedNode node = reader.readNormalizedNode(); Assert.assertEquals(testContainer, node); @@ -148,7 +162,7 @@ public class NormalizedNodeStreamReaderWriterTest { @Test(expected = InvalidNormalizedNodeStreamException.class, timeout = 10000) public void testInvalidNormalizedNodeStream() throws IOException { - byte[] invalidBytes = {1,2,3}; + byte[] invalidBytes = {1, 2, 3}; NormalizedNodeDataInput reader = NormalizedNodeInputOutput.newDataInput( ByteStreams.newDataInput(invalidBytes)); @@ -195,8 +209,10 @@ public class NormalizedNodeStreamReaderWriterTest { nnout.writeNormalizedNode(anyXmlContainer); - NormalizedNodeDataInput nnin = NormalizedNodeInputOutput.newDataInput(ByteStreams.newDataInput( - bos.toByteArray())); + final byte[] bytes = bos.toByteArray(); + assertEquals(229, bytes.length); + + NormalizedNodeDataInput nnin = NormalizedNodeInputOutput.newDataInput(ByteStreams.newDataInput(bytes)); ContainerNode deserialized = (ContainerNode)nnin.readNormalizedNode(); @@ -207,24 +223,105 @@ public class NormalizedNodeStreamReaderWriterTest { StreamResult xmlOutput = new StreamResult(new StringWriter()); Transformer transformer = TransformerFactory.newInstance().newTransformer(); transformer.setOutputProperty(OutputKeys.OMIT_XML_DECLARATION, "yes"); - transformer.transform(((AnyXmlNode)child.get()).getValue(), xmlOutput); + transformer.transform(((DOMSourceAnyxmlNode)child.get()).getValue(), xmlOutput); assertEquals("XML", xml, xmlOutput.getWriter().toString()); - assertEquals("http://www.w3.org/TR/html4/", ((AnyXmlNode)child.get()).getValue().getNode().getNamespaceURI()); + assertEquals("http://www.w3.org/TR/html4/", + ((DOMSourceAnyxmlNode)child.get()).getValue().getNode().getNamespaceURI()); } @Test - public void testSchemaPathSerialization() throws Exception { + public void testSchemaPathSerialization() throws IOException { final SchemaPath expected = SchemaPath.create(true, TestModel.ANY_XML_QNAME); ByteArrayOutputStream bos = new ByteArrayOutputStream(); - NormalizedNodeDataOutput nnout = NormalizedNodeInputOutput.newDataOutput(ByteStreams.newDataOutput(bos)); - nnout.writeSchemaPath(expected); + try (NormalizedNodeDataOutput nnout = NormalizedNodeInputOutput.newDataOutput(ByteStreams.newDataOutput(bos))) { + nnout.writeSchemaPath(expected); + } + + final byte[] bytes = bos.toByteArray(); + assertEquals(99, bytes.length); + + NormalizedNodeDataInput nnin = NormalizedNodeInputOutput.newDataInput(ByteStreams.newDataInput(bytes)); + assertEquals(expected, nnin.readSchemaPath()); + } + + @Test + public void testWritePathArgument() throws IOException { + final NodeIdentifier expected = new NodeIdentifier(TestModel.BOOLEAN_LEAF_QNAME); + + ByteArrayOutputStream bos = new ByteArrayOutputStream(); + + try (NormalizedNodeDataOutput nnout = NormalizedNodeInputOutput.newDataOutput(ByteStreams.newDataOutput(bos))) { + nnout.writePathArgument(expected); + } + + final byte[] bytes = bos.toByteArray(); + assertEquals(103, bytes.length); - NormalizedNodeDataInput nnin = NormalizedNodeInputOutput.newDataInput(ByteStreams.newDataInput( - bos.toByteArray())); - SchemaPath actual = nnin.readSchemaPath(); - assertEquals(expected, actual); + NormalizedNodeDataInput nnin = NormalizedNodeInputOutput.newDataInput(ByteStreams.newDataInput(bytes)); + assertEquals(expected, nnin.readPathArgument()); + } + + /* + * This tests the encoding of a MapNode with a lot of entries, each of them having the key leaf (a string) + * and an integer. + */ + @Test + public void testHugeEntries() throws IOException { + final CollectionNodeBuilder mapBuilder = Builders.mapBuilder() + .withNodeIdentifier(new NodeIdentifier(TestModel.TEST_QNAME)); + final DataContainerNodeBuilder entryBuilder = + Builders.mapEntryBuilder().withChild(ImmutableNodes.leafNode(TestModel.DESC_QNAME, (byte) 42)); + + for (int i = 0; i < 100_000; ++i) { + final String key = "xyzzy" + i; + mapBuilder.addChild(entryBuilder + .withNodeIdentifier(NodeIdentifierWithPredicates.of(TestModel.TEST_QNAME, + TestModel.CHILD_NAME_QNAME, key)) + .withChild(ImmutableNodes.leafNode(TestModel.CHILD_NAME_QNAME, key)) + .build()); + } + + final MapNode expected = mapBuilder.build(); + final ByteArrayOutputStream bos = new ByteArrayOutputStream(); + + try (NormalizedNodeDataOutput nnout = NormalizedNodeInputOutput.newDataOutput(ByteStreams.newDataOutput(bos))) { + nnout.writeNormalizedNode(expected); + } + + final byte[] bytes = bos.toByteArray(); + assertEquals(2_289_103, bytes.length); + + NormalizedNodeDataInput nnin = NormalizedNodeInputOutput.newDataInput(ByteStreams.newDataInput(bytes)); + assertEquals(expected, nnin.readNormalizedNode()); + } + + @Test + public void testAugmentationIdentifier() throws IOException { + final List qnames = new ArrayList<>(); + for (int i = 0; i < 257; ++i) { + qnames.add(QName.create(TestModel.TEST_QNAME, "a" + i)); + } + + for (int i = 0; i < qnames.size(); ++i) { + final AugmentationIdentifier expected = AugmentationIdentifier.create( + ImmutableSet.copyOf(qnames.subList(0, i))); + + ByteArrayOutputStream bos = new ByteArrayOutputStream(); + + try (NormalizedNodeDataOutput nnout = + NormalizedNodeInputOutput.newDataOutput(ByteStreams.newDataOutput(bos), + NormalizedNodeStreamVersion.SODIUM_SR1)) { + nnout.writePathArgument(expected); + } + + final byte[] bytes = bos.toByteArray(); + + NormalizedNodeDataInput nnin = NormalizedNodeInputOutput.newDataInput(ByteStreams.newDataInput(bytes)); + PathArgument arg = nnin.readPathArgument(); + assertEquals(expected, arg); + } } private static String largeString(final int pow) {