*/
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.Mockito.verify;
import static org.mockito.Mockito.when;
+
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
+import org.mockito.Mockito;
import org.mockito.runners.MockitoJUnitRunner;
import org.opendaylight.controller.md.sal.binding.api.NotificationPublishService;
import org.opendaylight.controller.sal.binding.api.BindingAwareBroker;
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.device.XidSequencer;
import org.opendaylight.openflowplugin.api.openflow.rpc.RpcContext;
import org.opendaylight.openflowplugin.api.openflow.statistics.ofpspecific.MessageSpy;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeContext;
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.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 BindingAwareBroker.ProviderContext rpcProviderRegistry;
@Mock
private DeviceState deviceState;
@Mock
- private DeviceContext deviceContext;
+ private XidSequencer xidSequencer;
@Mock
private MessageSpy messageSpy;
@Mock
+ private DeviceContext deviceContext;
+ @Mock
+ private BindingAwareBroker.RoutedRpcRegistration<TestRpcService> routedRpcReg;
+ @Mock
private NotificationPublishService notificationPublishService;
+ @Mock
+ private TestRpcService serviceInstance;
+ @Mock
+ private DeviceInfo deviceInfo;
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);
- }
- @Test
- public void invokeRpcTest() {
+ rpcContext = new RpcContextImpl(
+ deviceInfo,
+ rpcProviderRegistry,
+ deviceContext,
+ messageSpy,
+ MAX_REQUESTS,
+ nodeInstanceIdentifier);
+
+ when(rpcProviderRegistry.addRoutedRpcImplementation(TestRpcService.class, serviceInstance)).thenReturn(routedRpcReg);
}
@Test
public void testStoreOrFail() throws Exception {
- try (final RpcContext rpcContext = new RpcContextImpl(mockedRpcProviderRegistry, deviceContext,
- 100, false, notificationPublishService)) {
+ try (final RpcContext rpcContext = new RpcContextImpl(
+ deviceInfo,
+ rpcProviderRegistry,
+ xidSequencer,
+ messageSpy,
+ 100,
+ nodeInstanceIdentifier)) {
final RequestContext<?> requestContext = rpcContext.createRequestContext();
assertNotNull(requestContext);
}
@Test
public void testStoreOrFailThatFails() throws Exception {
- try (final RpcContext rpcContext = new RpcContextImpl(mockedRpcProviderRegistry, deviceContext, 0,
- false, notificationPublishService)) {
+ try (final RpcContext rpcContext = new RpcContextImpl(
+ deviceInfo,
+ rpcProviderRegistry,
+ xidSequencer,
+ messageSpy,
+ 0,
+ nodeInstanceIdentifier)) {
final RequestContext<?> requestContext = rpcContext.createRequestContext();
assertNull(requestContext);
}
}
+
+ @Test
+ public void testStoreAndCloseOrFail() throws Exception {
+ try (final RpcContext rpcContext = new RpcContextImpl(
+ deviceInfo,
+ rpcProviderRegistry,
+ deviceContext,
+ messageSpy,
+ 100,
+ nodeInstanceIdentifier)) {
+ final RequestContext<?> requestContext = rpcContext.createRequestContext();
+ assertNotNull(requestContext);
+ requestContext.close();
+ verify(messageSpy).spyMessage(RpcContextImpl.class, MessageSpy.STATISTIC_GROUP.REQUEST_STACK_FREED);
+ }
+ }
+
+ public void testRegisterRpcServiceImplementation() {
+ rpcContext.registerRpcServiceImplementation(TestRpcService.class, serviceInstance);
+ verify(rpcProviderRegistry, Mockito.times(1)).addRoutedRpcImplementation(TestRpcService.class,serviceInstance);
+ verify(routedRpcReg,Mockito.times(1)).registerPath(NodeContext.class,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() {
+ rpcContext.registerRpcServiceImplementation(TestRpcService.class, serviceInstance);
+ rpcContext.close();
+ assertEquals(rpcContext.isEmptyRpcRegistrations(), true);
+ }
+
+ /**
+ * When deviceContext.reserveXidForDeviceMessage returns null, null should be returned
+ * @throws InterruptedException
+ */
+ @Test
+ public void testCreateRequestContext1() throws InterruptedException {
+ when(deviceContext.reserveXidForDeviceMessage()).thenReturn(null);
+ assertEquals(rpcContext.createRequestContext(),null);
+ }
+
+ /**
+ * When deviceContext.reserveXidForDeviceMessage returns value, AbstractRequestContext should be returned
+ * @throws InterruptedException
+ */
+
+ @Test
+ public void testCreateRequestContext2() throws InterruptedException {
+ RequestContext temp = rpcContext.createRequestContext();
+ temp.close();
+ verify(messageSpy).spyMessage(RpcContextImpl.class,MessageSpy.STATISTIC_GROUP.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 {}
}