-/**
+/*
* Copyright (c) 2015 Cisco Systems, Inc. and others. All rights reserved.
*
* This program and the accompanying materials are made available under the
*/
package org.opendaylight.openflowplugin.impl.rpc;
+import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertNull;
+import static org.mockito.ArgumentMatchers.anySet;
+import static org.mockito.ArgumentMatchers.eq;
+import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;
+
+import com.google.common.collect.ImmutableSet;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
-import org.mockito.runners.MockitoJUnitRunner;
-import org.opendaylight.controller.md.sal.binding.api.NotificationPublishService;
-import org.opendaylight.controller.sal.binding.api.BindingAwareBroker;
+import org.mockito.Mockito;
+import org.mockito.junit.MockitoJUnitRunner;
+import org.opendaylight.mdsal.binding.api.NotificationPublishService;
+import org.opendaylight.mdsal.binding.api.RpcProviderService;
+import org.opendaylight.openflowplugin.api.openflow.FlowGroupCacheManager;
import org.opendaylight.openflowplugin.api.openflow.device.DeviceContext;
+import org.opendaylight.openflowplugin.api.openflow.device.DeviceInfo;
import org.opendaylight.openflowplugin.api.openflow.device.DeviceState;
import org.opendaylight.openflowplugin.api.openflow.device.RequestContext;
import org.opendaylight.openflowplugin.api.openflow.rpc.RpcContext;
import org.opendaylight.openflowplugin.api.openflow.statistics.ofpspecific.MessageSpy;
+import org.opendaylight.openflowplugin.extension.api.core.extension.ExtensionConverterProvider;
+import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.ConvertorExecutor;
import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
+import org.opendaylight.yangtools.concepts.ObjectRegistration;
import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
import org.opendaylight.yangtools.yang.binding.KeyedInstanceIdentifier;
+import org.opendaylight.yangtools.yang.binding.RpcService;
-/**
- * @author joe
- */
@RunWith(MockitoJUnitRunner.class)
public class RpcContextImplTest {
+ private static final int MAX_REQUESTS = 5;
+ private RpcContextImpl rpcContext;
+
@Mock
- private BindingAwareBroker.ProviderContext mockedRpcProviderRegistry;
+ private FlowGroupCacheManager flowGroupCacheManager;
+ @Mock
+ private RpcProviderService rpcProviderRegistry;
@Mock
private DeviceState deviceState;
@Mock
+ private MessageSpy messageSpy;
+ @Mock
private DeviceContext deviceContext;
@Mock
- private MessageSpy messageSpy;
+ private ObjectRegistration<TestRpcService> routedRpcReg;
+
@Mock
private NotificationPublishService notificationPublishService;
+ @Mock
+ private TestRpcService serviceInstance;
+ @Mock
+ private DeviceInfo deviceInfo;
+ @Mock
+ private ExtensionConverterProvider extensionConverterProvider;
+ @Mock
+ private ConvertorExecutor convertorExecutor;
private KeyedInstanceIdentifier<Node, NodeKey> nodeInstanceIdentifier;
final NodeId nodeId = new NodeId("openflow:1");
nodeInstanceIdentifier = InstanceIdentifier.create(Nodes.class).child(Node.class, new NodeKey(nodeId));
- when(deviceState.getNodeInstanceIdentifier()).thenReturn(nodeInstanceIdentifier);
- when(deviceContext.getDeviceState()).thenReturn(deviceState);
- }
+ when(deviceInfo.getNodeInstanceIdentifier()).thenReturn(nodeInstanceIdentifier);
+ when(deviceContext.getMessageSpy()).thenReturn(messageSpy);
+ when(deviceContext.getDeviceInfo()).thenReturn(deviceInfo);
+ when(deviceInfo.getNodeInstanceIdentifier()).thenReturn(nodeInstanceIdentifier);
- @Test
- public void invokeRpcTest() {
+ rpcContext = new RpcContextImpl(
+ rpcProviderRegistry,
+ MAX_REQUESTS,
+ deviceContext,
+ extensionConverterProvider,
+ convertorExecutor,
+ notificationPublishService, true, flowGroupCacheManager);
+ when(rpcProviderRegistry.registerRpcImplementation(eq(TestRpcService.class), eq(serviceInstance), anySet()))
+ .thenReturn(routedRpcReg);
}
@Test
- public void testStoreOrFail() throws Exception {
- try (final RpcContext rpcContext = new RpcContextImpl(messageSpy, mockedRpcProviderRegistry, deviceContext,
- 100, false, notificationPublishService)) {
+ public void testStoreOrFail() {
+ try (RpcContext rpcContext = new RpcContextImpl(
+ rpcProviderRegistry,
+ 100,
+ deviceContext,
+ extensionConverterProvider,
+ convertorExecutor,
+ notificationPublishService, true, flowGroupCacheManager)) {
final RequestContext<?> requestContext = rpcContext.createRequestContext();
assertNotNull(requestContext);
}
}
@Test
- public void testStoreOrFailThatFails() throws Exception {
- try (final RpcContext rpcContext = new RpcContextImpl(messageSpy, mockedRpcProviderRegistry, deviceContext, 0,
- false, notificationPublishService)) {
+ public void testStoreOrFailThatFails() {
+ try (RpcContext rpcContext = new RpcContextImpl(
+ rpcProviderRegistry,
+ 0,
+ deviceContext,
+ extensionConverterProvider,
+ convertorExecutor,
+ notificationPublishService, true, flowGroupCacheManager)) {
final RequestContext<?> requestContext = rpcContext.createRequestContext();
assertNull(requestContext);
}
}
+
+ @Test
+ public void testStoreAndCloseOrFail() {
+ try (RpcContext rpcContext = new RpcContextImpl(
+ rpcProviderRegistry,
+ 100,
+ deviceContext,
+ extensionConverterProvider,
+ convertorExecutor,
+ notificationPublishService, true, flowGroupCacheManager)) {
+ final RequestContext<?> requestContext = rpcContext.createRequestContext();
+ assertNotNull(requestContext);
+ requestContext.close();
+ verify(messageSpy).spyMessage(RpcContextImpl.class, MessageSpy.StatisticsGroup.REQUEST_STACK_FREED);
+ }
+ }
+
+ public void testRegisterRpcServiceImplementation() {
+ rpcContext.registerRpcServiceImplementation(TestRpcService.class, serviceInstance);
+ verify(rpcProviderRegistry, Mockito.times(1)).registerRpcImplementation(TestRpcService.class, serviceInstance,
+ ImmutableSet.of(nodeInstanceIdentifier));
+ assertEquals(rpcContext.isEmptyRpcRegistrations(), false);
+ }
+
+ @Test
+ public void testLookupRpcService() {
+ when(routedRpcReg.getInstance()).thenReturn(serviceInstance);
+ rpcContext.registerRpcServiceImplementation(TestRpcService.class, serviceInstance);
+ TestRpcService temp = rpcContext.lookupRpcService(TestRpcService.class);
+ assertEquals(serviceInstance,temp);
+ }
+
+ @Test
+ public void testClose() {
+ when(routedRpcReg.getInstance()).thenReturn(serviceInstance);
+ rpcContext.registerRpcServiceImplementation(TestRpcService.class, serviceInstance);
+ rpcContext.close();
+ assertEquals(rpcContext.isEmptyRpcRegistrations(), true);
+ }
+
+ /**
+ * When deviceContext.reserveXidForDeviceMessage returns null, null should be returned.
+ */
+ @Test
+ public void testCreateRequestContext1() {
+ when(deviceInfo.reserveXidForDeviceMessage()).thenReturn(null);
+ assertEquals(rpcContext.createRequestContext(),null);
+ }
+
+ /**
+ * When deviceContext.reserveXidForDeviceMessage returns value, AbstractRequestContext should be returned.
+ */
+
+ @Test
+ public void testCreateRequestContext2() {
+ RequestContext temp = rpcContext.createRequestContext();
+ temp.close();
+ verify(messageSpy).spyMessage(RpcContextImpl.class, MessageSpy.StatisticsGroup.REQUEST_STACK_FREED);
+ }
+
+ @Test
+ public void testUnregisterRpcServiceImpl() {
+ rpcContext.registerRpcServiceImplementation(TestRpcService.class, serviceInstance);
+ assertEquals(rpcContext.isEmptyRpcRegistrations(), false);
+ rpcContext.unregisterRpcServiceImplementation(TestRpcService.class);
+ assertEquals(rpcContext.isEmptyRpcRegistrations(), true);
+ }
+
+ //Stub for RpcService class.
+ public class TestRpcService implements RpcService {
+
+ }
}