2 * Copyright (c) 2014, 2017 Cisco Systems, Inc. and others. All rights reserved.
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
8 package org.opendaylight.controller.remote.rpc;
10 import static org.junit.Assert.assertEquals;
11 import static org.junit.Assert.assertTrue;
12 import static org.mockito.ArgumentMatchers.any;
13 import static org.mockito.ArgumentMatchers.eq;
14 import static org.mockito.Mockito.doReturn;
15 import static org.mockito.Mockito.mock;
17 import akka.actor.Status.Failure;
18 import java.time.Duration;
19 import org.junit.Test;
20 import org.opendaylight.controller.remote.rpc.messages.ExecuteRpc;
21 import org.opendaylight.controller.remote.rpc.messages.RpcResponse;
22 import org.opendaylight.mdsal.dom.api.DOMRpcException;
23 import org.opendaylight.mdsal.dom.api.DOMRpcImplementationNotAvailableException;
24 import org.opendaylight.mdsal.dom.api.DOMRpcResult;
25 import org.opendaylight.mdsal.dom.spi.DefaultDOMRpcResult;
26 import org.opendaylight.yangtools.util.concurrent.FluentFutures;
27 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
29 public class OpsBrokerTest extends AbstractOpsTest {
31 public void testExecuteRpc() {
32 final DOMRpcResult rpcResult = new DefaultDOMRpcResult(makeRPCOutput("bar"));
33 doReturn(FluentFutures.immediateFluentFuture(rpcResult)).when(domRpcService1)
34 .invokeRpc(eq(TEST_RPC_TYPE), any());
35 final ExecuteRpc executeRpc = ExecuteRpc.from(TEST_RPC_ID, mock(ContainerNode.class));
37 rpcInvoker1.tell(executeRpc, rpcRegistry1Probe.getRef());
39 final RpcResponse rpcResponse = rpcRegistry1Probe.expectMsgClass(Duration.ofSeconds(5), RpcResponse.class);
41 assertEquals(rpcResult.getResult(), rpcResponse.getOutput());
45 public void testExecuteRpcFailureWithException() {
46 doReturn(FluentFutures.immediateFailedFluentFuture(new DOMRpcImplementationNotAvailableException("NOT FOUND")))
47 .when(domRpcService1).invokeRpc(eq(TEST_RPC_TYPE), any());
49 final ExecuteRpc executeMsg = ExecuteRpc.from(TEST_RPC_ID, mock(ContainerNode.class));
51 rpcInvoker1.tell(executeMsg, rpcRegistry1Probe.getRef());
53 final Failure rpcResponse = rpcRegistry1Probe.expectMsgClass(Duration.ofSeconds(5), Failure.class);
55 assertTrue(rpcResponse.cause() instanceof DOMRpcException);