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