Split up MessageTransformer
[netconf.git] / netconf / sal-netconf-connector / src / test / java / org / opendaylight / netconf / sal / connect / netconf / schema / mapping / BaseRpcSchemalessTransformerTest.java
1 /*
2  * Copyright (c) 2016 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.netconf.sal.connect.netconf.schema.mapping;
9
10 import static org.junit.Assert.assertEquals;
11 import static org.junit.Assert.assertNotNull;
12 import static org.junit.Assert.assertTrue;
13
14 import java.io.InputStream;
15 import java.util.Optional;
16 import javax.xml.transform.dom.DOMSource;
17 import org.custommonkey.xmlunit.Diff;
18 import org.custommonkey.xmlunit.XMLUnit;
19 import org.junit.Before;
20 import org.junit.Test;
21 import org.opendaylight.mdsal.dom.api.DOMRpcResult;
22 import org.opendaylight.netconf.api.NetconfMessage;
23 import org.opendaylight.netconf.api.xml.XmlUtil;
24 import org.opendaylight.netconf.sal.connect.netconf.AbstractBaseSchemasTest;
25 import org.opendaylight.netconf.sal.connect.netconf.util.NetconfMessageTransformUtil;
26 import org.opendaylight.netconf.sal.connect.util.MessageCounter;
27 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.netconf.base._1._0.rev110601.copy.config.input.target.ConfigTarget;
28 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.netconf.base._1._0.rev110601.edit.config.input.EditContent;
29 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.netconf.base._1._0.rev110601.edit.config.input.target.config.target.Candidate;
30 import org.opendaylight.yangtools.yang.common.Empty;
31 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
32 import org.opendaylight.yangtools.yang.data.api.schema.ChoiceNode;
33 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
34 import org.opendaylight.yangtools.yang.data.api.schema.DOMSourceAnyxmlNode;
35 import org.opendaylight.yangtools.yang.data.impl.schema.Builders;
36 import org.w3c.dom.Document;
37 import org.w3c.dom.Element;
38
39 public class BaseRpcSchemalessTransformerTest extends AbstractBaseSchemasTest {
40
41     static {
42         XMLUnit.setIgnoreWhitespace(true);
43         XMLUnit.setIgnoreAttributeOrder(true);
44     }
45
46     private static final String EXP_RPC = "<rpc message-id=\"m-0\" xmlns=\"urn:ietf:params:xml:ns:netconf:base:1.0\">\n"
47             + "   <edit-config>\n"
48             + "       <target>\n"
49             + "           <candidate/>\n"
50             + "       </target>\n"
51             + "       <config>\n"
52             + "           <top xmlns=\"http://example.com/schema/1.2/config\">\n"
53             + "               <users xmlns:ns0=\"urn:ietf:params:xml:ns:netconf:base:1.0\" ns0:operation=\"replace\">\n"
54             + "                   <user>\n"
55             + "                       <name>fred</name>\n"
56             + "                   </user>\n"
57             + "               </users>\n"
58             + "           </top>\n"
59             + "       </config>\n"
60             + "   </edit-config>\n"
61             + "</rpc>\n";
62
63     BaseRpcSchemalessTransformer transformer;
64
65     @Before
66     public void setUp() throws Exception {
67         transformer = new BaseRpcSchemalessTransformer(BASE_SCHEMAS, new MessageCounter());
68     }
69
70     @Test
71     public void toRpcRequest() throws Exception {
72         final Document doc =
73                 XmlUtil.readXmlToDocument(getClass().getResourceAsStream("/schemaless/edit-config/container.xml"));
74         final DOMSourceAnyxmlNode xml = Builders.anyXmlBuilder()
75                 .withNodeIdentifier(new YangInstanceIdentifier.NodeIdentifier(
76                         NetconfMessageTransformUtil.NETCONF_CONFIG_QNAME))
77                 .withValue(new DOMSource(doc.getDocumentElement()))
78                 .build();
79         final ChoiceNode editContent = Builders.choiceBuilder()
80                 .withNodeIdentifier(new YangInstanceIdentifier.NodeIdentifier(EditContent.QNAME))
81                 .withChild(xml)
82                 .build();
83         final ChoiceNode candidate = Builders.choiceBuilder().withNodeIdentifier(
84                 new YangInstanceIdentifier.NodeIdentifier(ConfigTarget.QNAME))
85                 .withChild(Builders.leafBuilder().withNodeIdentifier(
86                         new YangInstanceIdentifier.NodeIdentifier(Candidate.QNAME))
87                     .withValue(Empty.value()).build())
88                 .build();
89         final ContainerNode target = Builders.containerBuilder()
90                 .withNodeIdentifier(
91                         new YangInstanceIdentifier.NodeIdentifier(NetconfMessageTransformUtil.NETCONF_TARGET_QNAME))
92                 .withChild(candidate)
93                 .build();
94         final ContainerNode editConfig = Builders.containerBuilder()
95                 .withNodeIdentifier(
96                     new YangInstanceIdentifier.NodeIdentifier(NetconfMessageTransformUtil.NETCONF_EDIT_CONFIG_QNAME))
97                 .withChild(editContent)
98                 .withChild(target)
99                 .build();
100         final NetconfMessage msg = transformer.toRpcRequest(
101                 NetconfMessageTransformUtil.NETCONF_EDIT_CONFIG_QNAME, editConfig);
102         final Diff diff = XMLUnit.compareXML(EXP_RPC, XmlUtil.toString(msg.getDocument()));
103         assertTrue(diff.toString(), diff.similar());
104     }
105
106     @Test
107     public void toRpcResult() throws Exception {
108         final Document doc = XmlUtil.readXmlToDocument(
109                 "<rpc-reply message-id=\"101\" xmlns=\"urn:ietf:params:xml:ns:netconf:base:1.0\"/>");
110         final InputStream stream = getClass().getResourceAsStream("/schemaless/get-config/container.xml");
111         final Element dataElement = XmlUtil.readXmlToElement(stream);
112         final Element element = (Element) doc.importNode(dataElement, true);
113         doc.getDocumentElement().appendChild(element);
114         final NetconfMessage msg = new NetconfMessage(doc);
115         final DOMRpcResult result = transformer.toRpcResult(msg, NetconfMessageTransformUtil.NETCONF_GET_CONFIG_QNAME);
116         assertNotNull(result.getResult());
117         final ContainerNode rpcReply = (ContainerNode) result.getResult();
118         assertEquals(NetconfMessageTransformUtil.NETCONF_RPC_REPLY_QNAME, rpcReply.getIdentifier().getNodeType());
119         final Optional<?> dataOpt = rpcReply.findChildByArg(NetconfMessageTransformUtil.NETCONF_DATA_NODEID);
120         assertTrue(dataOpt.isPresent());
121         final DOMSourceAnyxmlNode data = (DOMSourceAnyxmlNode) dataOpt.get();
122         final Diff diff = XMLUnit.compareXML(dataElement.getOwnerDocument(), (Document) data.body().getNode());
123         assertTrue(diff.toString(), diff.similar());
124     }
125 }