Rework NormalizedNode type hierarchy
[yangtools.git] / yang / yang-data-codec-xml / src / test / java / org / opendaylight / yangtools / yang / data / codec / xml / Bug890Test.java
1 /*
2  * Copyright (c) 2018 Pantheon Technologies s.r.o. 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.assertNotNull;
12 import static org.junit.Assert.assertTrue;
13
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.util.xml.UntrustedXML;
22 import org.opendaylight.yangtools.yang.common.QName;
23 import org.opendaylight.yangtools.yang.common.QNameModule;
24 import org.opendaylight.yangtools.yang.common.Revision;
25 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
26 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifierWithPredicates;
27 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
28 import org.opendaylight.yangtools.yang.data.api.schema.DataContainerChild;
29 import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
30 import org.opendaylight.yangtools.yang.data.api.schema.MapNode;
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.test.util.YangParserTestUtils;
39
40 public class Bug890Test {
41     private static final QNameModule FOO_MODULE = QNameModule.create(URI.create("foo"), Revision.of("2018-07-10"));
42     private static final QName OUTGOING_LABELS_QNAME = QName.create(FOO_MODULE, "outgoing-labels");
43     private static final QName INDEX_QNAME = QName.create(FOO_MODULE, "index");
44
45     private EffectiveModelContext schemaContext;
46
47     @Before
48     public void setUp() throws Exception {
49         schemaContext = YangParserTestUtils.parseYangResource("/bug890/yang/foo.yang");
50     }
51
52     @Test
53     public void testinputXml() throws Exception {
54         final InputStream resourceAsStream = XmlToNormalizedNodesTest.class.getResourceAsStream("/bug890/xml/foo.xml");
55         final Module fooModule = schemaContext.getModules().iterator().next();
56         final Optional<DataSchemaNode> rootCont = fooModule.findDataChildByName(
57                 QName.create(fooModule.getQNameModule(), "root"));
58         assertTrue(rootCont.isPresent());
59         ContainerSchemaNode containerSchemaNode = (ContainerSchemaNode) rootCont.get();
60         assertNotNull(containerSchemaNode);
61
62         final XMLStreamReader reader = UntrustedXML.createXMLStreamReader(resourceAsStream);
63
64         final NormalizedNodeResult result = new NormalizedNodeResult();
65
66         final NormalizedNodeStreamWriter streamWriter = ImmutableNormalizedNodeStreamWriter.from(result);
67
68         final XmlParserStream xmlParser = XmlParserStream.create(streamWriter, schemaContext, containerSchemaNode);
69         xmlParser.parse(reader);
70
71         assertNotNull(result.getResult());
72         assertTrue(result.getResult() instanceof ContainerNode);
73         final ContainerNode rootContainer = (ContainerNode) result.getResult();
74
75         DataContainerChild myLeaf = rootContainer.childByArg(new NodeIdentifier(OUTGOING_LABELS_QNAME));
76         assertTrue(myLeaf instanceof ContainerNode);
77
78         ContainerNode outgoingLabelsContainer = (ContainerNode)myLeaf;
79         DataContainerChild outgoingLabelsList =
80                 outgoingLabelsContainer.childByArg(new NodeIdentifier(OUTGOING_LABELS_QNAME));
81         assertTrue(outgoingLabelsList instanceof MapNode);
82         MapNode outgoingLabelsMap = (MapNode) outgoingLabelsList;
83
84         assertEquals(2, outgoingLabelsMap.size());
85         Collection<MapEntryNode> labels = outgoingLabelsMap.body();
86         NodeIdentifierWithPredicates firstNodeId =
87                 NodeIdentifierWithPredicates.of(OUTGOING_LABELS_QNAME, INDEX_QNAME, 0);
88         NodeIdentifierWithPredicates secondNodeId =
89                 NodeIdentifierWithPredicates.of(OUTGOING_LABELS_QNAME, INDEX_QNAME, 1);
90         assertTrue(labels.stream().anyMatch(mapEntryNode -> mapEntryNode.getIdentifier().compareTo(firstNodeId) == 0));
91         assertTrue(labels.stream().anyMatch(mapEntryNode -> mapEntryNode.getIdentifier().compareTo(secondNodeId) == 0));
92     }
93 }