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