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