opendaylight/md-sal/sal-remoterpc-connector/src/test/java/org/opendaylight/controller/remote/rpc/RpcBrokerTest.java

   1 /*
   2  * Copyright (c) 2014 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
   9 package org.opendaylight.controller.remote.rpc;
  10
  11
  12 import static org.junit.Assert.assertEquals;
  13 import static org.mockito.Matchers.eq;
  14 import static org.mockito.Mockito.when;
  15
  16 import akka.actor.Status.Failure;
  17 import akka.testkit.JavaTestKit;
  18 import com.google.common.util.concurrent.Futures;
  19 import org.junit.Assert;
  20 import org.junit.Test;
  21 import org.mockito.Mockito;
  22 import org.opendaylight.controller.cluster.datastore.node.utils.serialization.NormalizedNodeSerializer;
  23 import org.opendaylight.controller.md.sal.dom.api.DOMRpcException;
  24 import org.opendaylight.controller.md.sal.dom.api.DOMRpcImplementationNotAvailableException;
  25 import org.opendaylight.controller.md.sal.dom.api.DOMRpcResult;
  26 import org.opendaylight.controller.md.sal.dom.spi.DefaultDOMRpcResult;
  27 import org.opendaylight.controller.remote.rpc.messages.ExecuteRpc;
  28 import org.opendaylight.controller.remote.rpc.messages.RpcResponse;
  29 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
  30 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  31
  32 public class RpcBrokerTest extends AbstractRpcTest {
  33
  34     @Test
  35     public void testExecuteRpc() {
  36         new JavaTestKit(node1) {
  37             {
  38
  39                 final ContainerNode invokeRpcResult = makeRPCOutput("bar");
  40                 final DOMRpcResult rpcResult = new DefaultDOMRpcResult(invokeRpcResult);
  41                 when(domRpcService1.invokeRpc(eq(TEST_RPC_TYPE), Mockito.<NormalizedNode<?, ?>>any())).thenReturn(
  42                         Futures.<DOMRpcResult, DOMRpcException>immediateCheckedFuture(rpcResult));
  43
  44                 final ExecuteRpc executeMsg = ExecuteRpc.from(TEST_RPC_ID, null);
  45
  46                 rpcBroker1.tell(executeMsg, getRef());
  47
  48                 final RpcResponse rpcResponse = expectMsgClass(duration("5 seconds"), RpcResponse.class);
  49
  50                 assertEquals(rpcResult.getResult(),
  51                         NormalizedNodeSerializer.deSerialize(rpcResponse.getResultNormalizedNode()));
  52             }
  53         };
  54     }
  55
  56     @Test
  57     public void testExecuteRpcFailureWithException() {
  58
  59         new JavaTestKit(node1) {
  60             {
  61
  62                 when(domRpcService1.invokeRpc(eq(TEST_RPC_TYPE), Mockito.<NormalizedNode<?, ?>>any()))
  63                         .thenReturn(
  64                                 Futures.<DOMRpcResult, DOMRpcException>immediateFailedCheckedFuture(new DOMRpcImplementationNotAvailableException(
  65                                         "NOT FOUND")));
  66
  67                 final ExecuteRpc executeMsg = ExecuteRpc.from(TEST_RPC_ID, null);
  68
  69                 rpcBroker1.tell(executeMsg, getRef());
  70
  71                 final Failure rpcResponse = expectMsgClass(duration("5 seconds"), akka.actor.Status.Failure.class);
  72
  73                 Assert.assertTrue(rpcResponse.cause() instanceof DOMRpcException);
  74             }
  75         };
  76
  77     }
  78
  79 }