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