yang/yang-data-codec-xml/src/test/java/org/opendaylight/yangtools/yang/data/codec/xml/YangModeledAnyXMLDeserializationTest.java

   1 /*
   2  * Copyright (c) 2016 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.yangtools.yang.data.codec.xml;
   9
  10 import static org.junit.Assert.assertEquals;
  11 import static org.junit.Assert.assertTrue;
  12
  13 import com.google.common.collect.ImmutableSet;
  14 import java.io.InputStream;
  15 import java.net.URI;
  16 import java.util.Collection;
  17 import java.util.Optional;
  18 import javax.xml.stream.XMLStreamReader;
  19 import org.junit.Before;
  20 import org.junit.Test;
  21 import org.opendaylight.yangtools.odlext.model.api.YangModeledAnyxmlSchemaNode;
  22 import org.opendaylight.yangtools.util.xml.UntrustedXML;
  23 import org.opendaylight.yangtools.yang.common.QName;
  24 import org.opendaylight.yangtools.yang.common.QNameModule;
  25 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
  26 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.PathArgument;
  27 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
  28 import org.opendaylight.yangtools.yang.data.api.schema.DataContainerChild;
  29 import org.opendaylight.yangtools.yang.data.api.schema.LeafNode;
  30 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  31 import org.opendaylight.yangtools.yang.data.api.schema.YangModeledAnyXmlNode;
  32 import org.opendaylight.yangtools.yang.data.api.schema.stream.NormalizedNodeStreamWriter;
  33 import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNormalizedNodeStreamWriter;
  34 import org.opendaylight.yangtools.yang.data.impl.schema.NormalizedNodeResult;
  35 import org.opendaylight.yangtools.yang.model.api.ContainerSchemaNode;
  36 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
  37 import org.opendaylight.yangtools.yang.model.api.Module;
  38 import org.opendaylight.yangtools.yang.model.api.MustDefinition;
  39 import org.opendaylight.yangtools.yang.model.api.RevisionAwareXPath;
  40 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
  41 import org.opendaylight.yangtools.yang.model.api.SchemaPath;
  42 import org.opendaylight.yangtools.yang.model.api.Status;
  43 import org.opendaylight.yangtools.yang.test.util.YangParserTestUtils;
  44
  45 public class YangModeledAnyXMLDeserializationTest {
  46
  47     private QNameModule fooModuleQName;
  48     private QNameModule barModuleQName;
  49     private QName myContainer1;
  50     private QName myContainer2;
  51     private QName innerContainer;
  52     private QName myLeaf3;
  53     private QName myLeaf2;
  54     private QName myLeaf1;
  55     private QName myAnyXMLDataBar;
  56     private QName myAnyXMLDataFoo;
  57     private SchemaContext schemaContext;
  58
  59     @Before
  60     public void setUp() {
  61         barModuleQName = QNameModule.create(URI.create("bar"));
  62         myContainer1 = QName.create(barModuleQName, "my-container-1");
  63         myLeaf1 = QName.create(barModuleQName, "my-leaf-1");
  64         myAnyXMLDataBar = QName.create(barModuleQName, "my-anyxml-data");
  65
  66         fooModuleQName = QNameModule.create(URI.create("foo"));
  67         myContainer2 = QName.create(fooModuleQName, "my-container-2");
  68         innerContainer = QName.create(fooModuleQName, "inner-container");
  69         myLeaf3 = QName.create(fooModuleQName, "my-leaf-3");
  70         myLeaf2 = QName.create(fooModuleQName, "my-leaf-2");
  71         myAnyXMLDataFoo = QName.create(fooModuleQName, "my-anyxml-data");
  72
  73         schemaContext = YangParserTestUtils.parseYangResourceDirectory("/anyxml-support/yang");
  74     }
  75
  76     @Test
  77     public void testRawAnyXMLFromBar() throws Exception {
  78         DataSchemaNode barContainer = schemaContext.findDataChildByName(QName.create(barModuleQName, "bar")).get();
  79         assertTrue(barContainer instanceof ContainerSchemaNode);
  80         final YangModeledAnyxmlSchemaNode yangModeledAnyXML = new YangModeledAnyXMLSchemaNodeImplTest(myAnyXMLDataBar,
  81                 (ContainerSchemaNode) barContainer);
  82
  83         final InputStream resourceAsStream = YangModeledAnyXMLDeserializationTest.class.getResourceAsStream(
  84                 "/anyxml-support/xml/bar.xml");
  85
  86         final XMLStreamReader reader = UntrustedXML.createXMLStreamReader(resourceAsStream);
  87         final NormalizedNodeResult result = new NormalizedNodeResult();
  88
  89         final NormalizedNodeStreamWriter streamWriter = ImmutableNormalizedNodeStreamWriter.from(result);
  90
  91         final XmlParserStream xmlParser = XmlParserStream.create(streamWriter, schemaContext, yangModeledAnyXML);
  92         xmlParser.parse(reader);
  93
  94         final NormalizedNode<?, ?> output = result.getResult();
  95         assertTrue(output instanceof YangModeledAnyXmlNode);
  96         final YangModeledAnyXmlNode yangModeledAnyXmlNode = (YangModeledAnyXmlNode) output;
  97
  98         Collection<DataContainerChild<? extends PathArgument, ?>> value = yangModeledAnyXmlNode.getValue();
  99         assertEquals(2, value.size());
 100
 101         Optional<DataContainerChild<? extends PathArgument, ?>> child = yangModeledAnyXmlNode
 102                 .getChild(new NodeIdentifier(myContainer1));
 103         assertTrue(child.orElse(null) instanceof ContainerNode);
 104         ContainerNode myContainerNode1 = (ContainerNode) child.get();
 105
 106         Optional<DataContainerChild<? extends PathArgument, ?>> child2 = myContainerNode1.getChild(new NodeIdentifier(
 107                 myLeaf1));
 108         assertTrue(child2.orElse(null) instanceof LeafNode);
 109         LeafNode<?> leafNode1 = (LeafNode<?>) child2.get();
 110
 111         Object leafNode1Value = leafNode1.getValue();
 112         assertEquals("value1", leafNode1Value);
 113     }
 114
 115     @Test
 116     public void testRealSchemaContextFromFoo() throws Exception {
 117         final InputStream resourceAsStream = YangModeledAnyXMLDeserializationTest.class.getResourceAsStream(
 118                 "/anyxml-support/xml/foo.xml");
 119         final Module foo = schemaContext.findModules("foo").iterator().next();
 120         final YangModeledAnyxmlSchemaNode myAnyXmlData = (YangModeledAnyxmlSchemaNode) foo.findDataChildByName(
 121                 QName.create(foo.getQNameModule(), "my-anyxml-data")).get();
 122
 123         final XMLStreamReader reader = UntrustedXML.createXMLStreamReader(resourceAsStream);
 124         final NormalizedNodeResult result = new NormalizedNodeResult();
 125
 126         final NormalizedNodeStreamWriter streamWriter = ImmutableNormalizedNodeStreamWriter.from(result);
 127
 128         final XmlParserStream xmlParser = XmlParserStream.create(streamWriter, schemaContext, myAnyXmlData);
 129         xmlParser.parse(reader);
 130
 131         final NormalizedNode<?, ?> output = result.getResult();
 132         assertTrue(output instanceof YangModeledAnyXmlNode);
 133         final YangModeledAnyXmlNode yangModeledAnyXmlNode = (YangModeledAnyXmlNode) output;
 134
 135         DataSchemaNode schemaOfAnyXmlData = yangModeledAnyXmlNode.getSchemaOfAnyXmlData();
 136         DataSchemaNode expectedSchemaOfAnyXmlData = schemaContext.findDataChildByName(myContainer2).get();
 137         assertEquals(expectedSchemaOfAnyXmlData, schemaOfAnyXmlData);
 138
 139         Collection<DataContainerChild<? extends PathArgument, ?>> value = yangModeledAnyXmlNode.getValue();
 140         assertEquals(2, value.size());
 141
 142         Optional<DataContainerChild<? extends PathArgument, ?>> child2 = yangModeledAnyXmlNode
 143                 .getChild(new NodeIdentifier(innerContainer));
 144         assertTrue(child2.orElse(null) instanceof ContainerNode);
 145         ContainerNode innerContainerNode = (ContainerNode) child2.get();
 146
 147         Optional<DataContainerChild<? extends PathArgument, ?>> child3 = innerContainerNode
 148                 .getChild(new NodeIdentifier(myLeaf2));
 149         assertTrue(child3.orElse(null) instanceof LeafNode);
 150         LeafNode<?> leafNode2 = (LeafNode<?>) child3.get();
 151
 152         Object leafNode2Value = leafNode2.getValue();
 153         assertEquals("any-xml-leaf-2-value", leafNode2Value);
 154
 155         Optional<DataContainerChild<? extends PathArgument, ?>> child4 = yangModeledAnyXmlNode
 156                 .getChild(new NodeIdentifier(myLeaf3));
 157         assertTrue(child4.orElse(null) instanceof LeafNode);
 158         LeafNode<?> leafNode3 = (LeafNode<?>) child4.get();
 159
 160         Object leafNode3Value = leafNode3.getValue();
 161         assertEquals("any-xml-leaf-3-value", leafNode3Value);
 162     }
 163
 164     private static final class YangModeledAnyXMLSchemaNodeImplTest implements YangModeledAnyxmlSchemaNode {
 165         private final QName qname;
 166         private final ContainerSchemaNode contentSchema;
 167
 168         YangModeledAnyXMLSchemaNodeImplTest(final QName qname, final ContainerSchemaNode contentSchema) {
 169             this.qname = qname;
 170             this.contentSchema = contentSchema;
 171         }
 172
 173         @Override
 174         public boolean isAugmenting() {
 175             return false;
 176         }
 177
 178         @Override
 179         public boolean isAddedByUses() {
 180             return false;
 181         }
 182
 183         @Override
 184         public boolean isConfiguration() {
 185             return false;
 186         }
 187
 188         @Override
 189         public boolean isMandatory() {
 190             return false;
 191         }
 192
 193         @Override
 194         public QName getQName() {
 195             return qname;
 196         }
 197
 198         @Override
 199         public SchemaPath getPath() {
 200             return null;
 201         }
 202
 203         @Override
 204         public Optional<String> getDescription() {
 205             return Optional.empty();
 206         }
 207
 208         @Override
 209         public Optional<String> getReference() {
 210             return Optional.empty();
 211         }
 212
 213         @Override
 214         public Status getStatus() {
 215             return Status.CURRENT;
 216         }
 217
 218         @Override
 219         public ContainerSchemaNode getSchemaOfAnyXmlData() {
 220             return contentSchema;
 221         }
 222
 223         @Override
 224         public Collection<MustDefinition> getMustConstraints() {
 225             return ImmutableSet.of();
 226         }
 227
 228         @Override
 229         public Optional<RevisionAwareXPath> getWhenCondition() {
 230             return Optional.empty();
 231         }
 232     }
 233 }