Peel DataSchemaNode-only users of InstanceIdentifierContext
[netconf.git] / restconf / restconf-nb-bierman02 / src / test / java / org / opendaylight / controller / sal / restconf / impl / nn / to / xml / test / NnInstanceIdentifierToXmlTest.java
1 /*
2  * Copyright (c) 2014 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.controller.sal.restconf.impl.nn.to.xml.test;
9
10 import static org.junit.Assert.assertNotNull;
11 import static org.junit.Assert.assertTrue;
12
13 import com.google.common.collect.Iterables;
14 import java.io.ByteArrayOutputStream;
15 import java.io.OutputStream;
16 import java.net.URISyntaxException;
17 import java.util.List;
18 import javax.ws.rs.core.MediaType;
19 import org.junit.BeforeClass;
20 import org.junit.Test;
21 import org.opendaylight.controller.sal.rest.impl.test.providers.AbstractBodyReaderTest;
22 import org.opendaylight.netconf.sal.rest.impl.NormalizedNodeContext;
23 import org.opendaylight.netconf.sal.rest.impl.NormalizedNodeXmlBodyWriter;
24 import org.opendaylight.netconf.sal.restconf.impl.ControllerContext;
25 import org.opendaylight.restconf.common.context.InstanceIdentifierContext;
26 import org.opendaylight.yangtools.yang.common.QName;
27 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
28 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifierWithPredicates;
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.MapEntryNode;
32 import org.opendaylight.yangtools.yang.data.api.schema.SystemLeafSetNode;
33 import org.opendaylight.yangtools.yang.data.api.schema.SystemMapNode;
34 import org.opendaylight.yangtools.yang.data.api.schema.builder.CollectionNodeBuilder;
35 import org.opendaylight.yangtools.yang.data.api.schema.builder.DataContainerNodeBuilder;
36 import org.opendaylight.yangtools.yang.data.api.schema.builder.ListNodeBuilder;
37 import org.opendaylight.yangtools.yang.data.impl.schema.SchemaAwareBuilders;
38 import org.opendaylight.yangtools.yang.model.api.ContainerSchemaNode;
39 import org.opendaylight.yangtools.yang.model.api.DataNodeContainer;
40 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
41 import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
42 import org.opendaylight.yangtools.yang.model.api.LeafListSchemaNode;
43 import org.opendaylight.yangtools.yang.model.api.LeafSchemaNode;
44 import org.opendaylight.yangtools.yang.model.api.ListSchemaNode;
45
46 public class NnInstanceIdentifierToXmlTest extends AbstractBodyReaderTest {
47     private static EffectiveModelContext schemaContext;
48
49     private final NormalizedNodeXmlBodyWriter xmlBodyWriter = new NormalizedNodeXmlBodyWriter();
50
51     public NnInstanceIdentifierToXmlTest() {
52         super(schemaContext, null);
53     }
54
55     @BeforeClass
56     public static void initialization() throws URISyntaxException {
57         schemaContext = schemaContextLoader("/instanceidentifier/yang", schemaContext);
58     }
59
60     @Test
61     public void nnAsYangInstanceIdentifierAugmentLeafList() throws Exception {
62         final NormalizedNodeContext normalizedNodeContext = prepareNNCLeafList();
63
64         final OutputStream output = new ByteArrayOutputStream();
65
66         xmlBodyWriter.writeTo(normalizedNodeContext, null, null, null, mediaType, null, output);
67
68         assertNotNull(output);
69
70         final String outputJson = output.toString();
71
72         assertTrue(outputJson.contains("<cont xmlns="));
73         assertTrue(outputJson.contains(
74                 '"' + "instance:identifier:module" + '"'));
75         assertTrue(outputJson.contains(">"));
76
77         assertTrue(outputJson.contains("<cont1>"));
78
79         assertTrue(outputJson.contains("<lf11 xmlns="));
80         assertTrue(outputJson.contains(
81                 '"' + "augment:module:leaf:list" + '"'));
82         assertTrue(outputJson.contains(">"));
83         assertTrue(outputJson.contains("/instanceidentifier/"));
84         assertTrue(outputJson.contains("</lf11>"));
85
86         assertTrue(outputJson.contains("<lflst11 xmlns="));
87         assertTrue(outputJson.contains(
88                 '"' + "augment:module:leaf:list" + '"'));
89         assertTrue(outputJson.contains(">"));
90         assertTrue(outputJson.contains("lflst11 value"));
91         assertTrue(outputJson.contains("</lflst11>"));
92
93         assertTrue(outputJson.contains("</cont1>"));
94         assertTrue(outputJson.contains("</cont>"));
95     }
96
97     private static NormalizedNodeContext prepareNNCLeafList() throws URISyntaxException {
98         final QName cont = QName.create("instance:identifier:module", "2014-01-17",
99                 "cont");
100         final QName cont1 = QName.create("instance:identifier:module", "2014-01-17",
101                 "cont1");
102         final QName lflst11 = QName.create("augment:module:leaf:list", "2014-01-17",
103                 "lflst11");
104         final QName lf11 = QName.create("augment:module:leaf:list", "2014-01-17",
105                 "lf11");
106
107         final DataSchemaNode schemaCont = schemaContext.getDataChildByName(cont);
108
109         final DataContainerNodeBuilder<NodeIdentifier, ContainerNode> dataCont = SchemaAwareBuilders
110                 .containerBuilder((ContainerSchemaNode) schemaCont);
111
112         final DataSchemaNode schemaCont1 = ((ContainerSchemaNode) schemaCont)
113                 .getDataChildByName(cont1);
114
115         final DataContainerNodeBuilder<NodeIdentifier, ContainerNode> dataCont1 = SchemaAwareBuilders
116                 .containerBuilder((ContainerSchemaNode) schemaCont1);
117
118         final List<DataSchemaNode> instanceLfLst11 = ControllerContext
119                 .findInstanceDataChildrenByName(
120                         (DataNodeContainer) schemaCont1, lflst11.getLocalName());
121
122         final DataSchemaNode lfLst11Schema = Iterables
123                 .getFirst(instanceLfLst11, null);
124         final ListNodeBuilder<Object, SystemLeafSetNode<Object>> lfLst11Data = SchemaAwareBuilders
125                 .leafSetBuilder((LeafListSchemaNode) lfLst11Schema);
126
127         lfLst11Data.withChild(SchemaAwareBuilders.leafSetEntryBuilder((LeafListSchemaNode) lfLst11Schema)
128                 .withValue("lflst11 value").build());
129         dataCont1.withChild(lfLst11Data.build());
130
131         final List<DataSchemaNode> instanceLf11 = ControllerContext
132                 .findInstanceDataChildrenByName(
133                         (DataNodeContainer) schemaCont1, lf11.getLocalName());
134         final DataSchemaNode lf11Schema = Iterables.getFirst(instanceLf11, null);
135
136         dataCont1.withChild(SchemaAwareBuilders.leafBuilder((LeafSchemaNode) lf11Schema)
137                 .withValue("/instanceidentifier/").build());
138         dataCont.withChild(dataCont1.build());
139
140         return new NormalizedNodeContext(InstanceIdentifierContext.ofDataSchemaNode(schemaContext, schemaCont),
141                 dataCont.build());
142     }
143
144     @Test
145     public void nnAsYangInstanceIdentifierAugment() throws Exception {
146
147         final NormalizedNodeContext normalizedNodeContext = preparNNC();
148         final OutputStream output = new ByteArrayOutputStream();
149
150         xmlBodyWriter.writeTo(normalizedNodeContext, null, null, null,
151                 mediaType, null, output);
152
153         assertNotNull(output);
154
155         final String outputJson = output.toString();
156
157         assertTrue(outputJson.contains("<cont xmlns="));
158         assertTrue(outputJson.contains(
159                 '"' + "instance:identifier:module" + '"'));
160         assertTrue(outputJson.contains(">"));
161
162         assertTrue(outputJson.contains("<cont1>"));
163
164         assertTrue(outputJson.contains("<lst11 xmlns="));
165         assertTrue(outputJson.contains('"' + "augment:module" + '"'));
166         assertTrue(outputJson.contains(">"));
167
168         assertTrue(outputJson.contains(
169                 "<keyvalue111>keyvalue111</keyvalue111>"));
170         assertTrue(outputJson.contains(
171                 "<keyvalue112>keyvalue112</keyvalue112>"));
172
173         assertTrue(outputJson.contains("<lf111 xmlns="));
174         assertTrue(outputJson.contains(
175                 '"' + "augment:augment:module" + '"'));
176         assertTrue(outputJson.contains(">/cont/cont1/lf12</lf111>"));
177
178         assertTrue(outputJson.contains("<lf112 xmlns="));
179         assertTrue(outputJson.contains(
180                 '"' + "augment:augment:module" + '"'));
181         assertTrue(outputJson.contains(">lf12 value</lf112>"));
182
183         assertTrue(outputJson.contains("</lst11></cont1></cont>"));
184     }
185
186     private static NormalizedNodeContext preparNNC() {
187         final QName cont = QName.create("instance:identifier:module", "2014-01-17",
188                 "cont");
189         final QName cont1 = QName.create("instance:identifier:module", "2014-01-17",
190                 "cont1");
191         final QName lst11 = QName.create("augment:module", "2014-01-17", "lst11");
192         final QName lf11 = QName.create("augment:augment:module", "2014-01-17",
193                 "lf111");
194         final QName lf12 = QName.create("augment:augment:module", "2014-01-17",
195                 "lf112");
196         final QName keyvalue111 = QName.create("augment:module", "2014-01-17",
197                 "keyvalue111");
198         final QName keyvalue112 = QName.create("augment:module", "2014-01-17",
199                 "keyvalue112");
200
201         final DataSchemaNode schemaCont = schemaContext.getDataChildByName(cont);
202
203         final DataContainerNodeBuilder<NodeIdentifier, ContainerNode> dataCont = SchemaAwareBuilders
204                 .containerBuilder((ContainerSchemaNode) schemaCont);
205
206         final DataSchemaNode schemaCont1 = ((ContainerSchemaNode) schemaCont)
207                 .getDataChildByName(cont1);
208
209         final DataContainerNodeBuilder<NodeIdentifier, ContainerNode> dataCont1 = SchemaAwareBuilders
210                 .containerBuilder((ContainerSchemaNode) schemaCont1);
211
212         final List<DataSchemaNode> instanceLst11 = ControllerContext
213                 .findInstanceDataChildrenByName(
214                         (DataNodeContainer) schemaCont1, lst11.getLocalName());
215         final DataSchemaNode lst11Schema = Iterables.getFirst(instanceLst11, null);
216
217         final CollectionNodeBuilder<MapEntryNode, SystemMapNode> dataLst11 = SchemaAwareBuilders
218                 .mapBuilder((ListSchemaNode) lst11Schema);
219
220         final DataContainerNodeBuilder<NodeIdentifierWithPredicates, MapEntryNode> dataLst11Vaule = SchemaAwareBuilders
221                 .mapEntryBuilder((ListSchemaNode) lst11Schema);
222
223         dataLst11Vaule.withChild(buildLeaf(lst11Schema, keyvalue111, dataLst11, "keyvalue111"));
224
225         dataLst11Vaule.withChild(buildLeaf(lst11Schema, keyvalue112, dataLst11, "keyvalue112"));
226
227         dataLst11Vaule.withChild(buildLeaf(lst11Schema, lf11, dataLst11, "/cont/cont1/lf12"));
228
229         dataLst11Vaule.withChild(buildLeaf(lst11Schema, lf12, dataLst11, "lf12 value"));
230
231         dataLst11.withChild(dataLst11Vaule.build());
232
233         dataCont1.withChild(dataLst11.build());
234         dataCont.withChild(dataCont1.build());
235
236         return new NormalizedNodeContext(InstanceIdentifierContext.ofDataSchemaNode(schemaContext, schemaCont),
237             dataCont.build());
238     }
239
240     private static DataContainerChild buildLeaf(final DataSchemaNode lst11Schema, final QName qname,
241             final CollectionNodeBuilder<MapEntryNode, SystemMapNode> dataLst11, final Object value) {
242
243         final List<DataSchemaNode> instanceLf = ControllerContext.findInstanceDataChildrenByName(
244             (DataNodeContainer) lst11Schema, qname.getLocalName());
245         final DataSchemaNode schemaLf = Iterables.getFirst(instanceLf, null);
246
247         return SchemaAwareBuilders.leafBuilder((LeafSchemaNode) schemaLf).withValue(value).build();
248     }
249
250     @Override
251     protected MediaType getMediaType() {
252         return null;
253     }
254 }