Rework BaseScheams
[netconf.git] / plugins / netconf-client-mdsal / src / test / java / org / opendaylight / netconf / client / mdsal / spi / SchemalessNetconfDeviceRpcTest.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.client.mdsal.spi;
9
10 import static org.junit.Assert.assertEquals;
11 import static org.mockito.ArgumentMatchers.any;
12 import static org.mockito.Mockito.doReturn;
13 import static org.mockito.Mockito.verify;
14
15 import com.google.common.util.concurrent.Futures;
16 import com.google.common.util.concurrent.ListenableFuture;
17 import java.net.InetSocketAddress;
18 import java.util.Set;
19 import javax.xml.transform.dom.DOMSource;
20 import org.junit.Before;
21 import org.junit.Test;
22 import org.junit.runner.RunWith;
23 import org.mockito.ArgumentCaptor;
24 import org.mockito.Mock;
25 import org.mockito.junit.MockitoJUnitRunner;
26 import org.opendaylight.netconf.api.messages.NetconfMessage;
27 import org.opendaylight.netconf.api.xml.XmlUtil;
28 import org.opendaylight.netconf.client.mdsal.AbstractBaseSchemasTest;
29 import org.opendaylight.netconf.client.mdsal.api.NetconfSessionPreferences;
30 import org.opendaylight.netconf.client.mdsal.api.RemoteDeviceCommunicator;
31 import org.opendaylight.netconf.client.mdsal.api.RemoteDeviceId;
32 import org.opendaylight.netconf.client.mdsal.impl.BaseRpcSchemalessTransformer;
33 import org.opendaylight.netconf.client.mdsal.impl.MessageCounter;
34 import org.opendaylight.netconf.client.mdsal.impl.SchemalessMessageTransformer;
35 import org.opendaylight.yangtools.yang.common.QName;
36 import org.opendaylight.yangtools.yang.common.RpcResult;
37
38 @RunWith(MockitoJUnitRunner.StrictStubs.class)
39 public class SchemalessNetconfDeviceRpcTest extends AbstractBaseSchemasTest {
40     @Mock
41     private RemoteDeviceCommunicator listener;
42
43     private SchemalessNetconfDeviceRpc deviceRpc;
44
45     @Before
46     public void setUp() throws Exception {
47         RpcResult<NetconfMessage> msg = null;
48         ListenableFuture<RpcResult<NetconfMessage>> future = Futures.immediateFuture(msg);
49         doReturn(future).when(listener).sendRequest(any(), any());
50         final MessageCounter counter = new MessageCounter();
51         deviceRpc = new SchemalessNetconfDeviceRpc(
52                 new RemoteDeviceId("device1", InetSocketAddress.createUnresolved("0.0.0.0", 17830)), listener,
53                 new BaseRpcSchemalessTransformer(
54                     BASE_SCHEMAS.baseSchemaForCapabilities(NetconfSessionPreferences.fromStrings(Set.of())), counter),
55                 new SchemalessMessageTransformer(counter));
56     }
57
58     @Test
59     public void testInvokeRpc() throws Exception {
60         final QName qName = QName.create("urn:ietf:params:xml:ns:netconf:base:1.0", "2011-06-01", "get-config");
61         DOMSource src = new DOMSource(XmlUtil.readXmlToDocument("<get-config xmlns=\"dd\">\n"
62                 + "    <source>\n"
63                 + "      <running/>\n"
64                 + "    </source>\n"
65                 + "    <filter type=\"subtree\">\n"
66                 + "      <mainroot xmlns=\"urn:dummy:mod-0\">\n"
67                 + "        <maincontent/>\n"
68                 + "<choiceList></choiceList>\n"
69                 + "      </mainroot>\n"
70                 + "    </filter>\n"
71                 + "  </get-config>"));
72         deviceRpc.schemalessRpcService().invokeRpc(qName, src);
73         final var msgCaptor = ArgumentCaptor.forClass(NetconfMessage.class);
74         final var qnameCaptor = ArgumentCaptor.forClass(QName.class);
75         verify(listener).sendRequest(msgCaptor.capture(), qnameCaptor.capture());
76         assertEquals("""
77             <rpc xmlns="urn:ietf:params:xml:ns:netconf:base:1.0" message-id="m-0">
78                 <get-config xmlns="dd">
79                     <source>
80                         <running/>
81                     </source>
82                     <filter type="subtree">
83                         <mainroot xmlns="urn:dummy:mod-0">
84                             <maincontent/>
85                             <choiceList/>
86                         </mainroot>
87                     </filter>
88                 </get-config>
89             </rpc>
90             """, XmlUtil.toString(msgCaptor.getValue().getDocument()));
91     }
92 }