opendaylight/md-sal/sal-rest-connector/src/test/java/org/opendaylight/controller/sal/restconf/impl/test/InvokeRpcMethodTest.java

   1 package org.opendaylight.controller.sal.restconf.impl.test;
   2
   3 import static org.junit.Assert.assertEquals;
   4 import static org.junit.Assert.assertNotNull;
   5 import static org.junit.Assert.assertTrue;
   6 import static org.mockito.Matchers.any;
   7 import static org.mockito.Mockito.mock;
   8 import static org.mockito.Mockito.when;
   9
  10 import java.net.URI;
  11 import java.net.URISyntaxException;
  12 import java.util.Set;
  13
  14 import org.junit.BeforeClass;
  15 import org.junit.Test;
  16 import org.mockito.invocation.InvocationOnMock;
  17 import org.mockito.stubbing.Answer;
  18 import org.opendaylight.controller.sal.restconf.impl.BrokerFacade;
  19 import org.opendaylight.controller.sal.restconf.impl.ControllerContext;
  20 import org.opendaylight.controller.sal.restconf.impl.RestconfImpl;
  21 import org.opendaylight.controller.sal.restconf.impl.StructuredData;
  22 import org.opendaylight.yangtools.yang.common.QName;
  23 import org.opendaylight.yangtools.yang.common.RpcResult;
  24 import org.opendaylight.yangtools.yang.data.api.CompositeNode;
  25 import org.opendaylight.yangtools.yang.data.api.ModifyAction;
  26 import org.opendaylight.yangtools.yang.data.api.MutableCompositeNode;
  27 import org.opendaylight.yangtools.yang.data.api.MutableSimpleNode;
  28 import org.opendaylight.yangtools.yang.data.api.Node;
  29 import org.opendaylight.yangtools.yang.data.impl.NodeFactory;
  30 import org.opendaylight.yangtools.yang.model.api.Module;
  31
  32 public class InvokeRpcMethodTest {
  33
  34     private static Set<Module> modules;
  35
  36     private class AnswerImpl implements Answer<RpcResult<CompositeNode>> {
  37         @Override
  38         public RpcResult<CompositeNode> answer(InvocationOnMock invocation) throws Throwable {
  39             CompositeNode compNode = (CompositeNode) invocation.getArguments()[1];
  40             return new DummyRpcResult.Builder<CompositeNode>().result(compNode).isSuccessful(true).build();
  41         }
  42     }
  43
  44     @BeforeClass
  45     public static void initialization() {
  46         modules = TestUtils.loadModulesFrom("/invoke-rpc");
  47         assertEquals(1, modules.size());
  48         Module module = TestUtils.resolveModule("invoke-rpc-module", modules);
  49         assertNotNull(module);
  50     }
  51
  52     /**
  53      * Test method invokeRpc in RestconfImpl class tests if composite node as
  54      * input parameter of method invokeRpc (second argument) is wrapped to
  55      * parent composite node which has QName equals to QName of rpc (resolved
  56      * from string - first argument).
  57      */
  58     @Test
  59     public void invokeRpcMethodTest() {
  60         ControllerContext contContext = ControllerContext.getInstance();
  61         contContext.onGlobalContextUpdated(TestUtils.loadSchemaContext(modules));
  62         try {
  63             contContext.findModuleByNamespace(new URI("invoke:rpc:module"));
  64         } catch (URISyntaxException e) {
  65             assertTrue("Uri wasn't created sucessfuly", false);
  66         }
  67
  68         BrokerFacade mockedBrokerFacade = mock(BrokerFacade.class);
  69
  70         RestconfImpl restconf = RestconfImpl.getInstance();
  71         restconf.setBroker(mockedBrokerFacade);
  72         restconf.setControllerContext(contContext);
  73
  74         when(mockedBrokerFacade.invokeRpc(any(QName.class), any(CompositeNode.class))).thenAnswer(new AnswerImpl());
  75
  76         StructuredData structData = restconf.invokeRpc("invoke-rpc-module:rpc-test", preparePayload());
  77
  78         CompositeNode rpcCompNode = structData.getData();
  79         CompositeNode cont = null;
  80         assertEquals("invoke:rpc:module", rpcCompNode.getNodeType().getNamespace().toString());
  81         assertEquals("rpc-test", rpcCompNode.getNodeType().getLocalName());
  82
  83         for (Node<?> node : rpcCompNode.getChildren()) {
  84             if (node.getNodeType().getLocalName().equals("cont")
  85                     && node.getNodeType().getNamespace().toString().equals("nmspc")) {
  86                 if (node instanceof CompositeNode) {
  87                     cont = (CompositeNode) node;
  88                 }
  89             }
  90         }
  91         assertNotNull(cont);
  92
  93     }
  94
  95     private CompositeNode preparePayload() {
  96         MutableCompositeNode cont = NodeFactory.createMutableCompositeNode(
  97                 TestUtils.buildQName("cont", "nmspc", "2013-12-04"), null, null, ModifyAction.CREATE, null);
  98         MutableSimpleNode<?> lf = NodeFactory.createMutableSimpleNode(
  99                 TestUtils.buildQName("lf", "nmspc", "2013-12-04"), cont, "any value", ModifyAction.CREATE, null);
 100         cont.getChildren().add(lf);
 101         cont.init();
 102
 103         return cont;
 104     }
 105
 106 }