Cleanup testing AugmentationIdentifier instantiations
[yangtools.git] / yang / yang-data-impl / src / test / java / org / opendaylight / yangtools / yang / data / impl / schema / ImmutableNormalizedNodeStreamWriterTest.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.impl.schema;
10
11 import static org.junit.Assert.assertEquals;
12 import static org.junit.Assert.assertNotNull;
13
14 import com.google.common.collect.ImmutableSet;
15 import java.io.IOException;
16 import java.net.URI;
17 import java.net.URISyntaxException;
18 import java.text.ParseException;
19 import java.util.ArrayList;
20 import java.util.HashMap;
21 import java.util.List;
22 import java.util.Map;
23 import javax.xml.transform.dom.DOMSource;
24 import org.junit.Before;
25 import org.junit.Test;
26 import org.opendaylight.yangtools.yang.common.QName;
27 import org.opendaylight.yangtools.yang.common.QNameModule;
28 import org.opendaylight.yangtools.yang.common.Revision;
29 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.AugmentationIdentifier;
30 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
31 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifierWithPredicates;
32 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeWithValue;
33 import org.opendaylight.yangtools.yang.data.api.schema.AnyXmlNode;
34 import org.opendaylight.yangtools.yang.data.api.schema.AugmentationNode;
35 import org.opendaylight.yangtools.yang.data.api.schema.ChoiceNode;
36 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
37 import org.opendaylight.yangtools.yang.data.api.schema.LeafNode;
38 import org.opendaylight.yangtools.yang.data.api.schema.LeafSetNode;
39 import org.opendaylight.yangtools.yang.data.api.schema.MapNode;
40 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
41 import org.opendaylight.yangtools.yang.data.api.schema.OrderedMapNode;
42 import org.opendaylight.yangtools.yang.data.api.schema.UnkeyedListEntryNode;
43 import org.opendaylight.yangtools.yang.data.api.schema.UnkeyedListNode;
44 import org.opendaylight.yangtools.yang.data.api.schema.stream.NormalizedNodeStreamWriter;
45 import org.opendaylight.yangtools.yang.data.api.schema.stream.NormalizedNodeWriter;
46
47 public class ImmutableNormalizedNodeStreamWriterTest {
48
49     private QNameModule bazModule;
50
51     private QName outerContainer;
52
53     private QName myContainer1;
54     private QName myKeyedList;
55     private QName myKeyLeaf;
56     private QName myLeafInList1;
57     private QName myLeafInList2;
58     private QName myOrderedList;
59     private QName myKeyLeafInOrderedList;
60     private QName myLeafInOrderedList1;
61     private QName myLeafInOrderedList2;
62     private QName myUnkeyedList;
63     private QName myLeafInUnkeyedList;
64     private QName myLeaf1;
65     private QName myLeafList;
66     private QName myOrderedLeafList;
67
68     private QName myContainer2;
69     private QName innerContainer;
70     private QName myLeaf2;
71     private QName myLeaf3;
72     private QName myChoice;
73     private QName myLeafInCase2;
74     private QName myAnyxml;
75
76     private QName myContainer3;
77     private QName myDoublyKeyedList;
78     private QName myFirstKeyLeaf;
79     private QName mySecondKeyLeaf;
80     private QName myLeafInList3;
81
82     private DOMSource anyxmlDomSource;
83
84     @Before
85     public void setup() throws URISyntaxException, ParseException {
86         bazModule = QNameModule.create(new URI("baz-namespace"), Revision.of("1970-01-01"));
87
88         outerContainer = QName.create(bazModule, "outer-container");
89
90         myContainer1 = QName.create(bazModule, "my-container-1");
91         myKeyedList = QName.create(bazModule, "my-keyed-list");
92         myKeyLeaf = QName.create(bazModule, "my-key-leaf");
93         myLeafInList1 = QName.create(bazModule, "my-leaf-in-list-1");
94         myLeafInList2 = QName.create(bazModule, "my-leaf-in-list-2");
95         myOrderedList = QName.create(bazModule, "my-ordered-list");
96         myKeyLeafInOrderedList = QName.create(bazModule, "my-key-leaf-in-ordered-list");
97         myLeafInOrderedList1 = QName.create(bazModule, "my-leaf-in-ordered-list-1");
98         myLeafInOrderedList2 = QName.create(bazModule, "my-leaf-in-ordered-list-2");
99         myUnkeyedList = QName.create(bazModule, "my-unkeyed-list");
100         myLeafInUnkeyedList = QName.create(bazModule, "my-leaf-in-unkeyed-list");
101         myLeaf1 = QName.create(bazModule, "my-leaf-1");
102         myLeafList = QName.create(bazModule, "my-leaf-list");
103         myOrderedLeafList = QName.create(bazModule, "my-ordered-leaf-list");
104
105         myContainer2 = QName.create(bazModule, "my-container-2");
106         innerContainer = QName.create(bazModule, "inner-container");
107         myLeaf2 = QName.create(bazModule, "my-leaf-2");
108         myLeaf3 = QName.create(bazModule, "my-leaf-3");
109         myChoice = QName.create(bazModule, "my-choice");
110         myLeafInCase2 = QName.create(bazModule, "my-leaf-in-case-2");
111         myAnyxml = QName.create(bazModule, "my-anyxml");
112
113         myContainer3 = QName.create(bazModule, "my-container-3");
114         myDoublyKeyedList = QName.create(bazModule, "my-doubly-keyed-list");
115         myFirstKeyLeaf = QName.create(bazModule, "my-first-key-leaf");
116         mySecondKeyLeaf = QName.create(bazModule, "my-second-key-leaf");
117         myLeafInList3 = QName.create(bazModule, "my-leaf-in-list-3");
118
119         anyxmlDomSource = new DOMSource();
120     }
121
122     @Test
123     public void testImmutableNormalizedNodeStreamWriter() throws IOException {
124         final NormalizedNodeResult result = new NormalizedNodeResult();
125         final NormalizedNodeStreamWriter immutableNormalizedNodeStreamWriter = ImmutableNormalizedNodeStreamWriter
126                 .from(result);
127         final NormalizedNodeWriter normalizedNodeWriter = NormalizedNodeWriter
128                 .forStreamWriter(immutableNormalizedNodeStreamWriter);
129         normalizedNodeWriter.write(buildOuterContainerNode());
130
131         final NormalizedNode<?, ?> output = result.getResult();
132         assertNotNull(output);
133
134         final NormalizedNode<?, ?> expectedNormalizedNode = buildOuterContainerNode();
135         assertNotNull(expectedNormalizedNode);
136
137         assertEquals(expectedNormalizedNode, output);
138     }
139
140     private NormalizedNode<?, ?> buildOuterContainerNode() {
141         // my-container-1
142         MapNode myKeyedListNode = Builders.mapBuilder().withNodeIdentifier(new NodeIdentifier(myKeyedList))
143                 .withChild(Builders.mapEntryBuilder().withNodeIdentifier(
144                         NodeIdentifierWithPredicates.of(myKeyedList, myKeyLeaf, "listkeyvalue1"))
145                         .withChild(Builders.leafBuilder().withNodeIdentifier(new NodeIdentifier(myLeafInList1))
146                                 .withValue("listleafvalue1").build())
147                         .withChild(Builders.leafBuilder().withNodeIdentifier(new NodeIdentifier(myLeafInList2))
148                                 .withValue("listleafvalue2").build()).build())
149                 .withChild(Builders.mapEntryBuilder().withNodeIdentifier(
150                         NodeIdentifierWithPredicates.of(myKeyedList, myKeyLeaf, "listkeyvalue2"))
151                         .withChild(Builders.leafBuilder().withNodeIdentifier(new NodeIdentifier(myLeafInList1))
152                                 .withValue("listleafvalue12").build())
153                         .withChild(Builders.leafBuilder().withNodeIdentifier(new NodeIdentifier(myLeafInList2))
154                                 .withValue("listleafvalue22").build()).build()).build();
155
156         OrderedMapNode myOrderedListNode = Builders.orderedMapBuilder().withNodeIdentifier(
157             new NodeIdentifier(myOrderedList))
158                 .withChild(Builders.mapEntryBuilder().withNodeIdentifier(
159                         NodeIdentifierWithPredicates.of(myOrderedList, myKeyLeafInOrderedList, "olistkeyvalue1"))
160                         .withChild(Builders.leafBuilder().withNodeIdentifier(new NodeIdentifier(myLeafInOrderedList1))
161                                 .withValue("olistleafvalue1").build())
162                         .withChild(Builders.leafBuilder().withNodeIdentifier(new NodeIdentifier(myLeafInOrderedList2))
163                                 .withValue("olistleafvalue2").build()).build())
164                 .withChild(Builders.mapEntryBuilder().withNodeIdentifier(
165                         NodeIdentifierWithPredicates.of(myOrderedList, myKeyLeafInOrderedList, "olistkeyvalue2"))
166                         .withChild(Builders.leafBuilder().withNodeIdentifier(new NodeIdentifier(myLeafInOrderedList1))
167                                 .withValue("olistleafvalue12").build())
168                         .withChild(Builders.leafBuilder().withNodeIdentifier(new NodeIdentifier(myLeafInOrderedList2))
169                                 .withValue("olistleafvalue22").build()).build()).build();
170
171         final UnkeyedListEntryNode unkeyedListEntry = Builders.unkeyedListEntryBuilder()
172                 .withNodeIdentifier(new NodeIdentifier(myLeafInUnkeyedList))
173                 .withChild(ImmutableNodes.leafNode(myLeafInUnkeyedList, "foo")).build();
174
175         final List<UnkeyedListEntryNode> unkeyedListEntries = new ArrayList<>();
176         unkeyedListEntries.add(unkeyedListEntry);
177
178         UnkeyedListNode myUnkeyedListNode = Builders.unkeyedListBuilder().withNodeIdentifier(
179                 new NodeIdentifier(myUnkeyedList))
180                 .withValue(unkeyedListEntries).build();
181
182         LeafNode<?> myLeaf1Node = Builders.leafBuilder().withNodeIdentifier(new NodeIdentifier(myLeaf1))
183                 .withValue("value1").build();
184
185         LeafSetNode<?> myLeafListNode = Builders.leafSetBuilder().withNodeIdentifier(new NodeIdentifier(myLeafList))
186                 .withChild(Builders.leafSetEntryBuilder().withNodeIdentifier(
187                         new NodeWithValue<>(myLeafList, "lflvalue1")).withValue("lflvalue1").build())
188                 .withChild(Builders.leafSetEntryBuilder().withNodeIdentifier(
189                         new NodeWithValue<>(myLeafList, "lflvalue2")).withValue("lflvalue2").build()).build();
190
191         LeafSetNode<?> myOrderedLeafListNode = Builders.orderedLeafSetBuilder().withNodeIdentifier(
192                 new NodeIdentifier(myOrderedLeafList))
193                 .withChild(Builders.leafSetEntryBuilder().withNodeIdentifier(
194                         new NodeWithValue<>(myOrderedLeafList, "olflvalue1")).withValue("olflvalue1").build())
195                 .withChild(Builders.leafSetEntryBuilder().withNodeIdentifier(
196                         new NodeWithValue<>(myOrderedLeafList, "olflvalue2")).withValue("olflvalue2").build()).build();
197
198         ContainerNode myContainer1Node = Builders.containerBuilder().withNodeIdentifier(
199                 new NodeIdentifier(myContainer1))
200                 .withChild(myKeyedListNode)
201                 .withChild(myOrderedListNode)
202                 .withChild(myUnkeyedListNode)
203                 .withChild(myLeaf1Node)
204                 .withChild(myLeafListNode)
205                 .withChild(myOrderedLeafListNode)
206                 .build();
207
208         // my-container-2
209         ContainerNode innerContainerNode = Builders.containerBuilder().withNodeIdentifier(
210                 new NodeIdentifier(innerContainer))
211                 .withChild(Builders.leafBuilder().withNodeIdentifier(new NodeIdentifier(myLeaf2))
212                         .withValue("value2").build()).build();
213
214         LeafNode<?> myLeaf3Node = Builders.leafBuilder().withNodeIdentifier(new NodeIdentifier(myLeaf3))
215                 .withValue("value3").build();
216
217         ChoiceNode myChoiceNode = Builders.choiceBuilder().withNodeIdentifier(new NodeIdentifier(myChoice))
218                 .withChild(Builders.leafBuilder().withNodeIdentifier(new NodeIdentifier(myLeafInCase2))
219                         .withValue("case2value").build()).build();
220
221         AnyXmlNode myAnyxmlNode = Builders.anyXmlBuilder().withNodeIdentifier(new NodeIdentifier(myAnyxml))
222                 .withValue(anyxmlDomSource).build();
223
224         ContainerNode myContainer2Node = Builders.containerBuilder().withNodeIdentifier(
225                 new NodeIdentifier(myContainer2))
226                 .withChild(innerContainerNode)
227                 .withChild(myLeaf3Node)
228                 .withChild(myChoiceNode)
229                 .withChild(myAnyxmlNode).build();
230
231         // my-container-3
232         Map<QName, Object> keys = new HashMap<>();
233         keys.put(myFirstKeyLeaf, "listkeyvalue1");
234         keys.put(mySecondKeyLeaf, "listkeyvalue2");
235
236         MapNode myDoublyKeyedListNode = Builders.mapBuilder().withNodeIdentifier(new NodeIdentifier(myDoublyKeyedList))
237                 .withChild(Builders.mapEntryBuilder().withNodeIdentifier(
238                         NodeIdentifierWithPredicates.of(myDoublyKeyedList, keys))
239                         .withChild(Builders.leafBuilder().withNodeIdentifier(
240                                 new NodeIdentifier(myLeafInList3)).withValue("listleafvalue1").build()).build())
241                 .build();
242
243         AugmentationNode myDoublyKeyedListAugNode = Builders.augmentationBuilder().withNodeIdentifier(
244                 new AugmentationIdentifier(ImmutableSet.of(myDoublyKeyedList)))
245                 .withChild(myDoublyKeyedListNode).build();
246
247         ContainerNode myContainer3Node = Builders.containerBuilder().withNodeIdentifier(
248                 new NodeIdentifier(myContainer3))
249                 .withChild(myDoublyKeyedListAugNode).build();
250
251         AugmentationNode myContainer3AugNode = Builders.augmentationBuilder().withNodeIdentifier(
252                 new AugmentationIdentifier(ImmutableSet.of(myContainer3)))
253                 .withChild(myContainer3Node).build();
254
255         ContainerNode outerContainerNode = Builders.containerBuilder().withNodeIdentifier(
256                 new NodeIdentifier(outerContainer))
257                 .withChild(myContainer1Node)
258                 .withChild(myContainer2Node)
259                 .withChild(myContainer3AugNode).build();
260
261         return outerContainerNode;
262     }
263 }