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;
11 import com.google.common.base.Preconditions;
12 import com.google.common.collect.Iterables;
13 import java.io.ByteArrayOutputStream;
14 import java.io.IOException;
15 import java.io.OutputStream;
16 import java.net.URISyntaxException;
17 import java.util.List;
18 import javax.ws.rs.WebApplicationException;
19 import javax.ws.rs.core.MediaType;
20 import org.junit.BeforeClass;
21 import org.junit.Test;
22 import org.opendaylight.controller.sal.rest.impl.NormalizedNodeXmlBodyWriter;
23 import org.opendaylight.controller.sal.rest.impl.test.providers.AbstractBodyReaderTest;
24 import org.opendaylight.controller.sal.restconf.impl.ControllerContext;
25 import org.opendaylight.controller.sal.restconf.impl.InstanceIdentifierContext;
26 import org.opendaylight.controller.sal.restconf.impl.NormalizedNodeContext;
27 import org.opendaylight.yangtools.yang.common.QName;
28 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
29 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
30 import org.opendaylight.yangtools.yang.data.impl.schema.Builders;
31 import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.DataContainerNodeAttrBuilder;
32 import org.opendaylight.yangtools.yang.model.api.ContainerSchemaNode;
33 import org.opendaylight.yangtools.yang.model.api.DataNodeContainer;
34 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
35 import org.opendaylight.yangtools.yang.model.api.LeafSchemaNode;
36 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
38 public class NnToXmlWithDataFromSeveralModulesTest extends
39 AbstractBodyReaderTest {
41 private final NormalizedNodeXmlBodyWriter xmlBodyWriter;
42 private static SchemaContext schemaContext;
44 public NnToXmlWithDataFromSeveralModulesTest() throws NoSuchFieldException,
47 xmlBodyWriter = new NormalizedNodeXmlBodyWriter();
51 public static void initialize() {
52 schemaContext = schemaContextLoader(
53 "/nn-to-xml/data-of-several-modules/yang", schemaContext);
54 controllerContext.setSchemas(schemaContext);
58 public void dataFromSeveralModulesToXmlTest()
59 throws WebApplicationException, IOException, URISyntaxException {
60 final NormalizedNodeContext normalizedNodeContext = prepareNormalizedNodeContext();
61 final OutputStream output = new ByteArrayOutputStream();
62 xmlBodyWriter.writeTo(normalizedNodeContext, null, null, null,
63 mediaType, null, output);
65 final String outputString = output.toString();
67 assertTrue(outputString
70 + "urn:ietf:params:xml:ns:netconf:base:1.0"
73 assertTrue(outputString.contains(
74 "<cont_m2 xmlns=" + '"' + "module:two" + '"' + '>'));
75 assertTrue(outputString.contains("<lf1_m2>lf1 m2 value</lf1_m2>"));
76 assertTrue(outputString.contains("<contB_m2></contB_m2>"));
77 assertTrue(outputString.contains("</cont_m2>"));
80 assertTrue(outputString.contains(
81 "<cont_m1 xmlns=" + '"' + "module:one" + '"' + '>'));
82 assertTrue(outputString.contains("<contB_m1></contB_m1>"));
83 assertTrue(outputString.contains("<lf1_m1>lf1 m1 value</lf1_m1>"));
84 assertTrue(outputString.contains("</cont_m1>"));
87 assertTrue(output.toString().contains("</data>"));
91 protected MediaType getMediaType() {
92 // TODO Auto-generated method stub
96 private NormalizedNodeContext prepareNormalizedNodeContext() {
97 final String rev = "2014-01-17";
99 final DataSchemaNode schemaContNode = schemaContext;
101 assertTrue(schemaContNode instanceof ContainerSchemaNode);
103 final DataContainerNodeAttrBuilder<NodeIdentifier, ContainerNode> dataContSchemaContNode = Builders
104 .containerBuilder((ContainerSchemaNode) schemaContNode);
106 final DataContainerNodeAttrBuilder<NodeIdentifier, ContainerNode> modul1 = buildContBuilderMod1(
107 "module:one", rev, "cont_m1", "contB_m1", "lf1_m1",
109 dataContSchemaContNode.withChild(modul1.build());
111 final DataContainerNodeAttrBuilder<NodeIdentifier, ContainerNode> modul2 = buildContBuilderMod1(
112 "module:two", rev, "cont_m2", "contB_m2", "lf1_m2",
114 dataContSchemaContNode.withChild(modul2.build());
116 final NormalizedNodeContext testNormalizedNodeContext = new NormalizedNodeContext(
117 new InstanceIdentifierContext<DataSchemaNode>(null,
118 schemaContNode, null, schemaContext),
119 dataContSchemaContNode.build());
121 return testNormalizedNodeContext;
124 private DataContainerNodeAttrBuilder<NodeIdentifier, ContainerNode> buildContBuilderMod1(
125 final String uri, final String rev, final String cont, final String contB, final String lf1,
126 final String lf1_value) {
127 final QName contQname = QName.create(uri, rev, cont);
128 final QName contBQname = QName.create(uri, rev, contB);
129 final QName lf1Qname = QName.create(contQname, lf1);
131 final DataSchemaNode contSchemaNode = schemaContext
132 .getDataChildByName(contQname);
133 final DataContainerNodeAttrBuilder<NodeIdentifier, ContainerNode> dataContainerNodeAttrBuilder = Builders
134 .containerBuilder((ContainerSchemaNode) contSchemaNode);
136 Preconditions.checkState(contSchemaNode instanceof ContainerSchemaNode);
137 final List<DataSchemaNode> instanceLf1_m1 = ControllerContext
138 .findInstanceDataChildrenByName(
139 (DataNodeContainer) contSchemaNode,
140 lf1Qname.getLocalName());
141 final DataSchemaNode schemaLf1_m1 = Iterables.getFirst(instanceLf1_m1, null);
143 dataContainerNodeAttrBuilder.withChild(Builders
144 .leafBuilder((LeafSchemaNode) schemaLf1_m1)
145 .withValue(lf1_value).build());
147 final DataSchemaNode contBSchemaNode = ((ContainerSchemaNode) contSchemaNode)
148 .getDataChildByName(contBQname);
150 final DataContainerNodeAttrBuilder<NodeIdentifier, ContainerNode> dataContainerB = Builders
151 .containerBuilder((ContainerSchemaNode) contBSchemaNode);
153 return dataContainerNodeAttrBuilder.withChild(dataContainerB.build());