restconf/sal-rest-connector/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.NormalizedNodeXmlBodyWriter;
  23 import org.opendaylight.netconf.sal.restconf.impl.ControllerContext;
  24 import org.opendaylight.netconf.sal.restconf.impl.InstanceIdentifierContext;
  25 import org.opendaylight.netconf.sal.restconf.impl.NormalizedNodeContext;
  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.LeafSetEntryNode;
  32 import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
  33 import org.opendaylight.yangtools.yang.data.api.schema.MapNode;
  34 import org.opendaylight.yangtools.yang.data.impl.schema.Builders;
  35 import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.CollectionNodeBuilder;
  36 import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.DataContainerNodeAttrBuilder;
  37 import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.ListNodeBuilder;
  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.LeafListSchemaNode;
  42 import org.opendaylight.yangtools.yang.model.api.LeafSchemaNode;
  43 import org.opendaylight.yangtools.yang.model.api.ListSchemaNode;
  44 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
  45
  46 public class NnInstanceIdentifierToXmlTest extends AbstractBodyReaderTest {
  47
  48     private static SchemaContext schemaContext;
  49     NormalizedNodeXmlBodyWriter xmlBodyWriter;
  50
  51     public NnInstanceIdentifierToXmlTest() throws NoSuchFieldException,
  52             SecurityException {
  53         super();
  54         xmlBodyWriter = new NormalizedNodeXmlBodyWriter();
  55     }
  56
  57     @BeforeClass
  58     public static void initialization() throws URISyntaxException {
  59         schemaContext = schemaContextLoader("/instanceidentifier/yang",
  60                 schemaContext);
  61         controllerContext.setSchemas(schemaContext);
  62     }
  63
  64     @Test
  65     public void nnAsYangInstanceIdentifierAugmentLeafList() throws Exception {
  66         final NormalizedNodeContext normalizedNodeContext = prepareNNCLeafList();
  67
  68         final OutputStream output = new ByteArrayOutputStream();
  69
  70         xmlBodyWriter.writeTo(normalizedNodeContext, null, null, null,
  71                 mediaType, null, output);
  72
  73         assertNotNull(output);
  74
  75         final String outputJson = output.toString();
  76
  77         assertTrue(outputJson.contains("<cont xmlns="));
  78         assertTrue(outputJson.contains(
  79                 '"' + "instance:identifier:module" + '"'));
  80         assertTrue(outputJson.contains(">"));
  81
  82         assertTrue(outputJson.contains("<cont1>"));
  83
  84         assertTrue(outputJson.contains("<lf11 xmlns="));
  85         assertTrue(outputJson.contains(
  86                 '"' + "augment:module:leaf:list" + '"'));
  87         assertTrue(outputJson.contains(">"));
  88         assertTrue(outputJson.contains("/instanceidentifier/"));
  89         assertTrue(outputJson.contains("</lf11>"));
  90
  91         assertTrue(outputJson.contains("<lflst11 xmlns="));
  92         assertTrue(outputJson.contains(
  93                 '"' + "augment:module:leaf:list" + '"'));
  94         assertTrue(outputJson.contains(">"));
  95         assertTrue(outputJson.contains("lflst11 value"));
  96         assertTrue(outputJson.contains("</lflst11>"));
  97
  98         assertTrue(outputJson.contains("</cont1>"));
  99         assertTrue(outputJson.contains("</cont>"));
 100     }
 101
 102     private NormalizedNodeContext prepareNNCLeafList()
 103             throws URISyntaxException {
 104         final QName cont = QName.create("instance:identifier:module", "2014-01-17",
 105                 "cont");
 106         final QName cont1 = QName.create("instance:identifier:module", "2014-01-17",
 107                 "cont1");
 108         final QName lflst11 = QName.create("augment:module:leaf:list", "2014-01-17",
 109                 "lflst11");
 110         final QName lf11 = QName.create("augment:module:leaf:list", "2014-01-17",
 111                 "lf11");
 112
 113         final DataSchemaNode schemaCont = schemaContext.getDataChildByName(cont);
 114
 115         final DataContainerNodeAttrBuilder<NodeIdentifier, ContainerNode> dataCont = Builders
 116                 .containerBuilder((ContainerSchemaNode) schemaCont);
 117
 118         final DataSchemaNode schemaCont1 = ((ContainerSchemaNode) schemaCont)
 119                 .getDataChildByName(cont1);
 120
 121         final DataContainerNodeAttrBuilder<NodeIdentifier, ContainerNode> dataCont1 = Builders
 122                 .containerBuilder((ContainerSchemaNode) schemaCont1);
 123
 124         final List<DataSchemaNode> instanceLfLst11 = ControllerContext
 125                 .findInstanceDataChildrenByName(
 126                         (DataNodeContainer) schemaCont1, lflst11.getLocalName());
 127
 128         final DataSchemaNode lfLst11Schema = Iterables
 129                 .getFirst(instanceLfLst11, null);
 130         final ListNodeBuilder<Object, LeafSetEntryNode<Object>> lfLst11Data = Builders
 131                 .leafSetBuilder((LeafListSchemaNode) lfLst11Schema);
 132
 133         lfLst11Data.withChild(Builders
 134                 .leafSetEntryBuilder((LeafListSchemaNode) lfLst11Schema)
 135                 .withValue("lflst11 value").build());
 136         dataCont1.withChild(lfLst11Data.build());
 137
 138         final List<DataSchemaNode> instanceLf11 = ControllerContext
 139                 .findInstanceDataChildrenByName(
 140                         (DataNodeContainer) schemaCont1, lf11.getLocalName());
 141         final DataSchemaNode lf11Schema = Iterables.getFirst(instanceLf11, null);
 142
 143         dataCont1.withChild(Builders.leafBuilder((LeafSchemaNode) lf11Schema)
 144                 .withValue("/instanceidentifier/").build());
 145         dataCont.withChild(dataCont1.build());
 146
 147         final NormalizedNodeContext testNormalizedNodeContext = new NormalizedNodeContext(
 148                 new InstanceIdentifierContext<DataSchemaNode>(null, schemaCont,
 149                         null, schemaContext), dataCont.build());
 150
 151         return testNormalizedNodeContext;
 152     }
 153
 154     @Test
 155     public void nnAsYangInstanceIdentifierAugment() throws Exception {
 156
 157         final NormalizedNodeContext normalizedNodeContext = preparNNC();
 158         final NormalizedNodeXmlBodyWriter xmlBodyWriter = new NormalizedNodeXmlBodyWriter();
 159         final OutputStream output = new ByteArrayOutputStream();
 160
 161         xmlBodyWriter.writeTo(normalizedNodeContext, null, null, null,
 162                 mediaType, null, output);
 163
 164         assertNotNull(output);
 165
 166         final String outputJson = output.toString();
 167
 168         assertTrue(outputJson.contains("<cont xmlns="));
 169         assertTrue(outputJson.contains(
 170                 '"' + "instance:identifier:module" + '"'));
 171         assertTrue(outputJson.contains(">"));
 172
 173         assertTrue(outputJson.contains("<cont1>"));
 174
 175         assertTrue(outputJson.contains("<lst11 xmlns="));
 176         assertTrue(outputJson.contains('"' + "augment:module" + '"'));
 177         assertTrue(outputJson.contains(">"));
 178
 179         assertTrue(outputJson.contains(
 180                 "<keyvalue111>keyvalue111</keyvalue111>"));
 181         assertTrue(outputJson.contains(
 182                 "<keyvalue112>keyvalue112</keyvalue112>"));
 183
 184         assertTrue(outputJson.contains("<lf111 xmlns="));
 185         assertTrue(outputJson.contains(
 186                 '"' + "augment:augment:module" + '"'));
 187         assertTrue(outputJson.contains(">/cont/cont1/lf12</lf111>"));
 188
 189         assertTrue(outputJson.contains("<lf112 xmlns="));
 190         assertTrue(outputJson.contains(
 191                 '"' + "augment:augment:module" + '"'));
 192         assertTrue(outputJson.contains(">lf12 value</lf112>"));
 193
 194         assertTrue(outputJson.contains("</lst11></cont1></cont>"));
 195     }
 196
 197     private NormalizedNodeContext preparNNC() {
 198         final QName cont = QName.create("instance:identifier:module", "2014-01-17",
 199                 "cont");
 200         final QName cont1 = QName.create("instance:identifier:module", "2014-01-17",
 201                 "cont1");
 202         final QName lst11 = QName.create("augment:module", "2014-01-17", "lst11");
 203         final QName lf11 = QName.create("augment:augment:module", "2014-01-17",
 204                 "lf111");
 205         final QName lf12 = QName.create("augment:augment:module", "2014-01-17",
 206                 "lf112");
 207         final QName keyvalue111 = QName.create("augment:module", "2014-01-17",
 208                 "keyvalue111");
 209         final QName keyvalue112 = QName.create("augment:module", "2014-01-17",
 210                 "keyvalue112");
 211
 212         final DataSchemaNode schemaCont = schemaContext.getDataChildByName(cont);
 213
 214         final DataContainerNodeAttrBuilder<NodeIdentifier, ContainerNode> dataCont = Builders
 215                 .containerBuilder((ContainerSchemaNode) schemaCont);
 216
 217         final DataSchemaNode schemaCont1 = ((ContainerSchemaNode) schemaCont)
 218                 .getDataChildByName(cont1);
 219
 220         final DataContainerNodeAttrBuilder<NodeIdentifier, ContainerNode> dataCont1 = Builders
 221                 .containerBuilder((ContainerSchemaNode) schemaCont1);
 222
 223         final List<DataSchemaNode> instanceLst11 = ControllerContext
 224                 .findInstanceDataChildrenByName(
 225                         (DataNodeContainer) schemaCont1, lst11.getLocalName());
 226         final DataSchemaNode lst11Schema = Iterables.getFirst(instanceLst11, null);
 227
 228         final CollectionNodeBuilder<MapEntryNode, MapNode> dataLst11 = Builders
 229                 .mapBuilder((ListSchemaNode) lst11Schema);
 230
 231         final DataContainerNodeAttrBuilder<NodeIdentifierWithPredicates, MapEntryNode> dataLst11Vaule = Builders
 232                 .mapEntryBuilder((ListSchemaNode) lst11Schema);
 233
 234         dataLst11Vaule.withChild(buildLeaf(lst11Schema, keyvalue111, dataLst11,
 235                 "keyvalue111"));
 236
 237         dataLst11Vaule.withChild(buildLeaf(lst11Schema, keyvalue112, dataLst11,
 238                 "keyvalue112"));
 239
 240         dataLst11Vaule.withChild(buildLeaf(lst11Schema, lf11, dataLst11,
 241                 "/cont/cont1/lf12"));
 242
 243         dataLst11Vaule.withChild(buildLeaf(lst11Schema, lf12, dataLst11,
 244                 "lf12 value"));
 245
 246         dataLst11.withChild(dataLst11Vaule.build());
 247
 248         dataCont1.withChild(dataLst11.build());
 249         dataCont.withChild(dataCont1.build());
 250
 251         final NormalizedNodeContext testNormalizedNodeContext = new NormalizedNodeContext(
 252                 new InstanceIdentifierContext<DataSchemaNode>(null, schemaCont,
 253                         null, schemaContext), dataCont.build());
 254
 255         return testNormalizedNodeContext;
 256     }
 257
 258     private DataContainerChild<?, ?> buildLeaf(final DataSchemaNode lst11Schema,
 259             final QName qname,
 260             final CollectionNodeBuilder<MapEntryNode, MapNode> dataLst11, final Object value) {
 261
 262         final List<DataSchemaNode> instanceLf = ControllerContext
 263                 .findInstanceDataChildrenByName(
 264                         (DataNodeContainer) lst11Schema, qname.getLocalName());
 265         final DataSchemaNode schemaLf = Iterables.getFirst(instanceLf, null);
 266
 267         return Builders.leafBuilder((LeafSchemaNode) schemaLf).withValue(value)
 268                 .build();
 269     }
 270
 271     @Override
 272     protected MediaType getMediaType() {
 273         return null;
 274     }
 275 }