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.when;
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 {
32 public void testExecuteRpc() {
33 final ContainerNode invokeRpcResult = makeRPCOutput("bar");
34 final DOMRpcResult rpcResult = new DefaultDOMRpcResult(invokeRpcResult);
35 doReturn(FluentFutures.immediateFluentFuture(rpcResult)).when(domRpcService1)
36 .invokeRpc(eq(TEST_RPC_TYPE), any());
37 final ExecuteRpc executeRpc = ExecuteRpc.from(TEST_RPC_ID, null);
39 rpcInvoker1.tell(executeRpc, rpcRegistry1Probe.getRef());
41 final RpcResponse rpcResponse = rpcRegistry1Probe.expectMsgClass(Duration.ofSeconds(5), RpcResponse.class);
43 assertEquals(rpcResult.getResult(), rpcResponse.getOutput());
47 public void testExecuteRpcFailureWithException() {
48 when(domRpcService1.invokeRpc(eq(TEST_RPC_TYPE), any())).thenReturn(FluentFutures.immediateFailedFluentFuture(
49 new DOMRpcImplementationNotAvailableException("NOT FOUND")));
51 final ExecuteRpc executeMsg = ExecuteRpc.from(TEST_RPC_ID, null);
53 rpcInvoker1.tell(executeMsg, rpcRegistry1Probe.getRef());
55 final Failure rpcResponse = rpcRegistry1Probe.expectMsgClass(Duration.ofSeconds(5), Failure.class);
57 assertTrue(rpcResponse.cause() instanceof DOMRpcException);