Bug 7159: Add yang-test-util artifact
[yangtools.git] / 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 Init() 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 = (ImmutableContainerNodeBuilder) ImmutableContainerNodeSchemaAwareBuilder
95                         .create(yangModeledAnyXML.getSchemaOfAnyXmlData());
96
97         final NormalizedNodeStreamWriter streamWriter = ImmutableNormalizedNodeStreamWriter.from(result);
98
99         final XmlParserStream xmlParser = XmlParserStream.create(streamWriter, schemaContext, yangModeledAnyXML);
100         xmlParser.parse(reader);
101
102         final NormalizedNode<?, ?> output = result.build();
103
104         Collection<DataContainerChild<? extends PathArgument, ?>> value = ((ContainerNode) output).getValue();
105         assertEquals(2, value.size());
106
107         Optional<DataContainerChild<? extends PathArgument, ?>> child = ((ContainerNode) output)
108                 .getChild(new NodeIdentifier(myContainer1));
109         assertTrue(child.orNull() instanceof ContainerNode);
110         ContainerNode myContainerNode1 = (ContainerNode) child.get();
111
112         Optional<DataContainerChild<? extends PathArgument, ?>> child2 = myContainerNode1.getChild(new NodeIdentifier(
113                 myLeaf1));
114         assertTrue(child2.orNull() instanceof LeafNode);
115         LeafNode<?> LeafNode1 = (LeafNode<?>) child2.get();
116
117         Object leafNode1Value = LeafNode1.getValue();
118         assertEquals("value1", leafNode1Value);
119     }
120
121     @Test
122     public void testRealSchemaContextFromFoo() throws Exception {
123         final InputStream resourceAsStream = YangModeledAnyXMLDeserializationTest.class.getResourceAsStream(
124                 "/anyxml-support/xml/foo.xml");
125
126         final XMLInputFactory factory = XMLInputFactory.newInstance();
127         final XMLStreamReader reader = factory.createXMLStreamReader(resourceAsStream);
128
129         final ImmutableContainerNodeBuilder result = (ImmutableContainerNodeBuilder) ImmutableContainerNodeSchemaAwareBuilder
130                 .create(schemaContext);
131
132         final NormalizedNodeStreamWriter streamWriter = ImmutableNormalizedNodeStreamWriter.from(result);
133
134         final XmlParserStream xmlParser = XmlParserStream.create(streamWriter, schemaContext);
135         xmlParser.parse(reader);
136
137         final NormalizedNode<?, ?> output = result.build();
138
139         Optional<DataContainerChild<? extends PathArgument, ?>> child = ((ContainerNode) output).getChild(
140                 new NodeIdentifier(myAnyXMLDataFoo));
141         assertTrue(child.orNull() instanceof YangModeledAnyXmlNode);
142         YangModeledAnyXmlNode yangModeledAnyXmlNode = (YangModeledAnyXmlNode) child.get();
143
144         DataSchemaNode schemaOfAnyXmlData = yangModeledAnyXmlNode.getSchemaOfAnyXmlData();
145         DataSchemaNode expectedSchemaOfAnyXmlData = schemaContext.getDataChildByName(myContainer2);
146         assertEquals(expectedSchemaOfAnyXmlData, schemaOfAnyXmlData);
147
148         Collection<DataContainerChild<? extends PathArgument, ?>> value = yangModeledAnyXmlNode.getValue();
149         assertEquals(2, value.size());
150
151         Optional<DataContainerChild<? extends PathArgument, ?>> child2 = yangModeledAnyXmlNode
152                 .getChild(new NodeIdentifier(innerContainer));
153         assertTrue(child2.orNull() instanceof ContainerNode);
154         ContainerNode innerContainerNode = (ContainerNode) child2.get();
155
156         Optional<DataContainerChild<? extends PathArgument, ?>> child3 = innerContainerNode
157                 .getChild(new NodeIdentifier(myLeaf2));
158         assertTrue(child3.orNull() instanceof LeafNode);
159         LeafNode<?> LeafNode2 = (LeafNode<?>) child3.get();
160
161         Object leafNode2Value = LeafNode2.getValue();
162         assertEquals("any-xml-leaf-2-value", leafNode2Value);
163
164         Optional<DataContainerChild<? extends PathArgument, ?>> child4 = yangModeledAnyXmlNode
165                 .getChild(new NodeIdentifier(myLeaf3));
166         assertTrue(child4.orNull() instanceof LeafNode);
167         LeafNode<?> LeafNode3 = (LeafNode<?>) child4.get();
168
169         Object leafNode3Value = LeafNode3.getValue();
170         assertEquals("any-xml-leaf-3-value", leafNode3Value);
171     }
172
173     private static class YangModeledAnyXMLSchemaNodeImplTest implements YangModeledAnyXmlSchemaNode {
174         private final QName qName;
175         private final ContainerSchemaNode contentSchema;
176
177         private YangModeledAnyXMLSchemaNodeImplTest(QName qName, ContainerSchemaNode contentSchema) {
178             this.qName = qName;
179             this.contentSchema = contentSchema;
180         }
181
182         @Override
183         public boolean isAugmenting() {
184             return false;
185         }
186
187         @Override
188         public boolean isAddedByUses() {
189             return false;
190         }
191
192         @Override
193         public boolean isConfiguration() {
194             return false;
195         }
196
197         @Override
198         public ConstraintDefinition getConstraints() {
199             return null;
200         }
201
202         @Nonnull
203         @Override
204         public QName getQName() {
205             return qName;
206         }
207
208         @Nonnull
209         @Override
210         public SchemaPath getPath() {
211             return null;
212         }
213
214         @Nonnull
215         @Override
216         public List<UnknownSchemaNode> getUnknownSchemaNodes() {
217             return null;
218         }
219
220         @Override
221         public String getDescription() {
222             return null;
223         }
224
225         @Override
226         public String getReference() {
227             return null;
228         }
229
230         @Nonnull
231         @Override
232         public Status getStatus() {
233             return null;
234         }
235
236         @Nonnull
237         @Override
238         public ContainerSchemaNode getSchemaOfAnyXmlData() {
239             return contentSchema;
240         }
241     }
242 }