plugins/sal-netconf-connector/src/test/java/org/opendaylight/netconf/sal/connect/netconf/schema/mapping/SchemalessMessageTransformerTest.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.assertNull;
  12 import static org.junit.Assert.assertTrue;
  13
  14 import javax.xml.transform.dom.DOMSource;
  15 import org.custommonkey.xmlunit.Diff;
  16 import org.custommonkey.xmlunit.XMLUnit;
  17 import org.junit.Before;
  18 import org.junit.Test;
  19 import org.opendaylight.mdsal.dom.api.DOMNotification;
  20 import org.opendaylight.netconf.api.NetconfMessage;
  21 import org.opendaylight.netconf.api.xml.XmlUtil;
  22 import org.opendaylight.netconf.sal.connect.netconf.util.NetconfMessageTransformUtil;
  23 import org.opendaylight.netconf.sal.connect.util.MessageCounter;
  24 import org.opendaylight.yangtools.yang.common.QName;
  25 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
  26 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
  27 import org.opendaylight.yangtools.yang.data.api.schema.DOMSourceAnyxmlNode;
  28 import org.w3c.dom.Document;
  29 import org.w3c.dom.Node;
  30
  31 public class SchemalessMessageTransformerTest {
  32
  33     static {
  34         XMLUnit.setIgnoreWhitespace(true);
  35     }
  36
  37     private static final String EXP_REQUEST =
  38             "<rpc message-id=\"m-0\" xmlns=\"urn:ietf:params:xml:ns:netconf:base:1.0\">\n"
  39                     + "<test-rpc xmlns=\"test-ns\">\n"
  40                     + "<input>aaa</input>\n"
  41                     + "</test-rpc>\n"
  42                     + "</rpc>";
  43     private static final String EXP_REPLY =
  44             "<rpc-reply message-id=\"m-0\" xmlns=\"urn:ietf:params:xml:ns:netconf:base:1.0\">\n"
  45                     + "<output xmlns=\"test-ns\">aaa</output>\n"
  46                     + "</rpc-reply>";
  47     private static final String OK_REPLY = "<rpc-reply message-id=\"101\"\n"
  48             + "xmlns=\"urn:ietf:params:xml:ns:netconf:base:1.0\">\n"
  49             + "<ok/>\n"
  50             + "</rpc-reply>\n";
  51
  52     private static final QName TEST_RPC = QName.create("test-ns", "2016-10-13", "test-rpc");
  53
  54     private SchemalessMessageTransformer transformer;
  55
  56     @Before
  57     public void setUp() throws Exception {
  58         transformer = new SchemalessMessageTransformer(new MessageCounter());
  59     }
  60
  61     @Test
  62     public void toNotification() throws Exception {
  63         final Document payload = XmlUtil.readXmlToDocument(getClass().getResourceAsStream("/notification-payload.xml"));
  64         final NetconfMessage netconfMessage = new NetconfMessage(payload);
  65         final DOMNotification domNotification = transformer.toNotification(netconfMessage);
  66         assertEquals(domNotification.getType().lastNodeIdentifier(),
  67                 SchemalessMessageTransformer.SCHEMALESS_NOTIFICATION_PAYLOAD.getNodeType());
  68         final QName qName =
  69                 QName.create("org:opendaylight:notification:test:ns:yang:user-notification", "user-visited-page");
  70         final DOMSourceAnyxmlNode dataContainerChild = (DOMSourceAnyxmlNode) domNotification.getBody()
  71                 .findChildByArg(new NodeIdentifier(qName)).get();
  72         final Diff diff = XMLUnit.compareXML(payload, dataContainerChild.body().getNode().getOwnerDocument());
  73         assertTrue(diff.toString(), diff.similar());
  74
  75     }
  76
  77     @Test
  78     public void toRpcRequest() throws Exception {
  79         final Node src = XmlUtil.readXmlToDocument("<test-rpc xmlns=\"test-ns\"><input>aaa</input></test-rpc>");
  80         final NetconfMessage netconfMessage = transformer.toRpcRequest(TEST_RPC, new DOMSource(src));
  81         final Diff diff = XMLUnit.compareXML(XmlUtil.readXmlToDocument(EXP_REQUEST), netconfMessage.getDocument());
  82         assertTrue(diff.toString(), diff.similar());
  83     }
  84
  85     @Test
  86     public void toRpcResult() throws Exception {
  87         final Document doc = XmlUtil.readXmlToDocument(EXP_REPLY);
  88         final NetconfMessage netconfMessage = new NetconfMessage(doc);
  89         final DOMSource result = transformer.toRpcResult(RpcResultBuilder.success(netconfMessage).build(), TEST_RPC);
  90         final Document domSourceDoc = (Document) result.getNode();
  91         final Diff diff = XMLUnit.compareXML(XmlUtil.readXmlToDocument(EXP_REPLY), domSourceDoc);
  92         assertTrue(diff.toString(), diff.similar());
  93     }
  94
  95     @Test
  96     public void toEmptyRpcResult() throws Exception {
  97         final Document doc = XmlUtil.readXmlToDocument(OK_REPLY);
  98         final DOMSource result = transformer.toRpcResult(RpcResultBuilder.success(new NetconfMessage(doc)).build(),
  99             NetconfMessageTransformUtil.NETCONF_COMMIT_QNAME);
 100         assertNull(result);
 101     }
 102 }