+
+ @Test
+ public void testStoreAndCloseOrFail() throws Exception {
+ try (final RpcContext rpcContext = new RpcContextImpl(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 {}