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