2 * Copyright (c) 2014 Cisco Systems, Inc. and others. All rights reserved.
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
8 package org.opendaylight.controller.sal.restconf.impl.nn.to.xml.test;
10 import static org.junit.Assert.assertTrue;
12 import com.google.common.base.Preconditions;
13 import com.google.common.collect.Iterables;
14 import java.io.ByteArrayOutputStream;
15 import java.io.IOException;
16 import java.io.OutputStream;
17 import java.net.URISyntaxException;
18 import java.util.List;
19 import javax.ws.rs.WebApplicationException;
20 import javax.ws.rs.core.MediaType;
21 import org.junit.BeforeClass;
22 import org.junit.Test;
23 import org.opendaylight.controller.sal.rest.impl.test.providers.AbstractBodyReaderTest;
24 import org.opendaylight.netconf.sal.rest.impl.NormalizedNodeXmlBodyWriter;
25 import org.opendaylight.netconf.sal.restconf.impl.ControllerContext;
26 import org.opendaylight.restconf.common.context.InstanceIdentifierContext;
27 import org.opendaylight.restconf.common.context.NormalizedNodeContext;
28 import org.opendaylight.yangtools.yang.common.QName;
29 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
30 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
31 import org.opendaylight.yangtools.yang.data.impl.schema.Builders;
32 import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.DataContainerNodeBuilder;
33 import org.opendaylight.yangtools.yang.model.api.ContainerSchemaNode;
34 import org.opendaylight.yangtools.yang.model.api.DataNodeContainer;
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.LeafSchemaNode;
39 public class NnToXmlWithDataFromSeveralModulesTest extends
40 AbstractBodyReaderTest {
42 private final NormalizedNodeXmlBodyWriter xmlBodyWriter;
43 private static EffectiveModelContext schemaContext;
45 public NnToXmlWithDataFromSeveralModulesTest() {
46 super(schemaContext, null);
47 xmlBodyWriter = new NormalizedNodeXmlBodyWriter();
51 public static void initialize() {
52 schemaContext = schemaContextLoader("/nn-to-xml/data-of-several-modules/yang", schemaContext);
56 public void dataFromSeveralModulesToXmlTest()
57 throws WebApplicationException, IOException, URISyntaxException {
58 final NormalizedNodeContext normalizedNodeContext = prepareNormalizedNodeContext();
59 final OutputStream output = new ByteArrayOutputStream();
60 xmlBodyWriter.writeTo(normalizedNodeContext, null, null, null,
61 mediaType, null, output);
63 final String outputString = output.toString();
65 assertTrue(outputString
68 + "urn:ietf:params:xml:ns:netconf:base:1.0"
71 assertTrue(outputString.contains(
72 "<cont_m2 xmlns=" + '"' + "module:two" + '"' + '>'));
73 assertTrue(outputString.contains("<lf1_m2>lf1 m2 value</lf1_m2>"));
74 assertTrue(outputString.contains("<contB_m2/>"));
75 assertTrue(outputString.contains("</cont_m2>"));
78 assertTrue(outputString.contains(
79 "<cont_m1 xmlns=" + '"' + "module:one" + '"' + '>'));
80 assertTrue(outputString.contains("<contB_m1/>"));
81 assertTrue(outputString.contains("<lf1_m1>lf1 m1 value</lf1_m1>"));
82 assertTrue(outputString.contains("</cont_m1>"));
85 assertTrue(output.toString().contains("</data>"));
89 protected MediaType getMediaType() {
90 // TODO Auto-generated method stub
94 private static NormalizedNodeContext prepareNormalizedNodeContext() {
95 final String rev = "2014-01-17";
97 final DataContainerNodeBuilder<NodeIdentifier, ContainerNode> dataContSchemaContNode = Builders
98 .containerBuilder(schemaContext);
100 final DataContainerNodeBuilder<NodeIdentifier, ContainerNode> modul1 = buildContBuilderMod1(
101 "module:one", rev, "cont_m1", "contB_m1", "lf1_m1",
103 dataContSchemaContNode.withChild(modul1.build());
105 final DataContainerNodeBuilder<NodeIdentifier, ContainerNode> modul2 = buildContBuilderMod1(
106 "module:two", rev, "cont_m2", "contB_m2", "lf1_m2",
108 dataContSchemaContNode.withChild(modul2.build());
110 final NormalizedNodeContext testNormalizedNodeContext = new NormalizedNodeContext(
111 new InstanceIdentifierContext<>(null, schemaContext, null, schemaContext),
112 dataContSchemaContNode.build());
114 return testNormalizedNodeContext;
117 private static DataContainerNodeBuilder<NodeIdentifier, ContainerNode> buildContBuilderMod1(
118 final String uri, final String rev, final String cont, final String contB, final String lf1,
119 final String lf1Value) {
120 final QName contQname = QName.create(uri, rev, cont);
121 final QName contBQname = QName.create(uri, rev, contB);
122 final QName lf1Qname = QName.create(contQname, lf1);
124 final DataSchemaNode contSchemaNode = schemaContext
125 .getDataChildByName(contQname);
126 final DataContainerNodeBuilder<NodeIdentifier, ContainerNode> dataContainerNodeAttrBuilder = Builders
127 .containerBuilder((ContainerSchemaNode) contSchemaNode);
129 Preconditions.checkState(contSchemaNode instanceof ContainerSchemaNode);
130 final List<DataSchemaNode> instanceLf1_m1 = ControllerContext
131 .findInstanceDataChildrenByName(
132 (DataNodeContainer) contSchemaNode,
133 lf1Qname.getLocalName());
134 final DataSchemaNode schemaLf1_m1 = Iterables.getFirst(instanceLf1_m1, null);
136 dataContainerNodeAttrBuilder.withChild(Builders
137 .leafBuilder((LeafSchemaNode) schemaLf1_m1)
138 .withValue(lf1Value).build());
140 final DataSchemaNode contBSchemaNode = ((ContainerSchemaNode) contSchemaNode)
141 .getDataChildByName(contBQname);
143 final DataContainerNodeBuilder<NodeIdentifier, ContainerNode> dataContainerB = Builders
144 .containerBuilder((ContainerSchemaNode) contBSchemaNode);
146 return dataContainerNodeAttrBuilder.withChild(dataContainerB.build());