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