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
9 package org.opendaylight.controller.remote.rpc;
11 import akka.actor.ActorRef;
12 import akka.actor.ActorSystem;
13 import akka.testkit.javadsl.TestKit;
14 import com.typesafe.config.ConfigFactory;
15 import java.util.Collections;
16 import org.junit.AfterClass;
17 import org.junit.BeforeClass;
18 import org.junit.Test;
19 import org.opendaylight.controller.remote.rpc.registry.RpcRegistry;
20 import org.opendaylight.mdsal.dom.api.DOMRpcIdentifier;
21 import org.opendaylight.yangtools.yang.common.QName;
22 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
23 import org.opendaylight.yangtools.yang.model.api.SchemaPath;
25 public class RpcListenerTest {
27 private static final QName TEST_QNAME = QName.create("test", "2015-06-12", "test");
28 private static final SchemaPath RPC_TYPE = SchemaPath.create(true, TEST_QNAME);
29 private static final YangInstanceIdentifier TEST_PATH = YangInstanceIdentifier
30 .create(new YangInstanceIdentifier.NodeIdentifier(TEST_QNAME));
31 private static final DOMRpcIdentifier RPC_ID = DOMRpcIdentifier.create(RPC_TYPE, TEST_PATH);
32 static ActorSystem system;
36 public static void setup() {
37 system = ActorSystem.create("opendaylight-rpc", ConfigFactory.load().getConfig("odl-cluster-rpc"));
41 public static void teardown() {
42 TestKit.shutdownActorSystem(system);
47 public void testRouteAdd() {
51 final TestKit probeReg = new TestKit(system);
52 final ActorRef rpcRegistry = probeReg.getRef();
54 final RpcListener rpcListener = new RpcListener(rpcRegistry);
55 rpcListener.onRpcAvailable(Collections.singleton(RPC_ID));
56 probeReg.expectMsgClass(RpcRegistry.Messages.AddOrUpdateRoutes.class);
62 public void testRouteRemove() {
66 final TestKit probeReg = new TestKit(system);
67 final ActorRef rpcRegistry = probeReg.getRef();
69 final RpcListener rpcListener = new RpcListener(rpcRegistry);
70 rpcListener.onRpcUnavailable(Collections.singleton(RPC_ID));
71 probeReg.expectMsgClass(RpcRegistry.Messages.RemoveRoutes.class);