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.netconf.sal.restconf.impl.InstanceIdentifierContext;
27 import org.opendaylight.netconf.sal.restconf.impl.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.DataContainerNodeAttrBuilder;
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.LeafSchemaNode;
37 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
39 public class NnToXmlWithDataFromSeveralModulesTest extends
40 AbstractBodyReaderTest {
42 private final NormalizedNodeXmlBodyWriter xmlBodyWriter;
43 private static SchemaContext schemaContext;
45 public NnToXmlWithDataFromSeveralModulesTest() throws NoSuchFieldException,
48 xmlBodyWriter = new NormalizedNodeXmlBodyWriter();
52 public static void initialize() {
53 schemaContext = schemaContextLoader(
54 "/nn-to-xml/data-of-several-modules/yang", schemaContext);
55 CONTROLLER_CONTEXT.setSchemas(schemaContext);
59 public void dataFromSeveralModulesToXmlTest()
60 throws WebApplicationException, IOException, URISyntaxException {
61 final NormalizedNodeContext normalizedNodeContext = prepareNormalizedNodeContext();
62 final OutputStream output = new ByteArrayOutputStream();
63 xmlBodyWriter.writeTo(normalizedNodeContext, null, null, null,
64 mediaType, null, output);
66 final String outputString = output.toString();
68 assertTrue(outputString
71 + "urn:ietf:params:xml:ns:netconf:base:1.0"
74 assertTrue(outputString.contains(
75 "<cont_m2 xmlns=" + '"' + "module:two" + '"' + '>'));
76 assertTrue(outputString.contains("<lf1_m2>lf1 m2 value</lf1_m2>"));
77 assertTrue(outputString.contains("<contB_m2></contB_m2>"));
78 assertTrue(outputString.contains("</cont_m2>"));
81 assertTrue(outputString.contains(
82 "<cont_m1 xmlns=" + '"' + "module:one" + '"' + '>'));
83 assertTrue(outputString.contains("<contB_m1></contB_m1>"));
84 assertTrue(outputString.contains("<lf1_m1>lf1 m1 value</lf1_m1>"));
85 assertTrue(outputString.contains("</cont_m1>"));
88 assertTrue(output.toString().contains("</data>"));
92 protected MediaType getMediaType() {
93 // TODO Auto-generated method stub
97 private static NormalizedNodeContext prepareNormalizedNodeContext() {
98 final String rev = "2014-01-17";
100 final DataSchemaNode schemaContNode = schemaContext;
102 assertTrue(schemaContNode instanceof ContainerSchemaNode);
104 final DataContainerNodeAttrBuilder<NodeIdentifier, ContainerNode> dataContSchemaContNode = Builders
105 .containerBuilder((ContainerSchemaNode) schemaContNode);
107 final DataContainerNodeAttrBuilder<NodeIdentifier, ContainerNode> modul1 = buildContBuilderMod1(
108 "module:one", rev, "cont_m1", "contB_m1", "lf1_m1",
110 dataContSchemaContNode.withChild(modul1.build());
112 final DataContainerNodeAttrBuilder<NodeIdentifier, ContainerNode> modul2 = buildContBuilderMod1(
113 "module:two", rev, "cont_m2", "contB_m2", "lf1_m2",
115 dataContSchemaContNode.withChild(modul2.build());
117 final NormalizedNodeContext testNormalizedNodeContext = new NormalizedNodeContext(
118 new InstanceIdentifierContext<>(null,
119 schemaContNode, null, schemaContext),
120 dataContSchemaContNode.build());
122 return testNormalizedNodeContext;
125 private static DataContainerNodeAttrBuilder<NodeIdentifier, ContainerNode> buildContBuilderMod1(
126 final String uri, final String rev, final String cont, final String contB, final String lf1,
127 final String lf1Value) {
128 final QName contQname = QName.create(uri, rev, cont);
129 final QName contBQname = QName.create(uri, rev, contB);
130 final QName lf1Qname = QName.create(contQname, lf1);
132 final DataSchemaNode contSchemaNode = schemaContext
133 .getDataChildByName(contQname);
134 final DataContainerNodeAttrBuilder<NodeIdentifier, ContainerNode> dataContainerNodeAttrBuilder = Builders
135 .containerBuilder((ContainerSchemaNode) contSchemaNode);
137 Preconditions.checkState(contSchemaNode instanceof ContainerSchemaNode);
138 final List<DataSchemaNode> instanceLf1_m1 = ControllerContext
139 .findInstanceDataChildrenByName(
140 (DataNodeContainer) contSchemaNode,
141 lf1Qname.getLocalName());
142 final DataSchemaNode schemaLf1_m1 = Iterables.getFirst(instanceLf1_m1, null);
144 dataContainerNodeAttrBuilder.withChild(Builders
145 .leafBuilder((LeafSchemaNode) schemaLf1_m1)
146 .withValue(lf1Value).build());
148 final DataSchemaNode contBSchemaNode = ((ContainerSchemaNode) contSchemaNode)
149 .getDataChildByName(contBQname);
151 final DataContainerNodeAttrBuilder<NodeIdentifier, ContainerNode> dataContainerB = Builders
152 .containerBuilder((ContainerSchemaNode) contBSchemaNode);
154 return dataContainerNodeAttrBuilder.withChild(dataContainerB.build());